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