Merge pull request #42 from rtzoeller/dir_support

Calling dir() on a Structure includes Fields

Author: posborne

suitcase/structure.py

@@ -293,6 +293,9 @@ def __setattr__(self, key, value):
             return field.setval(value)
         return object.__setattr__(self, key, value)
 
+    def __dir__(self):
+        return dir(type(self)) + [str(k) for k, v in self._sorted_fields]
+
     def __iter__(self):
         return iter(self._sorted_fields)
 

suitcase/test/test_fields.py

@@ -1189,5 +1189,14 @@ def test_unpack_greedy_after(self):
         self.assertEqual(m.last.value, b"Doe")
         self.assertEqual(m.greedy, b"Hello World!")
 
+
+class TestDiscoverableFields(unittest.TestCase):
+    def test_dir(self):
+        m = PascalString16()
+        attrs = dir(m)
+        self.assertTrue("length" in attrs)
+        self.assertTrue("value" in attrs)
+
+
 if __name__ == "__main__":
     unittest.main()