Merge branch 'master' into python3
[wrw.git] / wrw / resp.py
... / ...
CommitLineData
1from . import dispatch, proto, env
2from .sp import xhtml
3h = xhtml.cons()
4
5__all__ = ["skeleton", "skelfor", "setskel", "usererror"]
6
7class skeleton(object):
8 def page(self, title, *content):
9 return h.html(self.head(title), h.body(*content))
10
11 def head(self, title):
12 return xhtml.head(title=title)
13
14 def error(self, message, *detail):
15 return self.page(message, h.h1(message), h.p(*detail))
16
17 def message(self, message, *detail):
18 return self.page(message, h.h1(message), h.p(*detail))
19
20defskel = env.var(skeleton())
21
22def getskel(req):
23 return [defskel.val]
24def skelfor(req):
25 return req.item(getskel)[0]
26def setskel(req, skel):
27 req.item(getskel)[0] = skel
28
29class 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 xhtml.forreq(req, skelfor(req).error(self.message, *self.detail))
37
38class 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 xhtml.forreq(req, skelfor(req).error(self.message, *self.detail))
46
47class 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
60class notfound(httperror):
61 def __init__(self):
62 return super().__init__(404)
63
64class redirect(dispatch.restart):
65 def __init__(self, url, status = 303):
66 super().__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 []