From bf540630104adea0bafb7ae78a704feab7103efc 2005-12-31 22:06:42
From: fperez
Date: 2005-12-31 22:06:42
Subject: [PATCH] for modules which define __all__, complete only on those.  After a patch by
Jeffrey Collins <jcollins_boulder-AT-earthlink.net>.

---

diff --git a/IPython/completer.py b/IPython/completer.py
index 23acd6a..2bc3781 100644
--- a/IPython/completer.py
+++ b/IPython/completer.py
@@ -185,21 +185,29 @@ class Completer:
 
         if not m:
             return []
+        
         expr, attr = m.group(1, 3)
         try:
             object = eval(expr, self.namespace)
         except:
             object = eval(expr, self.global_namespace)
-        words = [w for w in dir(object) if isinstance(w, basestring)]
-        if hasattr(object,'__class__'):
-            words.append('__class__')
-            words.extend(get_class_members(object.__class__))
+
+        # for modules which define __all__, complete only on those.
+        if type(object) == types.ModuleType and hasattr(object, '__all__'):
+            words = getattr(object, '__all__')
+        else:
+            words = dir(object)
+            if hasattr(object,'__class__'):
+                words.append('__class__')
+                words.extend(get_class_members(object.__class__))
+
+        # filter out non-string attributes which may be stuffed by dir() calls
+        # and poor coding in third-party modules
+        words = [w for w in words
+                 if isinstance(w, basestring) and w != "__builtins__"]
+        # Build match list to return
         n = len(attr)
-        matches = []
-        for word in words:
-            if word[:n] == attr and word != "__builtins__":
-                matches.append("%s.%s" % (expr, word))
-        return matches
+        return ["%s.%s" % (expr, w) for w in words if w[:n] == attr ]
 
 class IPCompleter(Completer):
     """Extension of the completer class with IPython-specific features"""
diff --git a/doc/ChangeLog b/doc/ChangeLog
index 21159a0..72a6567 100644
--- a/doc/ChangeLog
+++ b/doc/ChangeLog
@@ -1,5 +1,10 @@
 2005-12-31  Fernando Perez  <Fernando.Perez@colorado.edu>
 
+	* IPython/completer.py (Completer.attr_matches): for modules which
+	define __all__, complete only on those.  After a patch by Jeffrey
+	Collins <jcollins-AT-boulder.net>.  Also, clean up and speed up
+	this routine.
+
 	* IPython/Logger.py (Logger.log): fix a history handling bug.  I
 	don't know if this is the end of it, but the behavior now is
 	certainly much more correct.  Note that coupled with macros,