wrw/resp.py

   1 from . 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().__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, message, *detail):
  40         super().__init__()
  41         self.message = message
  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().__init__(message, *detail)
  54         self.status = status
  55
  56     def handle(self, req):
  57         req.status(self.status, self.message)
  58         return super().handle(req)
  59
  60 class notfound(httperror):
  61     def __init__(self):
  62         return super().__init__(404)
  63
  64 class redirect(dispatch.restart):
  65     bases = {"url": proto.requrl,
  66              "script": proto.scripturl,
  67              "site": proto.siteurl}
  68
  69     def __init__(self, url, status=303, base="url"):
  70         super().__init__()
  71         self.url = url
  72         self.status = status
  73         self.bases[base]
  74         self.base = base
  75
  76     def handle(self, req):
  77         req.status(self.status, "Redirect")
  78         req.ohead["Location"] = proto.appendurl(self.bases[self.base](req), self.url)
  79         req.ohead["Content-Length"] = 0
  80         return []
  81
  82 class unmodified(dispatch.restart):
  83     def handle(self, req):
  84         req.status(304, "Not Modified")
  85         req.ohead["Content-Length"] = "0"
  86         return []