Mercurial > hg > flowerbed

annotate flowerbed/dispatcher.py @ 0:0613e2bb0ebe

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
initial creation of flowerbed
author Jeff Hammel <jhammel@mozilla.com>
date Mon, 20 Dec 2010 09:24:48 -0800
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
1 """
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
2 request dispatcher:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
3 data persisting across requests should go here
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
4 """
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
5
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
6 import os
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
7
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
8 from handlers import Index
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
9
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
10 from genshi.template import TemplateLoader
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
11 from paste.fileapp import FileApp
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
12 from pkg_resources import resource_filename
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
13 from webob import Request, Response, exc
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
14
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
15 class Dispatcher(object):
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
16
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
17 ### class level variables
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
18 defaults = { 'auto_reload': 'False',
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
19 'template_dirs': '',
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
20 'app': None,
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
21 'name': 'anonymous' }
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
22
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
23 def __init__(self, **kw):
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
24
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
25 # set instance parameters from kw and defaults
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
26 for key in self.defaults:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
27 setattr(self, key, kw.get(key, self.defaults[key]))
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
28 self.auto_reload = self.auto_reload.lower() == 'true'
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
29
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
30 # request handlers
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
31 self.handlers = [ Index ]
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
32
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
33 # endpoint app if used as middleware
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
34 if self.app:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
35 assert hasattr(self.app, '__call__')
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
36
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
37 # template loader
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
38 self.template_dirs = self.template_dirs.split()
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
39 self.template_dirs.append(resource_filename(__name__, 'templates'))
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
40 self.loader = TemplateLoader(self.template_dirs,
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
41 auto_reload=self.auto_reload)
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
42
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
43 def __call__(self, environ, start_response):
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
44
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
45 # get a request object
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
46 request = Request(environ)
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
47
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
48 # get the path
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
49 path = request.path_info.strip('/').split('/')
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
50 if path == ['']:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
51 path = []
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
52 request.environ['path'] = path
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
53
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
54 # match the request to a handler
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
55 for h in self.handlers:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
56 handler = h.match(self, request)
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
57 if handler is not None:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
58 break
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
59 else:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
60 if self.app:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
61 return self.app(environ, start_response)
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
62 handler = exc.HTTPNotFound
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
63
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
64 # add navigation links to handler [example]
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
65 if hasattr(handler, 'data'):
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
66 handler.data.setdefault('links', [])
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
67 for h in self.handlers:
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
68 handler.data['links'].append((handler.link(h.handler_path),
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
69 h.__name__))
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
70
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
71 # get response
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
72 res = handler()
0613e2bb0ebe initial creation of flowerbed
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
73 return res(environ, start_response)