annotate licenser/web.py @ 27:7e0c931a201d

web licenser now works
author Jeff Hammel <jhammel@mozilla.com>
date Fri, 25 Nov 2011 00:57:38 -0800
parents 0faf3e7b593a
children eb3749dbdbf5
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
25
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
1 #!/usr/bin/env python
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
2
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
3 """
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
4 web handler for MPL licenser
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
5 """
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
6
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
7 import licenses
26
0faf3e7b593a more stubbing
Jeff Hammel <jhammel@mozilla.com>
parents: 25
diff changeset
8 import tempfile
27
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
9 from StringIO import StringIO
25
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
10 from webob import Request, Response, exc
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
11
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
12 class Handler(object):
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
13
27
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
14 required = set(['author', 'email', 'url'])
25
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
15
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
16 template = """<html>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
17 <head><title>MPL licenser</title></head>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
18 <body>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
19 <h1>MPL Licenser</h1>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
20 <div><i>license your files with MPL. Currently works for files of type: %(filetypes)s</i></div>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
21 <div><form method="POST" enctype="multipart/form-data">
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
22 <p>Enter a file: <input type="file" name="file"/></p>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
23 <p>or a URL: <input type="text" name="url" value="%(url)s"/></p>
27
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
24 <p>And enter your name: <input type="text" name="author" value="%(author)s"/></p>
25
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
25 <p>and email: <input type="text" name="email" value="%(email)s"/></p>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
26 <input type="submit"/>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
27 </body>
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
28 </html>"""
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
29
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
30 def __init__(self, **kw):
27
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
31 self.license = licenses.MPL()
25
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
32 self.methods = {'GET': self.GET,
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
33 'POST': self.POST}
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
34
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
35 def __call__(self, environ, start_response):
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
36 request = Request(environ)
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
37 if request.method in self.methods:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
38 response = self.methods[request.method](request)
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
39 else:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
40 response = exc.HTTPMethodNotAllowed()
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
41 return response(environ, start_response)
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
42
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
43 def GET(self, request):
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
44 variables = {}
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
45 for i in self.required:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
46 if i in request.GET:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
47 variables[i] = request.GET[i]
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
48 if set(variables.keys()) == self.required:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
49 try:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
50 pass
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
51 except:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
52 pass
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
53 return self.render(**variables)
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
54
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
55 def render(self, **variables):
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
56 for i in self.required:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
57 if i not in variables:
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
58 variables[i] = ''
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
59 variables['filetypes'] = licenses.filetypes_string()
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
60 return Response(content_type='text/html', body=(self.template % variables))
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
61
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
62 def POST(self, request):
27
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
63 variables = dict([(i, request.params[i])
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
64 for i in self.required
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
65 if i in request.params])
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
66 if 'file' in request.POST and hasattr(request.POST['file'], 'filename'):
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
67 body = request.POST['file'].file.getvalue()
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
68 filename = request.POST['file'].filename
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
69 elif 'url' in request.POST:
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
70 try:
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
71 pass
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
72 except:
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
73 return self.render(**variables)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
74 if not set(['author', 'email']).issubset(request.params):
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
75 return self.render(**variables)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
76 author = request.params['author']
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
77 email = request.params['email']
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
78 retval = self.interpolate(filename, body, author, email)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
79 return Response(content_type='text/plain', body=retval)
25
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
80
27
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
81 def interpolate(self, filename, body, author, email):
26
0faf3e7b593a more stubbing
Jeff Hammel <jhammel@mozilla.com>
parents: 25
diff changeset
82
27
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
83 buffer = StringIO()
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
84 buffer.write(body)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
85 value = buffer.getvalue().splitlines()
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
86 if self.license.has_license(buffer):
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
87 return body
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
88 filetype = self.license.filetype(filename)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
89 if not filetype:
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
90 return body
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
91 variables = self.license.obtain_variables(author=author, email=email)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
92 license = self.license.interpolate_license(**variables)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
93 value = [i + '\r\n' for i in value]
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
94 filetype._lines = value
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
95 buffer = StringIO()
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
96 filetype.filename = buffer
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
97 filetype(license)
7e0c931a201d web licenser now works
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
98 return buffer.getvalue()
26
0faf3e7b593a more stubbing
Jeff Hammel <jhammel@mozilla.com>
parents: 25
diff changeset
99
25
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
100 if __name__ == '__main__':
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
101 from wsgiref import simple_server
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
102 app = Handler()
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
103 server = simple_server.make_server(host='0.0.0.0', port=8080, app=app)
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
104 server.serve_forever()
8729685a2cf6 stub web interface
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
105