Mercurial > hg > commentator
annotate commentator/middleware.py @ 3:d0d8524d9495
finish basic API...still doesnt expire results correctly
author | k0s <k0scist@gmail.com> |
---|---|
date | Fri, 26 Feb 2010 11:06:48 -0500 |
parents | 689b9d928dc8 |
children | 451169e51935 |
rev | line source |
---|---|
0 | 1 """ |
2 request dispatcher: | |
3 data persisting across requests should go here | |
4 """ | |
5 | |
6 import os | |
7 import re | |
8 | |
9 from handlers import PostComment | |
2
689b9d928dc8
add date, reflect api change of lxmlmiddleware
k0s <k0scist@gmail.com>
parents:
0
diff
changeset
|
10 #from model import CouchComments |
689b9d928dc8
add date, reflect api change of lxmlmiddleware
k0s <k0scist@gmail.com>
parents:
0
diff
changeset
|
11 from model import PickleComments |
0 | 12 from genshi.template import TemplateLoader |
13 from lxml import etree | |
14 from lxmlmiddleware import LXMLMiddleware | |
15 from paste.fileapp import FileApp | |
16 from pkg_resources import resource_filename | |
17 from string import Template | |
18 from webob import Request, Response, exc | |
19 | |
20 class LaxTemplate(Template): | |
21 idpattern = r'[_a-z0-9]+' | |
22 | |
23 class Commentator(LXMLMiddleware): | |
24 | |
25 ### class level variables | |
26 defaults = { 'auto_reload': 'False', | |
27 'database': 'commentator', | |
2
689b9d928dc8
add date, reflect api change of lxmlmiddleware
k0s <k0scist@gmail.com>
parents:
0
diff
changeset
|
28 'date_format': '%H:%M %m/%d/%Y', |
0 | 29 'template_dirs': '', |
30 'pattern': '.*', | |
31 'path': 'html', | |
32 'url': '.comment', | |
33 'template': 'comment.html' } | |
34 | |
35 def __init__(self, app, **kw): | |
36 | |
37 self.app = app | |
38 | |
39 # set instance parameters from kw and defaults | |
40 for key in self.defaults: | |
41 setattr(self, key, kw.get(key, self.defaults[key])) | |
42 self.auto_reload = self.auto_reload.lower() == 'true' | |
43 | |
44 # request handlers | |
45 self.handlers = [ PostComment ] | |
46 | |
47 # template loader | |
48 self.template_dirs = self.template_dirs.split() | |
49 self.template_dirs.append(resource_filename(__name__, 'templates')) | |
50 self.loader = TemplateLoader(self.template_dirs, | |
51 auto_reload=self.auto_reload) | |
52 | |
53 # URL,path | |
54 assert '#' in self.pattern | |
55 self.url_pattern, self.xpath_pattern = self.pattern.split('#', 1) | |
56 assert '->' in self.xpath_pattern | |
57 self.xpath_pattern, self.mapping = [i.strip() for i in self.xpath_pattern.split('->')] | |
58 | |
59 # string template for URL substitution | |
60 self.mapping = LaxTemplate(self.mapping) | |
61 | |
62 # backend: comment storage | |
63 self.model = PickleComments(self.database) | |
64 | |
65 def __call__(self, environ, start_response): | |
66 | |
67 # get a request object | |
68 request = Request(environ) | |
69 | |
70 # get the path | |
71 path = request.path_info.strip('/').split('/') | |
72 if path == ['']: | |
73 path = [] | |
74 request.environ['path'] = path | |
75 | |
76 # save the path; not sure why i need to do this | |
77 environ['commentator.path_info'] = request.path_info | |
78 | |
79 # match the request to a handler | |
80 for h in self.handlers: | |
81 handler = h.match(self, request) | |
82 if handler is not None: | |
83 break | |
84 else: | |
85 return LXMLMiddleware.__call__(self, environ, start_response) | |
86 | |
87 # get response | |
88 res = handler() | |
89 return res(environ, start_response) | |
90 | |
2
689b9d928dc8
add date, reflect api change of lxmlmiddleware
k0s <k0scist@gmail.com>
parents:
0
diff
changeset
|
91 def manipulate(self, request, response, tree): |
689b9d928dc8
add date, reflect api change of lxmlmiddleware
k0s <k0scist@gmail.com>
parents:
0
diff
changeset
|
92 url_match = re.match(self.url_pattern, request.environ['commentator.path_info']) |
0 | 93 if not url_match: |
94 return tree | |
95 | |
96 # make string template of the groups | |
97 groups_dict = dict([(str(index+1), value) | |
98 for index, value in enumerate(url_match.groups())]) | |
99 | |
3
d0d8524d9495
finish basic API...still doesnt expire results correctly
k0s <k0scist@gmail.com>
parents:
2
diff
changeset
|
100 last_modified = None |
0 | 101 for element in tree.findall(self.xpath_pattern): |
102 | |
103 # get url | |
104 str_dict = groups_dict.copy() | |
105 for key in element.keys(): | |
106 str_dict[key] = element.get(key) | |
107 uri = self.mapping.substitute(str_dict) | |
108 | |
109 # genshi data | |
110 data = {} | |
111 data['comments'] = self.model.comments(uri) | |
3
d0d8524d9495
finish basic API...still doesnt expire results correctly
k0s <k0scist@gmail.com>
parents:
2
diff
changeset
|
112 if data['comments']: |
d0d8524d9495
finish basic API...still doesnt expire results correctly
k0s <k0scist@gmail.com>
parents:
2
diff
changeset
|
113 last_modified = data['comments'][-1]['date'] |
0 | 114 data['action'] = '%s/%s' % (uri, self.url) |
2
689b9d928dc8
add date, reflect api change of lxmlmiddleware
k0s <k0scist@gmail.com>
parents:
0
diff
changeset
|
115 data['date_format'] = self.date_format |
689b9d928dc8
add date, reflect api change of lxmlmiddleware
k0s <k0scist@gmail.com>
parents:
0
diff
changeset
|
116 data['request'] = request |
0 | 117 |
118 # render template | |
119 template = self.loader.load(self.template) | |
120 comments = template.generate(**data).render() | |
121 comments = etree.fromstring(comments) | |
122 element.append(comments) | |
123 | |
3
d0d8524d9495
finish basic API...still doesnt expire results correctly
k0s <k0scist@gmail.com>
parents:
2
diff
changeset
|
124 if last_modified: |
d0d8524d9495
finish basic API...still doesnt expire results correctly
k0s <k0scist@gmail.com>
parents:
2
diff
changeset
|
125 import pdb; pdb.set_trace() |
0 | 126 return tree |