wrw/resp.py

   1 import dispatch, proto, env
   2 from sp import xhtml
   3 h = xhtml.cons()
   4
   5 __all__ = ["skeleton", "skelfor", "setskel", "usererror"]
   6
   7 class skeleton(object):
   8     def page(self, req, title, *content):
   9         return xhtml.forreq(req, h.html(self.head(req, title), h.body(*content)))
  10
  11     def head(self, req, title):
  12         return xhtml.head(title=title)
  13
  14     def error(self, req, message, *detail):
  15         return self.page(req, message, h.h1(message), h.p(*detail))
  16
  17     def message(self, req, message, *detail):
  18         return self.page(req, message, h.h1(message), h.p(*detail))
  19
  20 defskel = env.var(skeleton())
  21
  22 def getskel(req):
  23     return [defskel.val]
  24 def skelfor(req):
  25     return req.item(getskel)[0]
  26 def setskel(req, skel):
  27     req.item(getskel)[0] = skel
  28
  29 class usererror(dispatch.restart):
  30     def __init__(self, message, *detail):
  31         super(usererror, self).__init__()
  32         self.message = message
  33         self.detail = detail
  34
  35     def handle(self, req):
  36         return skelfor(req).error(req, self.message, *self.detail)
  37
  38 class message(dispatch.restart):
  39     def __init__(self, msg, *detail):
  40         super(message, self).__init__()
  41         self.message = msg
  42         self.detail = detail
  43
  44     def handle(self, req):
  45         return skelfor(req).message(req, self.message, *self.detail)
  46
  47 class httperror(usererror):
  48     def __init__(self, status, message = None, detail = None):
  49         if message is None:
  50             message = proto.statusinfo[status][0]
  51         if detail is None:
  52             detail = (proto.statusinfo[status][1],)
  53         super(httperror, self).__init__(message, *detail)
  54         self.status = status
  55
  56     def handle(self, req):
  57         req.status(self.status, self.message)
  58         return super(httperror, self).handle(req)
  59
  60 class notfound(httperror):
  61     def __init__(self):
  62         return super(notfound, self).__init__(404)
  63
  64 class redirect(dispatch.restart):
  65     def __init__(self, url, status = 303):
  66         super(redirect, self).__init__()
  67         self.url = url
  68         self.status = status
  69
  70     def handle(self, req):
  71         req.status(self.status, "Redirect")
  72         req.ohead["Location"] = proto.appendurl(proto.requrl(req), self.url)
  73         return []
  74
  75 class unmodified(dispatch.restart):
  76     def handle(self, req):
  77         req.status(304, "Not Modified")
  78         req.ohead["Content-Length"] = "0"
  79         return []