view web/index.py @ 338:12123e390169

More minor work
author Matt Johnston <matt@ucc.asn.au>
date Thu, 14 Jun 2012 00:00:57 +0800
parents f575ef538f5d
children
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.length) * 3600 * 24
    else:
        start_hour = datetime.datetime.strptime(request.query.hour, '%Y%m%d%H')
        start = time.mktime(start_hour.timetuple())
        length = int(request.query.length) * 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)

def main():
    bottle.debug(True)
    bottle.run(port=9999, reloader=True)

if __name__ == '__main__':
    main()