From: Fredrik Tolf <fredrik@dolda2000.com>
Date: Mon, 2 Sep 2013 01:19:03 +0000 (+0200)
Subject: Made redirections a bit more flexible.
X-Git-Url: http://git.dolda2000.com/gitweb/?a=commitdiff_plain;h=d0034deebc5dc1fc5a54d1e99fe917f9ffbe7d69;p=wrw.git

Made redirections a bit more flexible.
---

diff --git a/wrw/resp.py b/wrw/resp.py
index 50ac6b1..fbb056f 100644
--- a/wrw/resp.py
+++ b/wrw/resp.py
@@ -62,14 +62,20 @@ class notfound(httperror):
         return super(notfound, self).__init__(404)
 
 class redirect(dispatch.restart):
-    def __init__(self, url, status = 303):
+    bases = {"url": proto.requrl,
+             "script": proto.scripturl,
+             "site": proto.siteurl}
+
+    def __init__(self, url, status = 303, base = "url"):
         super(redirect, self).__init__()
         self.url = url
         self.status = status
+        self.bases[base]
+        self.base = base
 
     def handle(self, req):
         req.status(self.status, "Redirect")
-        req.ohead["Location"] = proto.appendurl(proto.requrl(req), self.url)
+        req.ohead["Location"] = proto.appendurl(self.bases[self.base](req), self.url)
         req.ohead["Content-Length"] = 0
         return []