ENH: add BagObj.__dir__ to numpy.lib.npyio

This allows dir(bagobj), and also enables tab-completion on a BagObj, which can be useful in an interpreter or IPython
aman-thakral · Nov 18, 2014 · 60de33d · 60de33d
1 parent c2d7845
commit 60de33d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/numpy/lib/npyio.py b/numpy/lib/npyio.py
@@ -122,6 +122,14 @@ def __getattribute__(self, key):
             return object.__getattribute__(self, '_obj')[key]
         except KeyError:
             raise AttributeError(key)
+
+    def __dir__(self):
+        """
+        Enables dir(bagobj) to list the files in an NpzFile.
+        
+        This also enables tab-completion in an interpreter or IPython.
+        """
+        return object.__getattribute__(self, '_obj').keys()
 
 
 def zipfile_factory(*args, **kwargs):