Mercurial > hg > WSGraph
annotate wsgraph/web.py @ 21:f56d27f03e4a
pretend we are slick but we can be so much slicker
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Mon, 10 Dec 2012 17:50:45 -0800 |
parents | 542232f858a9 |
children | a401bf63c938 |
rev | line source |
---|---|
0 | 1 #!/usr/bin/env python |
2 | |
3 """ | |
4 web handler for WSGraph | |
5 """ | |
6 | |
7 import json | |
12
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
8 import sys |
0 | 9 from webob import Request, Response, exc |
10 | |
5 | 11 # rank constants |
4 | 12 GRAPH = 0 |
13 NODE = 1 | |
14 EDGE = 2 | |
15 | |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
16 # XXX stubbing hack |
7 | 17 class JSONformatter(object): |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
18 |
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
19 def __init__(self, content_type='application/json', sort_keys=True): |
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
20 self.content_type = content_type |
7 | 21 self.sort_keys = sort_keys |
22 | |
19 | 23 def format(self, instance): |
24 """front-end to json.dumps""" | |
25 return json.dumps(instance, sort_keys=self.sort_keys) | |
7 | 26 |
20
542232f858a9
rearrange this all around for argument order
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
27 def node(self, request, graph, node): |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
28 return Response(content_type=self.content_type, |
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
29 body=self.format(graph.node(node))) |
7 | 30 |
20
542232f858a9
rearrange this all around for argument order
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
31 def edge(self, request, graph, node1, node2): |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
32 return Response(content_type=self.content_type, |
20
542232f858a9
rearrange this all around for argument order
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
33 body=self.format(graph.edge(node1, node2))) |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
34 |
20
542232f858a9
rearrange this all around for argument order
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
35 def graph(self, request, graph): |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
36 return Response(content_type=self.content_type, |
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
37 body='hello? is it me you\'re looking for?') |
0 | 38 |
20
542232f858a9
rearrange this all around for argument order
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
39 |
0 | 40 class Dispatcher(object): |
41 | |
42 def __init__(self, graph, | |
21
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
43 graph_formatters, |
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
44 node_formatters, |
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
45 edge_formatters, |
0 | 46 require_auth=False): |
47 self.graph = graph | |
48 self.require_auth = require_auth | |
5 | 49 |
13
7459702bf574
mostly spacing fix + comment
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
50 # view formatters |
5 | 51 self.formatters = {GRAPH: graph_formatters, |
52 NODE: node_formatters, | |
53 EDGE: edge_formatters} | |
16 | 54 for key, value in self.formatters.items(): |
5 | 55 # ensure default formatter |
56 assert None in value | |
0 | 57 |
3 | 58 self.methods = dict([(i, getattr(self, i)) |
59 for i in dir(self) | |
60 if i.isupper()]) | |
61 | |
0 | 62 def __call__(self, environ, start_response): |
63 request = Request(environ) | |
3 | 64 method = self.methods.get(request.method) |
65 if method is None: | |
66 return exc.HTTPMethodNotAllowed()(environ, start_response) | |
67 response = method(request) | |
0 | 68 return response(environ, start_response) |
69 | |
70 @staticmethod | |
5 | 71 def path_segments(path): |
72 """split a path into segments""" | |
73 segments = path.strip('/').split('/') | |
74 if segments == ['']: | |
75 segments = [] | |
76 return segments | |
0 | 77 |
78 # HTTP methods | |
79 | |
80 def GET(self, request): | |
2
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
81 """ |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
82 respond to a GET request |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
83 |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
84 Formatters are keyed off of |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
85 |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
86 formatters = {0: { |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
87 } |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
88 |
4 | 89 A graph formatter takes the following arguments: |
90 | |
6 | 91 def sample_graph_formatter(graph, request): |
92 | |
93 A node formatter takes the following arguments: | |
94 | |
95 def sample_node_formatter(node, graph, request): | |
96 | |
97 An edge formatter takes the following arguments: | |
98 | |
99 def sample_edge_formatter(edge, graph, request): | |
4 | 100 |
2
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
101 API: |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
102 |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
103 ?format=<format> |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
104 |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
105 """ |
0 | 106 |
7 | 107 # get resource requestor |
5 | 108 segments = self.path_segments(request.path_info) |
12
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
109 rank = len(segments) |
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
110 if rank not in (0,1,2): |
4 | 111 return exc.HTTPNotFound() |
0 | 112 |
17
1053290ba067
finish up this throwaway thing
Jeff Hammel <jhammel@mozilla.com>
parents:
16
diff
changeset
|
113 # is resource in the graph? |
12
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
114 if (rank == EDGE) or (rank == NODE): |
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
115 if tuple(segments) not in self.graph: |
14 | 116 return exc.HTTPNotFound() |
7 | 117 |
0 | 118 # formatter |
119 formatter = self.formatters[len(segments)] | |
20
542232f858a9
rearrange this all around for argument order
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
120 return formatter(request, self.graph, *segments) |
0 | 121 return Response(content_type='text/plain', |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
122 body=content) |
0 | 123 |
124 def POST(self, request): | |
2
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
125 """ |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
126 respond to a POST request |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
127 |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
128 API: |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
129 |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
130 ?update : |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
131 """ |
7a3b21cafc00
start to sketch out REST API
Jeff Hammel <jhammel@mozilla.com>
parents:
0
diff
changeset
|
132 |
4 | 133 return exc.HTTPSeeOther(location='/') # TODO: /path/to/self |
0 | 134 |
135 def HEAD(self, request): | |
136 """respond to a HEAD request""" | |
137 raise NotImplementedError | |
138 | |
3 | 139 def OPTIONS(self, request): |
4 | 140 return Response() # TODO: Allow=', '.join(self.methods); Content-Length=0 |
3 | 141 |
0 | 142 |
12
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
143 def main(args=sys.argv[1:]): |
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
144 """example web server""" |
3 | 145 |
0 | 146 # imports |
147 from wsgiref import simple_server | |
3 | 148 from model import MemoryCache |
149 from optparse import OptionParser | |
0 | 150 |
3 | 151 # parse command line options |
152 parser = OptionParser() | |
153 parser.add_option('-p', '--port', type='int', default=8080, | |
154 help="port to serve on") | |
155 options, args = parser.parse_args() | |
156 | |
157 # example model | |
0 | 158 graph = MemoryCache() |
159 | |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
160 # example formatter(s) |
21
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
161 formatter = JSONformatter(content_type='text/plain') |
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
162 _formatters = {} |
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
163 _formatters = dict([('%s_formatters' % i, {None: getattr(formatter, i)}) |
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
164 for i in 'graph', 'node', 'edge']) |
18
b710a2374c8d
more stubbing, goddamn content_type
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
165 |
3 | 166 # WSGI app |
21
f56d27f03e4a
pretend we are slick but we can be so much slicker
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
167 app = Dispatcher(graph=graph, **_formatters) |
3 | 168 server = simple_server.make_server(host='0.0.0.0', port=options.port, app=app) |
169 print 'http://localhost:%s/' % options.port | |
0 | 170 server.serve_forever() |
171 | |
12
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
172 if __name__ == '__main__': |
421d5119e324
reluctantly make this a CLI entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
173 main() |