Mercurial > hg > TagInTheMiddle
view taginthemiddle/handlers.py @ 5:bf4c763b0313
correct import
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Wed, 05 May 2010 22:10:30 -0700 |
parents | 1182315b18ac |
children | fc55d95be553 |
line wrap: on
line source
""" request handlers: these are instantiated for every request, then called """ from urlparse import urlparse 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 GenshiHandler(Handler): 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 Get(self): # needs to have self.template set template = self.app.loader.load(self.template) return Response(content_type='text/html', body=template.generate(**self.data).render('html')) class TagCloud(GenshiHandler): template = 'cloud.html' methods = set(['GET']) @classmethod def match(cls, app, request): # check the method if request.method not in cls.methods: return None # check the path if request.path_info.endswith('/%s' % app.tags_url): try: return cls(app, request) except HandlerMatchException: return None class ViewTags(GenshiHandler): # TODO: RSS template = 'tags.html' methods = set(['GET']) def __init__(self, app, request): GenshiHandler.__init__(self, app, request) path = request.path_info.split('/') try: index = path.find(app.tags_url) except ValueError: raise HandlerMatchException self.tags = path[index+1:] if not self.tags: raise HandlerMatchException @classmethod def match(cls, app, request): # check the method if request.method not in cls.methods: return None try: return cls(app, request) except HandlerMatchException: return None def Get(self): if self.request.GET.get('format') == 'rss': pass # TODO return GenshiHandler.Get(self) class PostTags(Handler): methods = set(['POST']) @classmethod def match(cls, app, request): # check the method if request.method not in cls.methods: return None # check auth if not request.remote_user: return None # check the path if request.path_info.endswith('/%s' % app.post_url): try: return cls(app, request) except HandlerMatchException: return None def __call__(self): # get the url url = self.request.path_info.rsplit('/' + self.app.post_url, 1)[0] # add and remove tags # TODO # redirect to original resource return exc.HTTPSeeOther(location=url) class Index(GenshiHandler): template = 'index.html' methods=set(['GET', 'POST']) def __init__(self, app, request): GenshiHandler.__init__(self, app, request) def Get(self): self.data['name'] = self.request.remote_user or self.app.name return GenshiHandler.Get(self) def Post(self): self.app.name = self.request.POST.get('name', self.app.name) self.redirect(self.link(self.handler_path))