Mercurial > templog
annotate web/log.py @ 378:16f96019a9e0
merge
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Sat, 30 Jun 2012 22:00:46 +0800 |
| parents | 55710361804b |
| children | 0a8639039453 |
| 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 |
| 349 | 11 import sqlite3 |
|
355
a1aa4176ca2c
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
351
diff
changeset
|
12 import traceback |
| 373 | 13 import datetime |
| 335 | 14 from colorsys import hls_to_rgb |
| 333 | 15 |
| 334 | 16 import config |
| 333 | 17 |
| 18 def sensor_rrd_path(s): | |
| 334 | 19 return '%s/sensor_%s.rrd' % (config.DATA_PATH, s) |
| 333 | 20 |
| 335 | 21 # returns (path, sensor_name) tuples |
| 22 def all_sensors(): | |
| 23 return [(r, os.path.basename(r[:-4])) | |
| 24 for r in glob.glob('%s/*.rrd' % config.DATA_PATH)] | |
| 25 | |
| 333 | 26 def create_rrd(sensor_id): |
|
350
03322c38b4a9
Create with a start date so that it works
Matt Johnston <matt@ucc.asn.au>
parents:
346
diff
changeset
|
27 # start date of 10 seconds into 1970 is used so that we can |
|
03322c38b4a9
Create with a start date so that it works
Matt Johnston <matt@ucc.asn.au>
parents:
346
diff
changeset
|
28 # update with prior values straight away. |
| 367 | 29 if 'voltage' in sensor_id: |
| 369 | 30 args = [ |
| 31 '--step', '3600', | |
| 32 'DS:temp:GAUGE:7200:1:10', | |
| 33 'RRA:AVERAGE:0.5:1:87600'] | |
| 367 | 34 else: |
| 369 | 35 args = [ |
| 36 '--step', '300', | |
| 37 'DS:temp:GAUGE:600:-10:100', | |
| 367 | 38 'RRA:AVERAGE:0.5:1:1051200'] |
| 333 | 39 |
| 367 | 40 rrdtool.create(sensor_rrd_path(sensor_id), |
| 369 | 41 '--start', 'now-60d', |
| 367 | 42 *args) |
| 335 | 43 |
| 44 # stolen from viewmtn, stolen from monotone-viz | |
| 45 def colour_from_string(str): | |
| 46 def f(off): | |
| 47 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
|
48 hashval = hashlib.sha1(str).digest() |
| 335 | 49 hue = f(5) |
| 50 li = f(1) * 0.15 + 0.55 | |
| 51 sat = f(2) * 0.5 + .5 | |
| 52 return ''.join(["%.2x" % int(x * 256) for x in hls_to_rgb(hue, li, sat)]) | |
| 53 | |
| 54 def graph_png(start, length): | |
| 55 rrds = all_sensors() | |
| 56 | |
| 57 graph_args = [] | |
| 367 | 58 have_volts = False |
| 335 | 59 for n, (rrdfile, sensor) in enumerate(rrds): |
| 367 | 60 if 'avrtemp' in sensor: |
| 61 continue | |
| 62 if 'voltage' in sensor: | |
| 63 have_volts = True | |
| 64 vname = 'scalevolts' | |
|
377
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
65 graph_args.append('DEF:%(vname)s=%(rrdfile)s:temp:AVERAGE:step=3600' % locals()) |
| 367 | 66 else: |
| 67 vname = 'temp%d' % n | |
|
377
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
68 graph_args.append('DEF:raw%(vname)s=%(rrdfile)s:temp:AVERAGE' % locals()) |
|
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
69 graph_args.append('CDEF:%(vname)s=raw%(vname)s,0.1,*,2,+' % locals()) |
| 335 | 70 width = config.LINE_WIDTH |
| 71 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
|
72 colour = config.SENSOR_COLOURS.get(legend, colour_from_string(sensor)) |
| 335 | 73 graph_args.append('LINE%(width)f:%(vname)s#%(colour)s:%(legend)s' % locals()) |
| 74 | |
| 373 | 75 end = int(start+length) |
| 76 start = int(start) | |
| 77 | |
| 335 | 78 tempf = tempfile.NamedTemporaryFile() |
| 373 | 79 dateformat = '%H:%M:%S %Y-%m-%d' |
| 80 watermark = ("Now %s\t" | |
| 81 "Start %s\t" | |
| 82 "End %s" % ( | |
| 83 datetime.datetime.now().strftime(dateformat), | |
| 84 datetime.datetime.fromtimestamp(start).strftime(dateformat), | |
| 85 datetime.datetime.fromtimestamp(end).strftime(dateformat) )) | |
| 86 | |
| 87 args = [tempf.name, '-s', str(start), | |
| 88 '-e', str(end), | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
89 '-w', str(config.GRAPH_WIDTH), |
|
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
90 '-h', str(config.GRAPH_HEIGHT), |
| 335 | 91 '--slope-mode', |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
92 '--border', '0', |
|
377
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
93 '--vertical-label', 'Voltage', |
|
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
94 '--y-grid', '0.1:1', |
|
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
95 '--dynamic-labels', |
| 362 | 96 '--grid-dash', '1:0', |
| 97 '--color', 'GRID#00000000', | |
| 98 '--color', 'MGRID#aaaaaa', | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
99 '--color', 'BACK#ffffff', |
| 373 | 100 '--disable-rrdtool-tag', |
| 101 '--watermark', watermark, | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
102 '--imgformat', 'PNG'] \ |
| 335 | 103 + graph_args |
| 373 | 104 args += ['--font', 'DEFAULT:12:%s' % config.GRAPH_FONT] |
| 105 args += ['--font', 'WATERMARK:10:%s' % config.GRAPH_FONT] | |
| 367 | 106 if have_volts: |
|
377
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
107 args += ['--right-axis', '10:-20', # matches the scalevolts CDEF above |
| 367 | 108 '--right-axis-format', '%.2lf', |
|
377
55710361804b
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
375
diff
changeset
|
109 '--right-axis-label', 'Temperature'] |
| 367 | 110 |
| 335 | 111 rrdtool.graph(*args) |
| 112 return tempf.read() | |
| 113 | |
| 333 | 114 def sensor_update(sensor_id, measurements, first_real_time, time_step): |
| 115 try: | |
| 116 open(sensor_rrd_path(sensor_id)) | |
| 117 except IOError, e: | |
| 118 create_rrd(sensor_id) | |
| 119 | |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
120 if measurements: |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
121 values = ['%d:%f' % p for p in |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
122 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
|
123 measurements)] |
| 333 | 124 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
125 rrdfile = sensor_rrd_path(sensor_id) |
| 346 | 126 # XXX what to do here when it fails... |
|
355
a1aa4176ca2c
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
351
diff
changeset
|
127 for v in values: |
|
a1aa4176ca2c
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
351
diff
changeset
|
128 try: |
|
a1aa4176ca2c
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
351
diff
changeset
|
129 rrdtool.update(rrdfile, v) |
| 369 | 130 except rrdtool.error, e: |
|
355
a1aa4176ca2c
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
351
diff
changeset
|
131 print>>sys.stderr, "Bad rrdtool update '%s'" % v |
|
a1aa4176ca2c
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
351
diff
changeset
|
132 traceback.print_exc(file=sys.stderr) |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
133 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
134 # be paranoid |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
135 f = file(rrdfile) |
|
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
136 os.fsync(f.fileno()) |
| 333 | 137 |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
138 def record_debug(lines): |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
139 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
|
140 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
|
141 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
|
142 f.flush() |
|
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
143 return f |
|
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
144 |
| 333 | 145 def parse(lines): |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
146 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
147 debugf = record_debug(lines) |
|
339
449272fc63a3
- Debug log file for server
Matt Johnston <matt@ucc.asn.au>
parents:
338
diff
changeset
|
148 |
| 333 | 149 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
|
150 if len(entries) != len(lines): |
| 333 | 151 raise Exception("Keys are not unique") |
| 152 | |
| 153 num_sensors = int(entries['sensors']) | |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
154 num_measurements = int(entries['measurements']) |
| 333 | 155 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
156 sensors = [entries['sensor_id%d' % n] for n in xrange(num_sensors)] |
| 333 | 157 |
| 158 meas = [] | |
| 159 for s in sensors: | |
| 160 meas.append([]) | |
| 161 | |
| 334 | 162 def val_scale(v): |
| 163 # convert decidegrees to degrees | |
| 164 return 0.1 * v | |
| 165 | |
| 333 | 166 for n in xrange(num_measurements): |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
167 vals = [val_scale(int(x)) for x in entries["meas%d" % n].strip().split()] |
| 333 | 168 if len(vals) != num_sensors: |
| 169 raise Exception("Wrong number of sensors for measurement %d" % n) | |
| 170 # we make an array of values for each sensor | |
| 171 for s in xrange(num_sensors): | |
| 172 meas[s].append(vals[s]) | |
| 173 | |
|
337
f575ef538f5d
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
335
diff
changeset
|
174 avr_now = float(entries['now']) |
| 333 | 175 avr_first_time = float(entries['first_time']) |
| 176 time_step = float(entries['time_step']) | |
| 177 | |
| 367 | 178 if 'avrtemp' in entries: |
| 179 avrtemp = val_scale(int(entries['avrtemp'])) | |
| 180 sensor_update('avrtemp', [avrtemp], time.time(), 1) | |
| 181 | |
| 182 if 'voltage' in entries: | |
| 183 voltage = 0.001 * int(entries['voltage']) | |
| 184 sensor_update('voltage', [voltage], time.time(), 1) | |
| 185 | |
| 346 | 186 #sqlite |
| 187 # - time | |
| 188 # - voltage | |
| 189 # - boot time | |
| 190 | |
| 333 | 191 first_real_time = time.time() - (avr_now - avr_first_time) |
| 192 | |
| 193 for sensor_id, measurements in zip(sensors, meas): | |
| 346 | 194 # XXX sqlite add |
| 333 | 195 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
|
196 |
|
344
ea1779d27641
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
339
diff
changeset
|
197 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
|
198 debugf.flush() |