|
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- # Copyright (c) Twisted Matrix Laboratories.
- # See LICENSE for details.
-
- #
- from twisted.web import resource
-
-
- class RewriterResource(resource.Resource):
- def __init__(self, orig, *rewriteRules):
- resource.Resource.__init__(self)
- self.resource = orig
- self.rewriteRules = list(rewriteRules)
-
- def _rewrite(self, request):
- for rewriteRule in self.rewriteRules:
- rewriteRule(request)
-
- def getChild(self, path, request):
- request.postpath.insert(0, path)
- request.prepath.pop()
- self._rewrite(request)
- path = request.postpath.pop(0)
- request.prepath.append(path)
- return self.resource.getChildWithDefault(path, request)
-
- def render(self, request):
- self._rewrite(request)
- return self.resource.render(request)
-
-
- def tildeToUsers(request):
- if request.postpath and request.postpath[0][:1] == "~":
- request.postpath[:1] = ["users", request.postpath[0][1:]]
- request.path = "/" + "/".join(request.prepath + request.postpath)
-
-
- def alias(aliasPath, sourcePath):
- """
- I am not a very good aliaser. But I'm the best I can be. If I'm
- aliasing to a Resource that generates links, and it uses any parts
- of request.prepath to do so, the links will not be relative to the
- aliased path, but rather to the aliased-to path. That I can't
- alias static.File directory listings that nicely. However, I can
- still be useful, as many resources will play nice.
- """
- sourcePath = sourcePath.split("/")
- aliasPath = aliasPath.split("/")
-
- def rewriter(request):
- if request.postpath[: len(aliasPath)] == aliasPath:
- after = request.postpath[len(aliasPath) :]
- request.postpath = sourcePath + after
- request.path = "/" + "/".join(request.prepath + request.postpath)
-
- return rewriter
|