| 1 | import sys, traceback |
| 2 | import env, req, proto |
| 3 | |
| 4 | __all__ = ["restart"] |
| 5 | |
| 6 | class restart(Exception): |
| 7 | def handle(self, req): |
| 8 | pass |
| 9 | |
| 10 | def mangle(result): |
| 11 | try: |
| 12 | iter(result) |
| 13 | except TypeError: |
| 14 | pass |
| 15 | else: |
| 16 | return result |
| 17 | return [str(result)] |
| 18 | |
| 19 | def defaulterror(req, excinfo): |
| 20 | import resp |
| 21 | traceback.print_exception(*excinfo) |
| 22 | raise resp.httperror(500) |
| 23 | |
| 24 | def wraphandler(handler, excinfo): |
| 25 | def wrapped(req): |
| 26 | return handler(req, excinfo) |
| 27 | return wrapped |
| 28 | |
| 29 | errorhandler = env.var(defaulterror) |
| 30 | |
| 31 | def handle(req, startreq, handler): |
| 32 | eh = errorhandler.val |
| 33 | try: |
| 34 | resp = [""] |
| 35 | while True: |
| 36 | try: |
| 37 | resp = handler(req) |
| 38 | break |
| 39 | except restart, i: |
| 40 | handler = i.handle |
| 41 | except Exception, i: |
| 42 | if eh is None: |
| 43 | raise |
| 44 | handler = wraphandler(eh, sys.exc_info()) |
| 45 | eh = None |
| 46 | req.commit(startreq) |
| 47 | return resp |
| 48 | finally: |
| 49 | req.cleanup() |
| 50 | |
| 51 | def handleenv(env, startreq, handler): |
| 52 | if not "HTTP_HOST" in env: |
| 53 | return proto.simpleerror(env, startreq, 400, "Bad Request", "Request must include Host header.") |
| 54 | r = req.origrequest(env) |
| 55 | return handle(r, startreq, handler) |