[llvm-commits] [LNT] r154934 - /lnt/trunk/lnt/lnttool/main.py

[Daniel Dunbar]

Author: ddunbar
Date: Tue Apr 17 12:50:16 2012
New Revision: 154934

URL: http://llvm.org/viewvc/llvm-project?rev=154934&view=rev
Log:
lnt runserver: Add a --profiler option.
 - Writes to a profiler.log in the local directory.

Modified:
    lnt/trunk/lnt/lnttool/main.py

Modified: lnt/trunk/lnt/lnttool/main.py
URL: http://llvm.org/viewvc/llvm-project/lnt/trunk/lnt/lnttool/main.py?rev=154934&r1=154933&r2=154934&view=diff
==============================================================================
--- lnt/trunk/lnt/lnttool/main.py (original)
+++ lnt/trunk/lnt/lnttool/main.py Tue Apr 17 12:50:16 2012
@@ -6,6 +6,8 @@
 import tempfile
 from optparse import OptionParser, OptionGroup
 
+import werkzeug.contrib.profiler
+
 import StringIO
 import lnt
 import lnt.util.ImportData
@@ -38,6 +40,8 @@
                       action="store_true", help="use WSGI reload monitor")
     parser.add_option("", "--debugger", dest="debugger", default=False,
                       action="store_true", help="use WSGI debugger")
+    parser.add_option("", "--profiler", dest="profiler", default=False,
+                      action="store_true", help="enable WSGI profiler")
     parser.add_option("", "--show-sql", dest="show_sql", default=False,
                       action="store_true", help="show all SQL queries")
     parser.add_option("", "--threaded", dest="threaded", default=False,
@@ -74,6 +78,9 @@
     app = lnt.server.ui.app.App.create_standalone(input_path,)
     if opts.debugger:
         app.debug = True
+    if opts.profiler:
+        app.wsgi_app = werkzeug.contrib.profiler.ProfilerMiddleware(
+            app.wsgi_app, stream = open('profiler.log', 'w'))
     app.run(opts.hostname, opts.port,
             use_reloader = opts.reloader,
             use_debugger = opts.debugger,