Support PEP-263-style coding declarations.
Default to UTF-8 per PEP-3120.
diff --git a/Lib/linecache.py b/Lib/linecache.py
index 0501a10..9a16acd 100644
--- a/Lib/linecache.py
+++ b/Lib/linecache.py
@@ -7,6 +7,7 @@
import sys
import os
+import re
__all__ = ["getline", "clearcache", "checkcache"]
@@ -131,6 +132,16 @@
except IOError as msg:
## print '*** Cannot open', fullname, ':', msg
return []
+ coding = "utf-8"
+ for line in lines[:2]:
+ m = re.search(r"coding[:=]\s*([-\w.]+)", line)
+ if m:
+ coding = m.group(1)
+ break
+ try:
+ lines = [unicode(line, coding) for line in lines]
+ except UnicodeError:
+ pass # Hope for the best
size, mtime = stat.st_size, stat.st_mtime
cache[filename] = size, mtime, lines, fullname
return lines