Mercurial > templog
annotate web/templog.py @ 236:88dfac30d065
gevent
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Wed, 06 May 2015 19:34:23 +0800 |
| parents | 59379b2bd056 |
| children | abc837a7e60f |
| rev | line source |
|---|---|
| 27 | 1 #!/usr/bin/env python2.7 |
| 2 | |
| 28 | 3 import binascii |
|
146
3b4277aaed3c
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
4 import json |
| 28 | 5 import hmac |
| 29 | 6 import zlib |
| 69 | 7 from datetime import datetime, timedelta |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
30
diff
changeset
|
8 import time |
| 69 | 9 import urllib |
| 10 import sys | |
| 182 | 11 import os |
| 185 | 12 import traceback |
| 13 import fcntl | |
| 28 | 14 |
| 27 | 15 import bottle |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
30
diff
changeset
|
16 from bottle import route, request, response |
| 27 | 17 |
| 28 | 18 import config |
| 19 import log | |
| 185 | 20 import secure |
| 191 | 21 import atomicfile |
| 28 | 22 |
| 69 | 23 DATE_FORMAT = '%Y%m%d-%H.%M' |
| 24 ZOOM_SCALE = 2.0 | |
| 25 | |
| 27 | 26 @route('/update', method='post') |
| 27 def update(): | |
|
146
3b4277aaed3c
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
28 js_enc = request.forms.data |
| 28 | 29 mac = request.forms.hmac |
| 30 | |
|
146
3b4277aaed3c
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
31 if hmac.new(config.HMAC_KEY, js_enc).hexdigest() != mac: |
| 85 | 32 raise bottle.HTTPError(code = 403, output = "Bad key") |
| 28 | 33 |
|
146
3b4277aaed3c
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
34 js = zlib.decompress(binascii.a2b_base64(js_enc)) |
| 28 | 35 |
|
146
3b4277aaed3c
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
36 params = json.loads(js) |
|
3b4277aaed3c
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
37 |
|
3b4277aaed3c
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
38 log.parse(params) |
| 28 | 39 |
| 40 return "OK" | |
| 27 | 41 |
| 42 @route('/graph.png') | |
| 43 def graph(): | |
| 69 | 44 length_minutes = int(request.query.length) |
| 45 end = datetime.strptime(request.query.end, DATE_FORMAT) | |
| 46 start = end - timedelta(minutes=length_minutes) | |
|
30
13fcf497f8b7
parse the arguments for start/length
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
47 |
|
13fcf497f8b7
parse the arguments for start/length
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
48 response.set_header('Content-Type', 'image/png') |
| 69 | 49 start_epoch = time.mktime(start.timetuple()) |
| 50 return log.graph_png(start_epoch, length_minutes * 60) | |
| 27 | 51 |
| 188 | 52 @route('/set/update', method='post') |
| 194 | 53 def set_update(): |
| 188 | 54 post_json = json.loads(request.forms.data) |
| 55 | |
| 56 csrf_blob = post_json['csrf_blob'] | |
| 57 | |
| 194 | 58 if not secure.check_csrf_blob(csrf_blob): |
| 59 bottle.response.status = 403 | |
| 60 return "Bad csrf" | |
| 61 | |
| 62 ret = log.update_params(post_json['params']) | |
| 63 if not ret is True: | |
| 64 bottle.response.status = 403 | |
| 65 return ret | |
| 66 | |
| 67 return "Good" | |
| 188 | 68 |
| 182 | 69 @route('/set') |
| 70 def set(): | |
|
211
59379b2bd056
key fingerprints are case- and whitespace-insensitive.
Matt Johnston <matt@ucc.asn.au>
parents:
202
diff
changeset
|
71 allowed = ["false", "true"][secure.check_user_hash(config.ALLOWED_USERS)] |
|
202
6dd157a12035
Add url link, improve atomicfile
Matt Johnston <matt@ucc.asn.au>
parents:
194
diff
changeset
|
72 response.set_header('Cache-Control', 'no-cache') |
| 185 | 73 return bottle.template('set', |
| 74 inline_data = log.get_params(), | |
| 189 | 75 csrf_blob = secure.get_csrf_blob(), |
| 76 allowed = allowed) | |
| 182 | 77 |
| 27 | 78 @route('/') |
| 79 def top(): | |
| 69 | 80 |
| 81 minutes = int(request.query.get('length', 26*60)) | |
| 82 | |
| 83 if 'end' in request.query: | |
| 84 end = datetime.strptime(request.query.end, DATE_FORMAT) | |
| 85 else: | |
| 86 end = datetime.now() | |
| 87 | |
| 88 if 'zoom' in request.query: | |
| 89 orig_start = end - timedelta(minutes=minutes) | |
| 90 orig_end = end | |
| 91 xpos = int(request.query.x) | |
| 77 | 92 xpos -= config.GRAPH_LEFT_MARGIN * config.ZOOM |
| 69 | 93 |
| 77 | 94 if xpos >= 0 and xpos < config.GRAPH_WIDTH * config.ZOOM: |
| 69 | 95 click_time = orig_start \ |
| 77 | 96 + timedelta(minutes=(float(xpos) / (config.GRAPH_WIDTH * config.ZOOM)) * minutes) |
| 69 | 97 minutes = int(minutes / ZOOM_SCALE) |
| 98 | |
| 99 end = click_time + timedelta(minutes=minutes/2) | |
| 100 else: | |
| 101 # zoom out | |
| 102 minutes = int(minutes*ZOOM_SCALE) | |
| 103 end += timedelta(minutes=minutes/2) | |
| 104 | |
| 105 if end > datetime.now(): | |
| 106 end = datetime.now() | |
| 107 | |
| 108 request.query.replace('length', minutes) | |
| 109 request.query.replace('end', end.strftime(DATE_FORMAT)) | |
| 110 | |
| 111 urlparams = urllib.urlencode(request.query) | |
| 112 return bottle.template('top', urlparams=urlparams, | |
| 113 end = end.strftime(DATE_FORMAT), | |
| 114 length = minutes) | |
| 27 | 115 |
| 103 | 116 @route('/debug') |
| 117 def debuglog(): | |
| 118 response.set_header('Content-Type', 'text/plain') | |
| 119 return log.tail_debug_log() | |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
31
diff
changeset
|
120 |
| 182 | 121 @route('/env') |
| 122 def env(): | |
| 123 response.set_header('Content-Type', 'text/plain') | |
| 189 | 124 #return '\n'.join(traceback.format_stack()) |
| 125 return '\n'.join(("%s %s" % k) for k in request.environ.items()) | |
| 182 | 126 #return str(request.environ) |
| 127 #yield "\n" | |
| 128 #var_lookup = environ['mod_ssl.var_lookup'] | |
| 129 #return var_lookup("SSL_SERVER_I_DN_O") | |
| 130 | |
| 131 @bottle.get('/<filename:re:.*\.js>') | |
| 132 def javascripts(filename): | |
| 185 | 133 response.set_header('Cache-Control', "public, max-age=1296000") |
| 182 | 134 return bottle.static_file(filename, root='static') |
| 135 | |
| 185 | 136 secure.setup_csrf() |
| 182 | 137 |
| 236 | 138 class TemplogBottle(Bottle): |
| 139 def run(*args, **argm): | |
| 140 argm['server'] = 'gevent' | |
| 141 super(TemplogBottle, self).run(*args, **argm) | |
| 142 print "ran custom bottle" | |
| 143 | |
| 144 bottle.app().push(TemplogBottle()) | |
| 145 | |
| 27 | 146 def main(): |
| 61 | 147 #bottle.debug(True) |
| 148 #bottle.run(reloader=True) | |
| 149 bottle.run(server='cgi', reloader=True) | |
| 40 | 150 #bottle.run(port=9999, reloader=True) |
| 27 | 151 |
| 152 if __name__ == '__main__': | |
| 153 main() | |
| 154 |