Made appropriate Python3 changes.
[wrw.git] / wrw / proto.py
CommitLineData
3ba5e96c
FT
1import time
2
5ef9e488 3statusinfo = {
001ec99e
FT
4 400: ("Bad Request", "Invalid HTTP request."),
5 401: ("Unauthorized", "Authentication must be provided for the requested resource."),
6 403: ("Forbidden", "You are not authorized to request the requested resource."),
5ef9e488 7 404: ("Not Found", "The requested resource was not found."),
001ec99e
FT
8 405: ("Method Not Allowed", "The request method is not recognized or permitted by the requested resource."),
9 500: ("Server Error", "An internal error occurred."),
10 501: ("Not Implemented", "The requested functionality has not been implemented."),
11 503: ("Service Unavailable", "Service is being denied at this time."),
5ef9e488
FT
12 }
13
14def httpdate(ts):
15 return time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime(ts))
16
17def phttpdate(dstr):
18 tz = dstr[-6:]
19 dstr = dstr[:-6]
20 if tz[0] != " " or (tz[1] != "+" and tz[1] != "-") or not tz[2:].isdigit():
21 return None
22 tz = int(tz[1:])
23 tz = (((tz / 100) * 60) + (tz % 100)) * 60
24 return time.mktime(time.strptime(dstr, "%a, %d %b %Y %H:%M:%S")) - tz - time.altzone
25
eed3cf12
FT
26def pmimehead(hstr):
27 def pws(p):
28 while p < len(hstr) and hstr[p].isspace():
29 p += 1
30 return p
31 def token(p, sep):
32 buf = ""
33 p = pws(p)
34 if p >= len(hstr):
35 return "", p
36 if hstr[p] == '"':
37 p += 1
38 while p < len(hstr):
39 if hstr[p] == '\\':
40 p += 1
41 if p < len(hstr):
42 buf += hstr[p]
43 p += 1
44 else:
45 break
46 elif hstr[p] == '"':
47 p += 1
48 break
49 else:
50 buf += hstr[p]
51 p += 1
52 return buf, pws(p)
53 else:
54 while p < len(hstr):
55 if hstr[p] in sep:
56 break
57 buf += hstr[p]
58 p += 1
59 return buf.strip(), pws(p)
60 p = 0
61 val, p = token(p, ";")
62 pars = {}
63 while p < len(hstr):
64 if hstr[p] != ';':
65 break
66 p += 1
67 k, p = token(p, "=")
68 if k == "" or hstr[p:p + 1] != '=':
69 break
70 p += 1
71 v, p = token(p, ';')
ea18777a 72 pars[k.lower()] = v
eed3cf12
FT
73 return val, pars
74
5ef9e488
FT
75def htmlq(html):
76 ret = ""
77 for c in html:
78 if c == "&":
79 ret += "&amp;"
80 elif c == "<":
81 ret += "&lt;"
82 elif c == ">":
83 ret += "&gt;"
84 else:
85 ret += c
86 return ret
87
d740aa68
FT
88def simpleerror(env, startreq, code, title, msg):
89 buf = """<?xml version="1.0" encoding="US-ASCII"?>
90<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">
91<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en-US">
92<head>
93<title>%s</title>
94</head>
95<body>
96<h1>%s</h1>
97<p>%s</p>
98</body>
99</html>
100""" % (title, title, htmlq(msg))
cb8bb5ed 101 buf = buf.encode("us-ascii")
d740aa68
FT
102 startreq("%i %s" % (code, title), [("Content-Type", "text/html"), ("Content-Length", str(len(buf)))])
103 return [buf]
104
5ef9e488
FT
105def urlq(url):
106 ret = ""
107 for c in url:
108 if c == "&" or c == "=" or c == "#" or c == "?" or c == "/" or (ord(c) <= 32):
109 ret += "%%%02X" % ord(c)
110 else:
111 ret += c
112 return ret
113
bb80acbd
FT
114class urlerror(ValueError):
115 pass
116
117def parseurl(url):
118 p = url.find("://")
119 if p < 0:
120 raise urlerror("Protocol not found in absolute URL `%s'" % url)
121 proto = url[:p]
122 l = url.find("/", p + 3)
123 if l < 0:
124 raise urlerror("Local part not found in absolute URL `%s'" % url)
125 host = url[p + 3:l]
126 local = url[l:]
127 q = local.find("?")
128 if q < 0:
129 query = ""
130 else:
131 query = local[q + 1:]
132 local = local[:q]
133 return proto, host, local, query
134
e21c4382 135def consurl(proto, host, local, query=""):
bb80acbd
FT
136 if len(local) < 1 and local[0] != '/':
137 raise urlerror("Local part of URL must begin with a slash")
138 ret = "%s://%s%s" % (proto, host, local)
139 if len(query) > 0:
140 ret += "?" + query
141 return ret
142
143def appendurl(url, other):
144 if "://" in other:
145 return other
146 proto, host, local, query = parseurl(url)
147 if len(other) > 0 and other[0] == '/':
148 return consurl(proto, host, other)
149 else:
150 p = local.rfind('/')
151 return consurl(proto, host, local[:p + 1] + other)
152
31d48e69 153def siteurl(req):
bb80acbd
FT
154 host = req.ihead.get("Host", None)
155 if host is None:
156 raise Exception("Could not reconstruct URL because no Host header was sent")
157 proto = "http"
158 if req.https:
159 proto = "https"
31d48e69
FT
160 return "%s://%s/" % (proto, host)
161
162def scripturl(req):
163 s = siteurl(req)
164 if req.uriname[0] != '/':
165 raise Exception("Malformed local part when reconstructing URL")
166 return siteurl(req) + req.uriname[1:]
167
168def requrl(req):
169 s = siteurl(req)
bb80acbd
FT
170 if req.uri[0] != '/':
171 raise Exception("Malformed local part when reconstructing URL")
31d48e69 172 return siteurl(req) + req.uri[1:]
bb80acbd 173
e21c4382 174def parstring(pars={}, **augment):
5ef9e488
FT
175 buf = ""
176 for key in pars:
177 if key in augment:
178 val = augment[key]
179 del augment[key]
180 else:
181 val = pars[key]
182 if buf != "": buf += "&"
183 buf += urlq(key) + "=" + urlq(str(val))
184 for key in augment:
185 if buf != "": buf += "&"
186 buf += urlq(key) + "=" + urlq(str(augment[key]))
187 return buf
08a4b741 188
e21c4382 189def parurl(url, pars={}, **augment):
08a4b741
FT
190 qs = parstring(pars, **augment)
191 if qs != "":
192 return url + "?" + qs
193 else:
194 return url