Mercurial > templog
annotate web/log.py @ 348:536128b90573
mostly works
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Sat, 23 Jun 2012 23:36:00 +0800 |
| parents | d6219df77c41 |
| children | ea99aae87884 96c336896201 |
| rev | line source |
|---|---|
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
1 #:vim:et:ts=4:sts=4:sw=4: |
| 333 | 2 import rrdtool |
| 3 import os | |
| 335 | 4 import os.path |
| 333 | 5 import sys |
| 335 | 6 import glob |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
7 import hashlib |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
8 import tempfile |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
9 import time |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
10 import syslog |
| 335 | 11 from colorsys import hls_to_rgb |
| 333 | 12 |
| 334 | 13 import config |
| 333 | 14 |
| 15 def sensor_rrd_path(s): | |
| 334 | 16 return '%s/sensor_%s.rrd' % (config.DATA_PATH, s) |
| 333 | 17 |
| 335 | 18 # returns (path, sensor_name) tuples |
| 19 def all_sensors(): | |
| 20 return [(r, os.path.basename(r[:-4])) | |
| 21 for r in glob.glob('%s/*.rrd' % config.DATA_PATH)] | |
| 22 | |
| 333 | 23 def create_rrd(sensor_id): |
| 24 rrdtool.create(sensor_rrd_path(sensor_id), '-s', '300', | |
| 25 'DS:temp:GAUGE:600:-10:100', | |
| 26 'RRA:AVERAGE:0.5:1:1051200') | |
| 27 | |
| 335 | 28 |
| 29 # stolen from viewmtn, stolen from monotone-viz | |
| 30 def colour_from_string(str): | |
| 31 def f(off): | |
| 32 return ord(hashval[off]) / 256.0 | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
33 hashval = hashlib.sha1(str).digest() |
| 335 | 34 hue = f(5) |
| 35 li = f(1) * 0.15 + 0.55 | |
| 36 sat = f(2) * 0.5 + .5 | |
| 37 return ''.join(["%.2x" % int(x * 256) for x in hls_to_rgb(hue, li, sat)]) | |
| 38 | |
| 39 def graph_png(start, length): | |
| 40 rrds = all_sensors() | |
| 41 | |
| 42 graph_args = [] | |
| 43 for n, (rrdfile, sensor) in enumerate(rrds): | |
| 44 vname = 'temp%d' % n | |
| 45 graph_args.append('DEF:%(vname)s=%(rrdfile)s:temp:AVERAGE' % locals()) | |
| 46 width = config.LINE_WIDTH | |
| 47 legend = config.SENSOR_NAMES.get(sensor, sensor) | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
48 colour = config.SENSOR_COLOURS.get(legend, colour_from_string(sensor)) |
| 335 | 49 graph_args.append('LINE%(width)f:%(vname)s#%(colour)s:%(legend)s' % locals()) |
| 50 | |
| 51 tempf = tempfile.NamedTemporaryFile() | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
52 args = [tempf.name, '-s', str(int(start)), |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
53 '-e', str(int(start+length)), |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
54 '-w', str(config.GRAPH_WIDTH), |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
55 '-h', str(config.GRAPH_HEIGHT), |
| 335 | 56 '--slope-mode', |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
57 '--border', '0', |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
58 '--color', 'BACK#ffffff', |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
59 '--imgformat', 'PNG'] \ |
| 335 | 60 + graph_args |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
61 if config.GRAPH_FONT: |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
62 args += ['--font', 'DEFAULT:0:%s' % config.GRAPH_FONT] |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
63 print>>sys.stderr, args |
| 335 | 64 rrdtool.graph(*args) |
| 65 return tempf.read() | |
| 66 | |
| 333 | 67 def sensor_update(sensor_id, measurements, first_real_time, time_step): |
| 68 try: | |
| 69 open(sensor_rrd_path(sensor_id)) | |
| 70 except IOError, e: | |
| 71 create_rrd(sensor_id) | |
| 72 | |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
73 print>>sys.stderr, sensor_id, measurements, first_real_time, time_step |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
74 |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
75 if measurements: |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
76 values = ['%d:%f' % p for p in |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
77 zip((first_real_time + time_step*t for t in xrange(len(measurements))), |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
78 measurements)] |
| 333 | 79 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
80 rrdfile = sensor_rrd_path(sensor_id) |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
81 print>>sys.stderr, values |
| 346 | 82 # XXX what to do here when it fails... |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
83 rrdtool.update(rrdfile, *values) |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
84 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
85 # be paranoid |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
86 f = file(rrdfile) |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
87 os.fsync(f.fileno()) |
| 333 | 88 |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
89 def record_debug(lines): |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
90 f = open('%s/debug.log' % config.DATA_PATH, 'a+') |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
91 f.write('===== %s =====\n' % time.strftime('%a, %d %b %Y %H:%M:%S')) |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
92 f.writelines(('%s\n' % s for s in lines)) |
|
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
93 f.flush() |
|
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
94 return f |
|
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
95 |
| 333 | 96 def parse(lines): |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
97 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
98 debugf = record_debug(lines) |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
99 |
| 333 | 100 entries = dict(l.split('=', 1) for l in lines) |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
101 if len(entries) != len(lines): |
| 333 | 102 raise Exception("Keys are not unique") |
| 103 | |
| 104 num_sensors = int(entries['sensors']) | |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
105 num_measurements = int(entries['measurements']) |
| 333 | 106 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
107 sensors = [entries['sensor_id%d' % n] for n in xrange(num_sensors)] |
| 333 | 108 |
| 109 meas = [] | |
| 110 for s in sensors: | |
| 111 meas.append([]) | |
| 112 | |
| 334 | 113 def val_scale(v): |
| 114 # convert decidegrees to degrees | |
| 115 return 0.1 * v | |
| 116 | |
| 333 | 117 for n in xrange(num_measurements): |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
118 vals = [val_scale(int(x)) for x in entries["meas%d" % n].strip().split()] |
| 333 | 119 if len(vals) != num_sensors: |
| 120 raise Exception("Wrong number of sensors for measurement %d" % n) | |
| 121 # we make an array of values for each sensor | |
| 122 for s in xrange(num_sensors): | |
| 123 meas[s].append(vals[s]) | |
| 124 | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
125 avr_now = float(entries['now']) |
| 333 | 126 avr_first_time = float(entries['first_time']) |
| 127 time_step = float(entries['time_step']) | |
| 128 | |
| 346 | 129 #sqlite |
| 130 # - time | |
| 131 # - voltage | |
| 132 # - boot time | |
| 133 | |
| 333 | 134 first_real_time = time.time() - (avr_now - avr_first_time) |
| 135 | |
| 136 for sensor_id, measurements in zip(sensors, meas): | |
| 346 | 137 # XXX sqlite add |
| 333 | 138 sensor_update(sensor_id, measurements, first_real_time, time_step) |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
139 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
140 debugf.write("Updated %d sensors\n" % len(sensors)) |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
141 debugf.flush() |