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().

1 files changed, 9 insertions, 1 deletions

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 @@ def compile(file, cfile=None, dfile=None):
     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')