added __all__ to completer.py and added basic tests for test_dir2

Signed-off-by: Tim Couper <[e-mail]>

IPython/core/completer.py

@@ -336,7 +336,6 b' class Completer(Configurable):'
         #io.rprint('Completer->attr_matches, txt=%r' % text) # dbg
         # Another option, seems to work great. Catches things like ''.<tab>
         m = re.match(r"(\S+(\.\w+)*)\.(\w*)$", text)
-
         if m:
             expr, attr = m.group(1, 3)
         elif self.greedy:
@@ -355,6 +354,9 b' class Completer(Configurable):'
             except:
                 return []
 
+        if self.limit_to__all__ and hasattr(obj, '__all__'):
+            words = get__all__entries(obj)
+        else: 
         words = dir2(obj)
 
         try:
@@ -371,6 +373,16 b' class Completer(Configurable):'
         return res
 
 
+def get__all__entries(obj):
+    """returns the strings in the __all__ attribute"""
+    try:
+        words = getattr(obj,'__all__')
+    except:
+        return []
+    
+    return [w for w in words if isinstance(w, basestring)]
+
+
 class IPCompleter(Completer):
     """Extension of the completer class with IPython-specific features"""
 
@@ -403,6 +415,16 b' class IPCompleter(Completer):'
         When 0: nothing will be excluded.
         """
     )
+    limit_to__all__ = Enum((0,1), default_value=1, config=True,
+        help="""Instruct the completer to use __all__ for the completion
+        
+        Specifically, when completing on ``object.<tab>``.
+        
+        When 1: only those names in obj.__all__ will be included.
+        
+        When 0 [default]: the values in the __all__ attribute are ignored
+        """
+    )
 
     def __init__(self, shell=None, namespace=None, global_namespace=None,
                  alias_table=None, use_readline=True,

IPython/core/tests/test_completer.py

@@ -18,6 +18,7 b' from IPython.core import completer'
 from IPython.external.decorators import knownfailureif
 from IPython.utils.tempdir import TemporaryDirectory
 from IPython.utils.generics import complete_object
+from IPython.testing.globalipapp import get_ipython
 
 #-----------------------------------------------------------------------------
 # Test functions
@@ -230,3 +231,14 b' def test_omit__names():'
     nt.assert_false('ip._hidden_attr' in matches)
     del ip._hidden_attr
     
+def test_get__all__entries_ok():
+  class A(object):
+    __all__ = ['x', 1]
+  words = completer.get__all__entries(A())
+  nt.assert_equal(words, ['x'])
+
+def test_get__all__entries_no__all__ok():
+  class A(object):
+      pass
+  words = completer.get__all__entries(A())
+  nt.assert_equal(words, [])
  No newline at end of file