Fix suggested by Sjoerd (long ago!) to get a better error message when
there's a syntax error. (In particular, display the correct
filename). This changes the API: if there's a syntax error, the
function now returns normally after dumping the error to sys.stderr.
I changed Sjoerd's use of string.join(string.split(...)) with
string.replace().
diff --git a/Lib/py_compile.py b/Lib/py_compile.py
index 98b3b21..a6d03d7 100644
--- a/Lib/py_compile.py
+++ b/Lib/py_compile.py
@@ -51,7 +51,15 @@
f.close()
if codestring and codestring[-1] != '\n':
codestring = codestring + '\n'
- codeobject = __builtin__.compile(codestring, dfile or file, 'exec')
+ try:
+ codeobject = __builtin__.compile(codestring, dfile or file, 'exec')
+ except SyntaxError, detail:
+ import traceback, sys, string
+ lines = traceback.format_exception_only(SyntaxError, detail)
+ for line in lines:
+ sys.stderr.write(string.replace(line, 'File "<string>"',
+ 'File "%s"' % (dfile or file)))
+ return
if not cfile:
cfile = file + (__debug__ and 'c' or 'o')
fc = open(cfile, 'wb')