minor changes to general parser- still undone

pixelneo · pixelneo · commit 0689d7270c17 · 2019-09-04T16:17:58.000+02:00
diff --git a/python/pydocstring.py b/python/pydocstring.py
@@ -75,10 +75,11 @@ def _whole_string(self):
         lines_it = self.env.lines_following_cursor()
         counter = 0
         while not valid:
-            if counter == self.max_lines:
-                raise InvalidSyntax(
-                    'The method either invalid or it is on > {} lines.'.format(str(self.max_lines)))
-            last_row, line = next(lines_it)
+            try:
+                last_row, line = next(lines_it)
+            except StopIteration as e:
+                pass # TODO do something (is_valid?)
+
             lines.append(line)
             data = ''.join(lines)
             valid, tree = self._is_valid(data)
@@ -87,6 +88,7 @@ def _whole_string(self):
         arguments = self._arguments(tree)
         func_indent = re.findall('^(\s*)', lines[0])[0]
 
+    # TODO: change, adding pass will go away
     def _is_valid(self, lines):
         func = ''.join([lines.lstrip(), '\n   pass'])
         try:
diff --git a/python/vimenv.py b/python/vimenv.py
@@ -96,8 +96,10 @@ def lines_following_cursor(self):
         import vim
         lines = []
         buffer = vim.current.buffer
-        cursor_row = vim.current.window.cursor[0]-1
+        cursor_row = vim.current.window.cursor[0] - 1
         current_row = cursor_row
         while True:
+            if current_row >= len(vim.current.buffer) - 1:
+                raise StopIteration('Buffer is out of lines.')
             yield current_row, buffer[current_row]
             current_row += 1
