X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Fdispatch.py;h=666d7a7cbd3fff980c10b3d53a55d900576f59c3;hb=refs%2Fheads%2Fpython2;hp=5276892ac9e1abbc8250a92f07d78cdc0722e052;hpb=dc590bea957bbffa808e6cf3f516d4ae9beda98a;p=wrw.git diff --git a/wrw/dispatch.py b/wrw/dispatch.py index 5276892..666d7a7 100644 --- a/wrw/dispatch.py +++ b/wrw/dispatch.py @@ -1,5 +1,5 @@ import sys, traceback -from . import env, req, proto +import env, req, proto __all__ = ["restart"] @@ -17,8 +17,9 @@ def mangle(result): return [str(result)] def defaulterror(req, excinfo): - from . import resp + import resp traceback.print_exception(*excinfo) + sys.stderr.flush() raise resp.httperror(500) def wraphandler(handler, excinfo): @@ -37,9 +38,9 @@ def handle(req, startreq, handler): try: resp = handler(req) break - except restart as i: + except restart, i: handler = i.handle - except Exception as i: + except Exception, i: if eh is None: raise handler = wraphandler(eh, sys.exc_info()) @@ -54,3 +55,15 @@ def handleenv(env, startreq, handler): return proto.simpleerror(env, startreq, 400, "Bad Request", "Request must include Host header.") r = req.origrequest(env) return handle(r, startreq, handler) + +def exterror(env, startreq): + def handler(req): + import resp + code = 404 + if "Response-Code" in req.ihead: + try: + code = int(req.ihead["Response-Code"]) + except ValueError: + pass + raise resp.httperror(code) + return handleenv(env, startreq, handler)