X-Git-Url: http://dolda2000.com/gitweb/?p=wrw.git;a=blobdiff_plain;f=wrw%2Fdispatch.py;h=2c97b9fbb30e0291a4fc05160fb5720ea5790b5b;hp=d428a77ce5f503ae871b3e8eb8d2f2874d22544c;hb=HEAD;hpb=b409a33843abb3221edd27016558c39cf33a6510

diff --git a/wrw/dispatch.py b/wrw/dispatch.py
index d428a77..2c97b9f 100644
--- a/wrw/dispatch.py
+++ b/wrw/dispatch.py
@@ -1,3 +1,6 @@
+import sys, traceback
+from . import env, req, proto
+
 __all__ = ["restart"]
 
 class restart(Exception):
@@ -13,16 +16,54 @@ def mangle(result):
         return result
     return [str(result)]
 
+def defaulterror(req, excinfo):
+    from . import resp
+    traceback.print_exception(*excinfo)
+    sys.stderr.flush()
+    raise resp.httperror(500)
+
+def wraphandler(handler, excinfo):
+    def wrapped(req):
+        return handler(req, excinfo)
+    wrapped.__wrapped__ = handler
+    return wrapped
+
+errorhandler = env.var(defaulterror)
+
 def handle(req, startreq, handler):
+    eh = errorhandler.val
     try:
         resp = [""]
         while True:
             try:
                 resp = handler(req)
                 break
-            except restart, i:
-                handler = i
+            except restart as i:
+                handler = i.handle
+            except Exception as i:
+                if eh is None:
+                    raise
+                handler = wraphandler(eh, sys.exc_info())
+                eh = None
         req.commit(startreq)
         return resp
     finally:
         req.cleanup()
+
+def handleenv(env, startreq, handler):
+    if not "HTTP_HOST" in env:
+        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):
+        from . 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)