Fix bitrot

[invirt/packages/invirt-images.git] / invirt-images

diff --git a/invirt-images b/invirt-images
index ea208a5..ec79d52 100755 (executable)
--- a/invirt-images
+++ b/invirt-images
@@ -41,19 +41,42 @@ def lvcreate(name, size):
         return 6
 
 def lvrename(dest, src):
-    lvr = subprocess.Popen(['lvrename', 'xenvg', src, dest],
+    lvr = subprocess.Popen(['lvchange', '-an', "xenvg/%s" % (src,)],
                            stderr=subprocess.PIPE,
                            stdout=getOutput()['stdout'])
     ret = lvr.wait()
-    if not ret:
-        return 0
-    stderr = lvr.stderr.read()
-    if 'not found in volume group' in stderr:
-        return 0
-    else:
+
+    if ret:
+       stderr = lvr.stderr.read()
+       if 'not found' in stderr:
+           return 0
+       if verbosity > 0:
+           print stderr
+       return ret
+
+    lvr = subprocess.Popen(['lvrename', "xenvg/%s" % (src,), "xenvg/%s" % (dest,)],
+                           stderr=subprocess.PIPE,
+                           stdout=getOutput()['stdout'])
+    ret = lvr.wait()
+
+    if ret:
+        stderr = lvr.stderr.read()
+        if not ('not found in volume group' in stderr):
+            if verbosity > 0:
+                print stderr
+            return ret
+
+    subprocess.Popen(['lvchange', '-ay', "xenvg/%s" % (dest,)],
+                     stderr=subprocess.PIPE,
+                     stdout=getOutput()['stdout'])
+    ret = lvr.wait()
+
+    if ret:
         if verbosity > 0:
-            print stderr
-        return ret
+           print lvr.stderr.read()
+
+    return ret
+
 
 def lv_random(func, pattern, *args):
     """
@@ -257,9 +280,9 @@ def main():
             database.session.commit()
     elif options.action == 'update':
         if len(args) > 0:
-            images = [database.CDROM.query().get(arg) for arg in args]
+            images = [database.CDROM.query.get(arg) for arg in args]
         else:
-            images = database.CDROM.query().all()
+            images = database.CDROM.query.all()
         for cdrom in images:
             if cdrom is not None:
                 load_image(cdrom)