Fix lint

klen · klen · commit 147846c73ca3 · 2013-12-04T15:32:48.000+07:00
diff --git a/pymode/lint.py b/pymode/lint.py
@@ -16,6 +16,10 @@ def code_check():
     from pylama.main import parse_options
     from pylama.tasks import check_path
 
+    if not env.curbuf.name:
+        env.stop()
+        return False
+
     options = parse_options(
         ignore=env.var('g:pymode_lint_ignore'),
         select=env.var('g:pymode_lint_select'),
@@ -36,7 +40,7 @@ def code_check():
         LOGGER.setLevel(logging.DEBUG)
 
     with silence_stderr():
-        errors = check_path(path, options=options, code=env.source)
+        errors = check_path(path, options=options, code='\n'.join(env.curbuf))
 
     env.debug("Find errors: ", len(errors))
     sort_rules = env.var('g:pymode_lint_sort')
diff --git a/t/lint.vim b/t/lint.vim
@@ -14,5 +14,17 @@ describe 'pymode check code'
         Expect g:pymode_lint == 1
     end
 
+    it 'lint code'
+        put =['# coding: utf-8', 'call_unknown_function()']
+        PymodeLint
+        Expect getloclist(0) ==  []
+    end
+
+    it 'lint code'
+        e t/test.py
+        PymodeLint
+        Expect getloclist(0) ==  [{'lnum': 6, 'bufnr': 1, 'col': 0, 'valid': 1, 'vcol': 0, 'nr': 0, 'type': 'E', 'pattern': '', 'text': 'W0612 local variable "unused" is assigned to but never used [pyflakes]'}, {'lnum': 8, 'bufnr': 1, 'col': 0, 'valid': 1, 'vcol': 0, 'nr': 0, 'type': 'E', 'pattern': '', 'text': 'E0602 undefined name "unknown" [pyflakes]'}]
+    end
+
 end
 
diff --git a/t/test.py b/t/test.py
@@ -0,0 +1,8 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+
+def main():
+    unused = 1
+
+unknown()
