You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.3 KiB
42 lines
1.3 KiB
6 years ago
|
from waitress.server import create_server
|
||
|
|
import logging
|
||
|
|
|
||
|
|
def serve(app, **kw):
|
||
|
|
_server = kw.pop('_server', create_server) # test shim
|
||
|
|
_quiet = kw.pop('_quiet', False) # test shim
|
||
|
|
_profile = kw.pop('_profile', False) # test shim
|
||
|
|
if not _quiet: # pragma: no cover
|
||
|
|
# idempotent if logging has already been set up
|
||
|
|
logging.basicConfig()
|
||
|
|
server = _server(app, **kw)
|
||
|
|
if not _quiet: # pragma: no cover
|
||
|
|
server.print_listen('Serving on http://{}:{}')
|
||
|
|
if _profile: # pragma: no cover
|
||
|
|
profile('server.run()', globals(), locals(), (), False)
|
||
|
|
else:
|
||
|
|
server.run()
|
||
|
|
|
||
|
|
def serve_paste(app, global_conf, **kw):
|
||
|
|
serve(app, **kw)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
def profile(cmd, globals, locals, sort_order, callers): # pragma: no cover
|
||
|
|
# runs a command under the profiler and print profiling output at shutdown
|
||
|
|
import os
|
||
|
|
import profile
|
||
|
|
import pstats
|
||
|
|
import tempfile
|
||
|
|
fd, fn = tempfile.mkstemp()
|
||
|
|
try:
|
||
|
|
profile.runctx(cmd, globals, locals, fn)
|
||
|
|
stats = pstats.Stats(fn)
|
||
|
|
stats.strip_dirs()
|
||
|
|
# calls,time,cumulative and cumulative,calls,time are useful
|
||
|
|
stats.sort_stats(*(sort_order or ('cumulative', 'calls', 'time')))
|
||
|
|
if callers:
|
||
|
|
stats.print_callers(.3)
|
||
|
|
else:
|
||
|
|
stats.print_stats(.3)
|
||
|
|
finally:
|
||
|
|
os.remove(fn)
|