Fix default parameter problem (#49)

Author: heavenshell

CHANGES.rst

@@ -1,3 +1,13 @@
+Version 0.4.0
+-------------
+Released on May 6th 2018
+
+- Bug fix
+
+  - Default parameter problem
+
+  see https://github.com/heavenshell/vim-pydocstring/issues/46
+
 Version 0.3.0
 -------------
 Released on Dec 10th 2017

autoload/pydocstring.vim

@@ -153,8 +153,10 @@ function! s:parse_func(type, line)
   let header = substitute(a:line, '\s\|(.*\|:', '', 'g')
 
   let args_str = substitute(a:line, '\s\|.*(\|).*', '', 'g')
-  if args_str =~ ':'
+  if args_str =~ ':' && args_str =~ '['
     let args = s:parse_args(args_str)
+  elseif args_str =~ ':'
+    let args = split(args_str, ',')
   else
     " No typed args.
     let args = split(args_str, ',')

doc/pydocstring.txt

@@ -1,6 +1,6 @@
 *pydocstring.txt*	Generate Python docstring to your Python code.
 
-Version: 0.3.0
+Version: 0.4.0
 Author: Shinya Ohynagi <[email protected]>
 Repository: http://github.com/heavenshell/vim-pydocstring/
 License: BSD, see LICENSE for more details.

test/type-hint.vader

@@ -247,3 +247,39 @@ Expect python:
         :type arg2: Callable[List[str, str], str, int]
         """
         pass
+
+Given python (def first has type, the rest is not):
+    def foo(a: int, b, c) -> int:
+        pass
+
+Execute:
+    Pydocstring
+
+Expect python:
+    def foo(a: int, b, c) -> int:
+        """foo
+
+        :param a:
+        :type a: int
+        :param b:
+        :param c:
+        :rtype: int
+        """
+        pass
+
+Given python (keyword args):
+    def foo(a, b: str = None):
+        pass
+
+Execute:
+    Pydocstring
+
+Expect python:
+    def foo(a, b: str = None):
+        """foo
+
+        :param a:
+        :param b:
+        :type b: str
+        """
+        pass