support for completions on return statements.

2014-06-22 12:05:22 +02:00
parent d752907290
commit b7bf8d515c
4 changed files with 9 additions and 0 deletions
@@ -274,6 +274,8 @@ class Script(object):
            stmt = r.module.statements[-1]
        except IndexError:
            raise NotFoundError()
+        if isinstance(stmt, pr.KeywordStatement):
+            stmt = stmt.stmt
        if not isinstance(stmt, pr.Statement):
            raise NotFoundError()