view uploader/dispatcher.py @ 3:bb2d34fb5de4

allow to be used as middleware
author k0s <k0scist@gmail.com>
date Sun, 27 Dec 2009 16:27:34 -0500
parents 0b5fce452087
children e5efa999911a
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)
        self.auth = self.auth.lower() == 'true'
        self.subpath = self.subpath.lower() == 'true'
        if self.subpath:
            if self.query_string:
                self.handlers = [ SubpathGet, SubpathPost ]
            else:
                self.handlers = [ 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)