import sys, traceback
-from . import env
+from . import env, req, proto
__all__ = ["restart"]
return result
return [str(result)]
-class iterproxy(object):
- # Makes sure iter(real).next() is called immediately, in order to
- # let generator code run.
- def __init__(self, real):
- self.bk = real
- self.bki = iter(real)
- self._next = [None]
- self.__next__()
-
- def __iter__(self):
- return self
-
- def __next__(self):
- if self._next is None:
- raise StopIteration()
- ret = self._next[0]
- try:
- self._next[:] = [self.bki.__next__()]
- except StopIteration:
- self._next = None
- return ret
-
- def close(self):
- if hasattr(self.bk, "close"):
- self.bk.close()
-
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)
resp = [""]
while True:
try:
- resp = iterproxy(handler(req))
+ resp = handler(req)
break
except restart as i:
handler = i.handle
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):
+ 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)