Update progress bar after each processed module

Author: khalim19

pypredef_generator_plugin.py

@@ -3,10 +3,13 @@
 (Eclipse IDE plug-in) for GIMP modules.
 """
 
+from __future__ import absolute_import, print_function, division
+
 import os
 
 import importlib
 
+import gimp
 import gimpfu
 
 import pypredef_generator
@@ -18,15 +21,22 @@
 def generate_predefined_completions_for_pydev(generate_from_modules, generate_from_pdb):
   if generate_from_modules:
     module_names = _get_module_names(pypredef_generator.MODULES_FILE_PATH)
-    
+  
+  gimp_progress = GimpProgress(
+    _get_num_progress_items(generate_from_modules, module_names, generate_from_pdb))
+  gimp_progress.initialize()
+  
+  if generate_from_modules:
     _make_dirs(pypredef_generator.PYPREDEF_FILES_DIR)
 
     for module_name in module_names:
       module = importlib.import_module(module_name)
       pypredef_generator.generate_predefined_completions(module)
+      gimp_progress.update()
 
   if generate_from_pdb:
     pypredef_generator_pdb.generate_predefined_completions_for_gimp_pdb()
+    gimp_progress.update()
 
 
 def _get_module_names(modules_file_path):
@@ -56,6 +66,43 @@ def _make_dirs(path):
     else:
       raise
 
+
+#===============================================================================
+
+
+class GimpProgress(object):
+  
+  def __init__(self, num_total_tasks=0):
+    self.num_total_tasks = num_total_tasks
+    self._num_finished_tasks = 0
+  
+  @property
+  def num_finished_tasks(self):
+    return self._num_finished_tasks
+  
+  def initialize(self, message=None):
+    gimp.progress_init(message if message is not None else "")
+  
+  def update(self, num_tasks=1):
+    if self._num_finished_tasks + num_tasks > self.num_total_tasks:
+      raise ValueError("number of finished tasks exceeds the number of total tasks")
+    
+    self._num_finished_tasks += num_tasks
+    
+    gimp.progress_update(self._num_finished_tasks / self.num_total_tasks)
+  
+
+def _get_num_progress_items(generate_from_modules, module_names, generate_from_pdb):
+  num_progress_items = 0
+  if generate_from_modules:
+    num_progress_items += len(module_names)
+  
+  if generate_from_pdb:
+    num_progress_items += 1
+  
+  return num_progress_items
+
+
 #===============================================================================
 
 gimpfu.register(