Changeset 285

Timestamp:

02/13/06 22:12:11 (3 years ago)

Author:

athomas

Message:

• Added logger mixin class.
  
• Fixed bug with empty columns in print_table.
  
• Added console.getch() for retrieving a single character from stdin in character buffered mode.
  
• Added loader.load_all() for loading all sub-modules of an existing module. eg. import plugins; load_all(plugins)
  

Files:

• pycrash/trunk/crash/console.py (modified) (3 diffs)
• pycrash/trunk/crash/loader.py (modified) (1 diff)
• pycrash/trunk/crash/logger.py (added)

pycrash/trunk/crash/console.py

@@ -204 +204 @@
-    colwidths = [0] * cols
-    for i in range(0, cols):
-
+[0] + 
-        if i < cols - 1:
-            colwidths[i] += seplen
@@ -234 +234 @@
-        xrow = [cwraptext(col.replace('^R', fmt), colwidths[i] - (i < cols - 1 and seplen or 0)) for i, col in enumerate(row)]
-        # Pad column rows out to the maximum of all columns
-
+[0] + 
-        for col in xrow:
-            col += [''] * (maxrows - len(col))
-
+[0] + 
-        # Emit
-        for i in range(0, realrows):
@@ -248 +248 @@
-            cwrite(sys.stdout, '^N\n')
-        rowalt += 1
+
+try:
+    import msvcrt
+    def getch():
+        return msvcrt.getch()
+except ImportError:
+    import tty, sys, termios
+
+    def getch():
+        fd = sys.stdin.fileno()
+        old_settings = termios.tcgetattr(fd)
+        try:
+            tty.setraw(sys.stdin.fileno())
+            ch = sys.stdin.read(1)
+        finally:
+            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+        return ch
+
+getch.__doc__ = " Read a single character from stdin. "

pycrash/trunk/crash/loader.py

@@ -49 +49 @@
-    # Return a reference to the class itself, not an instantiated object.
-    return aClass
+
+def load_all(module):
+    """ Load all python modules in a module directory. Returns a tuple of
+        (success, failed) modules. """
+    from dircache import listdir
+    import os
+    failed = []
+    succeeded = []
+    tried = []
+    for p in module.__path__:
+        for f in listdir(p):
+            fp = os.path.join(p, f)
+            f, ext = os.path.splitext(f)
+            plugin = '.'.join((module.__name__, f))
+            if plugin not in tried and not f.startswith('__') and (ext in 
+                    ('.so', '.py', '.pyc') or os.path.isdir(fp)):
+                tried.append(plugin)
+                try:
+                    m = load_module(plugin)
+                    succeeded.append(m)
+                except Exception, e:
+                    failed.append((plugin, e))
+    return (succeeded, failed)
+