starting to move to vim-py-docstring.py

Author: pixelneo

python/docstring.py

@@ -1,22 +1,10 @@
 #!/usr/bin/env python3
-import regex as re
+import re
 from string import Template
 
-def test(a, b="ahoj ): # svete"):
-    pass
 
 # TODO: class
 # TODO: class will have func, 'one_tab' to return indent of one tab more
-# TODO: template different docstrings
-def google(arguments, funcdef_indent, indent):
-    indent1 = ''.join([funcdef_indent, indent])  #  'one tab' more indented than funcdef
-    indent2 = ''.join([indent1, indent])
-
-    argstring = '\n'.join(['{ind2}{name}: '.format(ind2=indent2, name=arg) for arg in arguments])
-    docstring = '{ind}"""\n{ind}\n{ind}\n{ind}\n{ind}Args:\n{args}\n{ind}\n{ind}Returns:\n{ind2}\n{ind}\n{ind}"""'.format(\
-        ind=indent1, ind2=indent2, args=argstring)
-    return docstring
-
 def substitute_list(list_name, template, list_):
     result_lines = []
     for line in template.split('\n'):

python/google.txt

@@ -1,11 +1,10 @@
-$i"""
-$i
-$i
-$i
-${i}Args:
-#args:{${i}${i2}$item: ()\n}
-$i
-${i}Returns:
-$i$i2
-$i
-$i"""
+$i$i2"""
+$i$i2
+$i$i2
+$i$i2
+${i}${i2}Args:
+#args:{${i}${i2}${i2}$item: ()\n}
+${i}${i2}Returns:
+$i$i2$i2
+$i$i2
+$i$i2"""

python/vim-py-docstring.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
-import regex as re
-
+import re
+import ast
+from string import Template
 class MethodDocGenerator:
 
     def __init__(self, signature):