Merge pull request #20 from pixelneo/self

pixelneo · web-flow · commit 5a9d57038e6b · 2022-06-20T14:29:10.000+02:00
fix
diff --git a/python/asthelper.py b/python/asthelper.py
@@ -88,7 +88,7 @@ def _handle_functions(self, node):
                 if arg.annotation is not None:
                     type_hint = ast.unparse(arg.annotation)
                 self.arguments.append({'arg': arg.arg, 'type': type_hint})
-            if len(self.arguments) > 0 and (self.arguments[0] == 'self' or self.arguments[0] == 'cls'):
+            if len(self.arguments) > 0 and (self.arguments[0]['arg'] == 'self' or self.arguments[0]['arg'] == 'cls'):
                 self.arguments.pop(0)
 
             self.returns = new_visitor.returns
diff --git a/python/pydocstring.py b/python/pydocstring.py
@@ -200,17 +200,6 @@ def write_docstring(self, print_hints=False):
             method_indent, args, returns, yields, raises, print_hints)
         self.env.append_after_line(sig_line, docstring)
 
-    def _arguments(self, tree):
-        try:
-            args = []
-            for arg in tree.body[0].args.args:
-                args.append(arg.arg)
-            if args[0] == 'self' or args[0] == 'cls':
-                args.pop(0)
-            return args
-        except SyntaxError as e:
-            raise InvalidSyntax('The method has invalid syntax.')
-
 
 class ClassController(ObjectWithDocstring):
 
