Mercurial > templog
annotate web/templog.py @ 552:a335760ad447
uwsgi bits
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Sat, 06 Jun 2015 15:34:03 +0800 |
| parents | 0baa57b6d9ca |
| children | 8ef52f27cf95 |
| rev | line source |
|---|---|
| 333 | 1 #!/usr/bin/env python2.7 |
| 2 | |
| 334 | 3 import binascii |
|
445
5b9dc87c988f
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
409
diff
changeset
|
4 import json |
| 334 | 5 import hmac |
| 335 | 6 import zlib |
| 375 | 7 from datetime import datetime, timedelta |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
336
diff
changeset
|
8 import time |
| 375 | 9 import urllib |
| 10 import sys | |
| 482 | 11 import os |
| 485 | 12 import traceback |
| 13 import fcntl | |
| 541 | 14 import hashlib |
| 334 | 15 |
| 333 | 16 import bottle |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
336
diff
changeset
|
17 from bottle import route, request, response |
| 333 | 18 |
| 334 | 19 import config |
| 20 import log | |
| 485 | 21 import secure |
| 489 | 22 import atomicfile |
| 334 | 23 |
| 375 | 24 DATE_FORMAT = '%Y%m%d-%H.%M' |
| 25 ZOOM_SCALE = 2.0 | |
| 26 | |
| 539 | 27 class TemplogBottle(bottle.Bottle): |
| 28 def run(*args, **argm): | |
| 29 argm['server'] = 'gevent' | |
| 30 super(TemplogBottle, self).run(*args, **argm) | |
| 31 print "ran custom bottle" | |
| 32 | |
|
540
0f665a84b581
gevent doesn't work well with subprocess
Matt Johnston <matt@ucc.asn.au>
parents:
539
diff
changeset
|
33 #bottle.default_app.push(TemplogBottle()) |
| 539 | 34 |
| 35 secure.setup_csrf() | |
| 36 | |
| 333 | 37 @route('/update', method='post') |
| 38 def update(): | |
|
445
5b9dc87c988f
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
409
diff
changeset
|
39 js_enc = request.forms.data |
| 334 | 40 mac = request.forms.hmac |
| 41 | |
| 541 | 42 h = hmac.new(config.HMAC_KEY, js_enc.strip(), hashlib.sha256).hexdigest() |
| 43 if h != mac: | |
| 391 | 44 raise bottle.HTTPError(code = 403, output = "Bad key") |
| 334 | 45 |
|
445
5b9dc87c988f
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
409
diff
changeset
|
46 js = zlib.decompress(binascii.a2b_base64(js_enc)) |
| 334 | 47 |
|
445
5b9dc87c988f
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
409
diff
changeset
|
48 params = json.loads(js) |
|
5b9dc87c988f
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
409
diff
changeset
|
49 |
|
5b9dc87c988f
update web to handle new style params
Matt Johnston <matt@ucc.asn.au>
parents:
409
diff
changeset
|
50 log.parse(params) |
| 334 | 51 |
| 52 return "OK" | |
| 333 | 53 |
| 542 | 54 def make_graph(length, end): |
| 55 length_minutes = int(length) | |
| 56 end = datetime.strptime(end, DATE_FORMAT) | |
| 57 start = end - timedelta(minutes=length_minutes) | |
| 58 | |
| 59 start_epoch = time.mktime(start.timetuple()) | |
| 60 return log.graph_png(start_epoch, length_minutes * 60) | |
| 61 | |
| 62 def encode_data(data, mimetype): | |
| 63 return 'data:%s;base64,%s' % (mimetype, binascii.b2a_base64(data).rstrip()) | |
| 64 | |
| 65 | |
| 333 | 66 @route('/graph.png') |
| 67 def graph(): | |
|
336
ba4c4df13487
parse the arguments for start/length
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
68 response.set_header('Content-Type', 'image/png') |
| 549 | 69 minutes, endstr = get_request_zoom() |
| 70 return make_graph(minutes, endstr) | |
| 333 | 71 |
| 487 | 72 @route('/set/update', method='post') |
| 492 | 73 def set_update(): |
| 487 | 74 post_json = json.loads(request.forms.data) |
| 75 | |
| 76 csrf_blob = post_json['csrf_blob'] | |
| 77 | |
| 492 | 78 if not secure.check_csrf_blob(csrf_blob): |
| 79 bottle.response.status = 403 | |
| 80 return "Bad csrf" | |
| 81 | |
| 82 ret = log.update_params(post_json['params']) | |
| 83 if not ret is True: | |
| 84 bottle.response.status = 403 | |
| 85 return ret | |
| 86 | |
| 87 return "Good" | |
| 487 | 88 |
| 482 | 89 @route('/set') |
| 90 def set(): | |
|
505
ad846b9bdd10
key fingerprints are case- and whitespace-insensitive.
Matt Johnston <matt@ucc.asn.au>
parents:
501
diff
changeset
|
91 allowed = ["false", "true"][secure.check_user_hash(config.ALLOWED_USERS)] |
|
501
236e5d131b3e
Add url link, improve atomicfile
Matt Johnston <matt@ucc.asn.au>
parents:
492
diff
changeset
|
92 response.set_header('Cache-Control', 'no-cache') |
| 485 | 93 return bottle.template('set', |
| 94 inline_data = log.get_params(), | |
| 488 | 95 csrf_blob = secure.get_csrf_blob(), |
| 96 allowed = allowed) | |
| 482 | 97 |
| 549 | 98 def get_request_zoom(): |
| 99 """ returns (length, end) tuple. | |
| 100 length is in minutes, end is a DATE_FORMAT string """ | |
| 375 | 101 minutes = int(request.query.get('length', 26*60)) |
| 102 | |
| 103 if 'end' in request.query: | |
| 104 end = datetime.strptime(request.query.end, DATE_FORMAT) | |
| 105 else: | |
| 106 end = datetime.now() | |
| 107 | |
| 108 if 'zoom' in request.query: | |
| 109 orig_start = end - timedelta(minutes=minutes) | |
| 110 orig_end = end | |
| 548 | 111 scale = float(request.query.scaledwidth) / config.GRAPH_WIDTH |
| 112 xpos = int(request.query.x) / scale | |
| 383 | 113 xpos -= config.GRAPH_LEFT_MARGIN * config.ZOOM |
| 375 | 114 |
| 383 | 115 if xpos >= 0 and xpos < config.GRAPH_WIDTH * config.ZOOM: |
| 375 | 116 click_time = orig_start \ |
| 383 | 117 + timedelta(minutes=(float(xpos) / (config.GRAPH_WIDTH * config.ZOOM)) * minutes) |
| 375 | 118 minutes = int(minutes / ZOOM_SCALE) |
| 119 | |
| 120 end = click_time + timedelta(minutes=minutes/2) | |
| 121 else: | |
| 122 # zoom out | |
| 123 minutes = int(minutes*ZOOM_SCALE) | |
| 124 end += timedelta(minutes=minutes/2) | |
| 125 | |
| 126 if end > datetime.now(): | |
| 127 end = datetime.now() | |
| 549 | 128 |
| 129 endstr = end.strftime(DATE_FORMAT) | |
| 130 return (minutes, endstr) | |
| 131 | |
| 132 @route('/') | |
| 133 def top(): | |
| 134 minutes, endstr = get_request_zoom() | |
| 135 | |
| 375 | 136 request.query.replace('length', minutes) |
| 549 | 137 request.query.replace('end', endstr) |
| 375 | 138 |
| 139 urlparams = urllib.urlencode(request.query) | |
| 549 | 140 graphdata = encode_data(make_graph(minutes, endstr), 'image/png') |
| 375 | 141 return bottle.template('top', urlparams=urlparams, |
| 549 | 142 end = endstr, |
| 542 | 143 length = minutes, |
| 144 graphwidth = config.GRAPH_WIDTH, | |
| 145 graphdata = graphdata) | |
| 333 | 146 |
| 409 | 147 @route('/debug') |
| 148 def debuglog(): | |
| 149 response.set_header('Content-Type', 'text/plain') | |
| 150 return log.tail_debug_log() | |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
337
diff
changeset
|
151 |
| 482 | 152 @route('/env') |
| 153 def env(): | |
| 154 response.set_header('Content-Type', 'text/plain') | |
| 488 | 155 #return '\n'.join(traceback.format_stack()) |
| 156 return '\n'.join(("%s %s" % k) for k in request.environ.items()) | |
| 482 | 157 #return str(request.environ) |
| 158 #yield "\n" | |
| 159 #var_lookup = environ['mod_ssl.var_lookup'] | |
| 160 #return var_lookup("SSL_SERVER_I_DN_O") | |
| 161 | |
| 552 | 162 @route('/wait') |
| 163 def wait(): | |
| 164 response.set_header('Content-Type', 'text/plain') | |
| 165 yield 'done' | |
| 166 | |
| 482 | 167 @bottle.get('/<filename:re:.*\.js>') |
| 168 def javascripts(filename): | |
| 485 | 169 response.set_header('Cache-Control', "public, max-age=1296000") |
| 482 | 170 return bottle.static_file(filename, root='static') |
| 171 | |
| 333 | 172 def main(): |
| 552 | 173 """ for standalone testing """ |
| 367 | 174 #bottle.debug(True) |
| 175 #bottle.run(reloader=True) | |
| 176 bottle.run(server='cgi', reloader=True) | |
| 346 | 177 #bottle.run(port=9999, reloader=True) |
| 333 | 178 |
| 179 if __name__ == '__main__': | |
| 180 main() | |
| 181 |