Merge pull request #520 from chuan92/develop

klen · klen · commit 8a3b5e14ad69 · 2015-01-16T00:30:21.000+03:00
Fix #482, (PEP 263 Python Source Code Encodings)
diff --git a/pymode/run.py b/pymode/run.py
@@ -6,8 +6,7 @@
 from .environment import env
 
 
-encoding = re(r'#[^\w]+coding:\s+utf.*$')
-
+encoding = re(r'#.*coding[:=]\s*([-\w.]+)')
 
 def run_code():
     """ Run python code in current buffer.
@@ -18,9 +17,12 @@ def run_code():
     errors, err = [], ''
     line1, line2 = env.var('a:line1'), env.var('a:line2')
     lines = __prepare_lines(line1, line2)
-    for ix in (0, 1):
-        if encoding.match(lines[ix]):
-            lines.pop(ix)
+    if encoding.match(lines[0]):
+        lines.pop(0)
+        if encoding.match(lines[0]):
+            lines.pop(0)
+    elif encoding.match(lines[1]):
+        lines.pop(1)
 
     context = dict(
         __name__='__main__',
