Show progress while taking snapshot

pisi/api.py

@@ -353,11 +353,15 @@ def snapshot():
     installed. The record is kept by pisi history mechanism as it works automatically on install, remove 
     and upgrade operations.
     """
-    
-    installdb = pisi.db.installdb.InstallDB() 
+
+    installdb = pisi.db.installdb.InstallDB()
     historydb = pisi.db.historydb.HistoryDB()
     historydb.create_history("snapshot")
 
+    li = installdb.list_installed()
+    progress = ctx.ui.Progress(len(li))
+
+    processed = 0
     for name in installdb.list_installed():
         package = installdb.get_package(name)
         historydb.add_package(pkgBefore=package, operation="snapshot")
@@ -366,6 +370,11 @@ def snapshot():
             if f.type == "config" and pisi.util.config_changed(f):
                 historydb.save_config(name, "/%s" % f.path)
 
+        processed += 1
+        ctx.ui.display_progress(operation = "snapshot",
+                                percent = progress.update(processed),
+                                info = _("Taking snapshot of the system"))
+
     historydb.update_history()
 
 # ****** Danger Zone Below! Tressspassers' eyes will explode! ********** #

pisi/ui.py

@@ -26,6 +26,15 @@ class UI(object):
             except ArithmeticError:
                 self.percent = 0
 
+        def update(self, size):
+            if not self.totalsize:
+                return 100
+            try:
+                self.percent = (size * 100) / self.totalsize
+            except ArithmeticError:
+                self.percent = 0
+            return self.percent
+
     def __init__(self, debuggy = False, verbose = False):
         self.show_debug = debuggy
         self.show_verbose = verbose