More extract_pyi tweaks

evaherrada · evaherrada · commit e169da3532f6 · 2020-07-02T12:02:21.000-04:00
diff --git a/tools/extract_pyi.py b/tools/extract_pyi.py
@@ -59,17 +59,18 @@ def convert_folder(top_level, stub_directory):
         tree = astroid.parse(stub_contents)
         for i in tree.body:
             if 'name' in i.__dict__:
+                print()
                 print(i.__dict__['name'])
                 for j in i.body:
                     if isinstance(j, astroid.scoped_nodes.FunctionDef):
                         for k, l in zip(j.args.__dict__['annotations'], j.args.__dict__['args']):
                             # K is type, l is name
                             if l.name != 'self':
                                 if not k:
-                                    print(f"Missing parameter type: {j.__dict__['name']} on line {j.__dict__['lineno']}\n")
+                                    print(f"Missing parameter type: {j.__dict__['name']} on line {j.__dict__['lineno']}")
                                 elif str(type(k)) == "<class 'astroid.node_classes.Name'>":
                                    if k.name == 'Any':
-                                      print(f"'Any' parameter type: {j.__dict__['name']} on line {j.__dict__['lineno']}\n")
+                                      print(f"'Any' parameter type: {j.__dict__['name']} on line {j.__dict__['lineno']}")
                         if j.returns:
                             if 'Any' in j.returns.__dict__.values():
                                 print(f"Missing return type: {j.__dict__['name']} on line {j.__dict__['lineno']}")
