diff options
| -rw-r--r-- | examples/console/web/index.py | 52 |
1 files changed, 50 insertions, 2 deletions
diff --git a/examples/console/web/index.py b/examples/console/web/index.py index 81f2759..d181857 100644 --- a/examples/console/web/index.py +++ b/examples/console/web/index.py @@ -1,7 +1,34 @@ +from cStringIO import StringIO +import traceback +import pdb +import threading +import sys from console.sitepage import SitePage from console.JSONComponent import JSONComponent global_namespace = {} +namespace_lock = threading.Lock() + +# From doctest +class _OutputRedirectingPdb(pdb.Pdb): + """ + A specialized version of the python debugger that redirects stdout + to a given stream when interacting with the user. Stdout is *not* + redirected when traced code is executed. + """ + def __init__(self, out): + self.__out = out + pdb.Pdb.__init__(self) + + def trace_dispatch(self, *args): + # Redirect stdout to the given stream. + save_stdout = sys.stdout + sys.stdout = self.__out + # Call Pdb's trace dispatch method. + try: + return pdb.Pdb.trace_dispatch(self, *args) + finally: + sys.stdout = save_stdout class index(SitePage): @@ -14,5 +41,26 @@ class index(SitePage): def jsonMethods(self): return ['run'] - def run(self): - return 'Cool dude!' + def run(self, value): + out = StringIO() + namespace_lock.acquire() + save_stdout = sys.stdout + try: + debugger = _OutputRedirectingPdb(save_stdout) + debugger.reset() + pdb.set_trace = debugger.set_trace + sys.stdout = out + try: + code = compile(value, '<web>', "single", 0, 1) + exec code in global_namespace + debugger.set_continue() + except KeyboardInterrupt: + raise + except: + traceback.print_exc(file=out) + debugger.set_continue() + finally: + sys.stdout = save_stdout + namespace_lock.release() + value = out.getvalue() + return value |