Support a force=True argument on SwapSpace.create()

force=True will tear down the device, unmount if necessary, and then
reformat it.

1 files changed, 6 insertions, 2 deletions

diff --git a/storage/formats/swap.py b/storage/formats/swap.py
index 392ed11ae..e1dacb09d 100644
--- a/storage/formats/swap.py
+++ b/storage/formats/swap.py
@@ -132,10 +132,14 @@ class SwapSpace(DeviceFormat):
         """ Create the device. """
         log_method_call(self, device=self.device,
                         type=self.type, status=self.status)
-        if self.exists:
+        force = kwargs.get("force")
+
+        if not force and self.exists:
             raise SwapSpaceError("format already exists")
 
-        if self.status:
+        if force:
+            self.teardown()
+        elif self.status:
             raise SwapSpaceError("device exists and is active")
 
         DeviceFormat.create(self, *args, **kwargs)