badblocks support -- have to do this before even formatting swap as they

might want to check their swap for badblocks.  also, still needs to
parse the output and have a progress bar post-beta 2

1 files changed, 6 insertions, 1 deletions

diff --git a/partitioning.py b/partitioning.py
index 24aa96b97..49b1bc50c 100644
--- a/partitioning.py
+++ b/partitioning.py
@@ -503,7 +503,7 @@ class PartitionSpec:
                  format = None, options = None, 
                  constraint = None, migrate = None,
                  raidmembers = None, raidlevel = None, 
-                 raidspares = None):
+                 raidspares = None, badblocks = None):
         #
         # requesttype: REQUEST_PREEXIST or REQUEST_NEW or REQUEST_RAID
         #
@@ -529,6 +529,7 @@ class PartitionSpec:
         self.drive = drive
         self.primary = primary
         self.format = format
+        self.badblocks = badblocks
         self.migrate = migrate
         self.options = options
         self.constraint = constraint
@@ -600,6 +601,10 @@ class PartitionSpec:
 
         if self.migrate:
             entry.setMigrate(self.migrate)
+
+        if self.badblocks:
+            entry.setBadblocks(self.badblocks)
+            
         return entry
 
 class Partitions: