Fix {{_return_type_}} did not indented (#55)

Author: heavenshell

autoload/pydocstring.vim

@@ -308,18 +308,20 @@ function! s:build_docstring(strs, indent, nested_indent)
           call add(docstrings, a:indent . template)
         endfor
       endif
-    elseif line =~ '{{_indent_}}'
-      let arg = substitute(line, '{{_indent_}}', a:indent, 'g')
-      call add(docstrings, arg)
-    elseif line =~ '{{_returnType_}}\|{{_return_type_}}'
+    elseif match(line, '{{_returnType_}}\|{{_return_type_}}') != -1
       if strlen(return_type) != 0
+        let line = substitute(line, '{{_indent_}}', a:indent, 'g')
+        let line = substitute(line, '{{_nested_indent_}}', a:nested_indent, 'g')
         let rt = substitute(line, '{{_returnType_}}\|{{_return_type_}}', return_type, '')
         call add(docstrings, a:indent . rt)
       else
         if docstrings[-1] == ''
           call remove(docstrings, -1)
         endif
       endif
+    elseif line =~ '{{_indent_}}'
+      let arg = substitute(line, '{{_indent_}}', a:indent, 'g')
+      call add(docstrings, arg)
     elseif line == '"""' || line == "'''"
       call add(docstrings, a:indent . line)
     else

test/issue_return_type_indent.vader

@@ -0,0 +1,67 @@
+# vim:set et sw=4 ts=4 tw=79:
+Execute (Setup template dir):
+    Save g:pydocstring_templates_dir
+    let g:pydocstring_templates_dir = './templates/issue51/'
+
+Given python (def foo with 1 params):
+    def foo(arg1):
+        pass
+
+Execute:
+    Pydocstring
+
+Expect python:
+    def foo(arg1):
+        """foo
+
+        Args:
+            arg1:
+
+        Returns:
+        """
+        pass
+
+
+Given python (def foo with 2 params):
+    def foo(arg1, arg2):
+        pass
+
+Execute:
+    Pydocstring
+
+Expect python:
+    def foo(arg1, arg2):
+        """foo
+
+        Args:
+            arg1:
+            arg2:
+
+        Returns:
+        """
+        pass
+
+
+Given python (return type):
+    def foo(arg1: str, arg2: int) -> bool:
+        pass
+
+Execute:
+    Pydocstring
+
+Expect python:
+    def foo(arg1: str, arg2: int) -> bool:
+        """foo
+
+        Args:
+            arg1 (str):
+            arg2 (int):
+
+        Returns:
+            bool
+        """
+        pass
+
+
+Execute (Clear pydocstring_templates_dir):
+    Restore

test/templates/issue51/arg.txt

@@ -0,0 +1 @@
+{{_name_}} ({{_type_}}):

test/templates/issue51/comment.txt

@@ -0,0 +1 @@
+# 

test/templates/issue51/multi.txt

@@ -0,0 +1,8 @@
+"""{{_header_}}
+
+{{_indent_}}Args:
+{{_nested_indent_}}{{_args_}}:
+
+{{_indent_}}Returns:
+{{_nested_indent_}}{{_return_type_}}
+"""

test/templates/issue51/oneline.txt

@@ -0,0 +1 @@
+"""{{_header_}}"""