Merge pull request #29 from pixelneo/attr_order

do not reorder attributes

Author: pixelneo

python/asthelper.py

@@ -14,13 +14,13 @@ def visit_Call(self, node):
 class AttributeCollector(ast.NodeVisitor):
     def __init__(self, instance_name):
         self.instance_name = instance_name
-        self.data = set()
+        self.data = {}
         super().__init__()
 
     def visit_Attribute(self, node):
         if isinstance(node.value, ast.Name):
             if node.value.id == self.instance_name:
-                self.data.add(node.attr)
+                self.data[node.attr] = None
         else:
             self.generic_visit(node)
 
@@ -46,7 +46,7 @@ def generic_visit(self, node):
 class ClassVisitor(ast.NodeVisitor):
     def __init__(self, instance_name):
         super().__init__()
-        self.attributes = set()
+        self.attributes = {}
         self.instance_name = instance_name
 
     def visit_Assign(self, node):

python/pydocstring.py

@@ -232,7 +232,7 @@ def _process_tree(self, tree):
         x.visit(tree)
         v = ClassVisitor(x.instance_name)
         v.visit(tree)
-        att = list(v.attributes)
+        att = [attr_name for attr_name in v.attributes]
         return att
 
     def write_docstring(self, *args, **kwargs):