Add -p option to invoke Python profiler

jeremyhylton · jeremyhylton · commit bf80a033ee76 · 2001-09-17T18:08:40.000Z
diff --git a/Tools/compiler/compile.py b/Tools/compiler/compile.py
@@ -3,13 +3,14 @@
 
 from compiler import compile, visitor
 
-##import profile
+import profile
 
 def main():
     VERBOSE = 0
     DISPLAY = 0
+    PROFILE = 0
     CONTINUE = 0
-    opts, args = getopt.getopt(sys.argv[1:], 'vqdc')
+    opts, args = getopt.getopt(sys.argv[1:], 'vqdcp')
     for k, v in opts:
         if k == '-v':
             VERBOSE = 1
@@ -24,16 +25,21 @@ def main():
             DISPLAY = 1
         if k == '-c':
             CONTINUE = 1
+        if k == '-p':
+            PROFILE = 1
     if not args:
         print "no files to compile"
     else:
         for filename in args:
             if VERBOSE:
                 print filename
             try:
-                compile(filename, DISPLAY)
-##                profile.run('compile(%s, %s)' % (`filename`, `DISPLAY`),
-##                            filename + ".prof")
+                if PROFILE:
+                    profile.run('compile(%s, %s)' % (`filename`, `DISPLAY`),
+                                filename + ".prof")
+                else:
+                    compile(filename, DISPLAY)
+                    
             except SyntaxError, err:
                 print err
                 print err.lineno
