Changeset 62374 for grass/trunk

Timestamp:

Oct 24, 2014, 2:50:38 AM (10 years ago)

Author:

zarch

Message:

pygrass: back to a pickable Module instance

File:

• grass/trunk/lib/python/pygrass/modules/interface/module.py (modified) (6 diffs)

grass/trunk/lib/python/pygrass/modules/interface/module.py

@@ -4 +4 @@
 import sys
 from multiprocessing import cpu_count
+from functools import wraps
 
 if sys.version_info[0] == 2:
@@ -20 +21 @@
 from grass.pygrass.modules.interface.typedict import TypeDict
 from grass.pygrass.modules.interface.read import GETFROMTAG, DOC
-from grass.pygrass.messages import Messenger
+from grass.pygrass.messages import get_msgr
+
+
+def mdebug(level, msg='', extra=None):
+    """Debug decorators for class methods.
+
+    :param level: the debug level
+    :type level: int
+    :param msg: Debug message
+    :type msg: str
+    :param extra: Function that return a string
+    :type msg: func
+    """
+    msgr = get_msgr()
+
+    def decorator(method):
+
+        @wraps(method)
+        def wrapper(self, *args, **kargs):
+            sargs = ', ' + ' , '.join([repr(a) for a in args]) if args else ''
+            skargs = (' , '.join(['%s=%r' % (k, v) for k, v in kargs.items()])
+                      if kargs else '')
+            opts = "%s%s%s" % (sargs, ',' if sargs and skargs else '', skargs)
+            dmsg = "%s.%s(self%s): %s %s" % (self.__class__.__name__,
+                                             method.__name__,
+                                             opts, msg,
+                                             extra(self, *args, **kargs)
+                                             if extra else '')
+            msgr.debug(level, dmsg)
+            return method(self, *args, **kargs)
+        return wrapper
+    return decorator
+
+
+def _get_bash(self, *args, **kargs):
+    return self.get_bash()
+
 
 class ParallelModuleQueue(object):
@@ -338 +375 @@
     """
     def __init__(self, cmd, *args, **kargs):
-        self._msgr = Messenger()
-        
         if isinstance(cmd, unicode):
             self.name = str(cmd)
@@ -422 +457 @@
             self.run()
             return self
-        
+
         #
         # check for extra kargs, set attribute and remove from dictionary
@@ -457 +492 @@
             else:
                 raise ParameterError('%s is not a valid parameter.' % key)
-
-        #
-        # print debug message
-        #
-        self._msgr.debug(1, "Module.__call__(): %s" % (self.get_bash()))
 
         #
@@ -559 +589 @@
         return args
 
-    def run(self, node=None):
+    @mdebug(1, extra=_get_bash)
+    def run(self):
         """Run the module