Mercurial > hg > MakeItSo
view makeitso/handlers.py @ 36:0cba953a03ca
fixed single-file test case
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Sat, 01 Jan 2011 22:13:33 -0800 |
parents | 6d1c703c5ffc |
children |
line wrap: on
line source
""" request handlers: these are instantiated for every request, then called """ import os from pkg_resources import resource_filename from urlparse import urlparse from tempita import HTMLTemplate from webob import Response, exc class HandlerMatchException(Exception): """the handler doesn't match the request""" class Handler(object): methods = set(['GET']) # methods to listen to handler_path = [] # path elements to match @classmethod def match(cls, app, request): # check the method if request.method not in cls.methods: return None # check the path if request.environ['path'][:len(cls.handler_path)] != cls.handler_path: return None try: return cls(app, request) except HandlerMatchException: return None def __init__(self, app, request): self.app = app self.request = request self.application_path = urlparse(request.application_url)[2] def link(self, path=(), permanant=False): if isinstance(path, basestring): path = [ path ] path = [ i.strip('/') for i in path ] if permanant: application_url = [ self.request.application_url ] else: application_url = [ self.application_path ] path = application_url + path return '/'.join(path) def redirect(self, location): raise exc.HTTPSeeOther(location=location) class TempitaHandler(Handler): template_dirs = [ resource_filename(__name__, 'templates') ] def __init__(self, app, request): Handler.__init__(self, app, request) self.data = { 'request': request, 'link': self.link } def __call__(self): return getattr(self, self.request.method.title())() def find_template(self, template): for d in self.template_dirs: path = os.path.join(d, template) if os.path.exists(path): return HTMLTemplate.from_filename(path) def Get(self): # needs to have self.template set template = self.find_template(self.template) return Response(content_type='text/html', body=template.substitute(**self.data)) class Index(TempitaHandler): template = 'index.html' methods=set(['GET'])