view uploader/dispatcher.py @ 12:f60ad139acc1

better error message
author egj@socialplanning.org
date Tue, 02 Feb 2010 17:58:00 +0000
parents d2990750e5d6
children 657e8df14187
line wrap: on
line source

"""
request dispatcher
"""

import os
from handlers import Get, Post, SubpathGet, SubpathPost
from webob import Request, exc

class Dispatcher(object):

    ### class level variables
    defaults = { 'directory': None,
                 'auth': 'False',
                 'query_string': None,
                 'subpath': 'False',
                 'app': None}

    def __init__(self, **kw):
        for key in self.defaults:
            setattr(self, key, kw.get(key, self.defaults[key]))
        assert os.path.exists(self.directory), "The base directory %s does not exist; uploader needs it!" % self.directory
        if self.app:
            assert hasattr(self.app, '__call__')
        self.auth = self.auth.lower() == 'true'
        self.subpath = self.subpath.lower() == 'true'
        if self.subpath:
            self.handlers = [ SubpathGet, SubpathPost ]
        else:
            self.handlers = [ Get, Post ]

    ### methods dealing with HTTP
    def __call__(self, environ, start_response):
        request = Request(environ)
        if self.auth and not request.remote_user:
            return exc.HTTPUnauthorized()(environ, start_response)
        for h in self.handlers:
            if h.match(self, request):
                handler = h(self, request)
                break
        else:
            if self.app:
                return self.app(environ, start_response)
            handler = exc.HTTPNotFound
        res = handler()
        return res(environ, start_response)