view web/templog.py @ 362:8fea6144951b

improve graphs
author Matt Johnston <matt@ucc.asn.au>
date Sun, 24 Jun 2012 23:48:11 +0800
parents d6219df77c41
children 62112fc2af21
line wrap: on
line source

#!/usr/bin/env python2.7

import binascii
import hmac
import zlib
import datetime
import time

import bottle
from bottle import route, request, response

import config
import log

@route('/update', method='post')
def update():
    enc_lines = request.forms.lines
    mac = request.forms.hmac

    if hmac.new(config.HMAC_KEY, enc_lines).hexdigest() != mac:
        raise HTTPError(code = 403, output = "Bad key")

    lines = zlib.decompress(binascii.a2b_base64(enc_lines)).split('\n')

    log.parse(lines)

    return "OK"

@route('/graph.png')
def graph():
    # url takes time in hours or days
    if 'day' in request.query:
        start_day = datetime.datetime.strptime(request.query.day, '%Y%m%d')
        start = time.mktime(start_day.timetuple())
        length = int(request.query.get('length', 5)) * 3600 * 24
    else:
        if 'hour' in request.query:
            start_hour = datetime.datetime.strptime(request.query.hour, '%Y%m%d%H')
        else:
            start_hour = datetime.datetime.now() - datetime.timedelta(days=1)

        start = time.mktime(start_hour.timetuple())
        length = int(request.query.get('length', 36)) * 3600

    response.set_header('Content-Type', 'image/png')
    return log.graph_png(start, length)

@route('/')
def top():
    return bottle.template('top', urlparams=request.query_string)

@route('/test')
def test():
    import config
    import os
    f = open('%s/testout' % config.DATA_PATH, 'a+')
    f.write("more")
    f.flush()
    f.close()

    return 'done'

def main():
    bottle.debug(True)
    bottle.run(server='cgi')
    #bottle.run(port=9999, reloader=True)

if __name__ == '__main__':
    main()