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))