Mercurial > templog
annotate web/log.py @ 114:b0ec7b783c0b
start of fridge bits
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Wed, 03 Oct 2012 12:04:36 +0000 |
| parents | 7f3fc0980df1 |
| children | 397ac5c079bf |
| rev | line source |
|---|---|
| 78 | 1 # -*- coding: utf-8 -*- |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
2 #:vim:et:ts=4:sts=4:sw=4: |
| 27 | 3 import rrdtool |
| 4 import os | |
| 29 | 5 import os.path |
| 27 | 6 import sys |
| 29 | 7 import glob |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
8 import hashlib |
|
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
9 import tempfile |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
10 import time |
|
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
11 import syslog |
| 44 | 12 import sqlite3 |
|
49
206294a354a7
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
45
diff
changeset
|
13 import traceback |
| 67 | 14 import datetime |
|
75
ca08442635ca
report raw ds18b20 values instead
Matt Johnston <matt@ucc.asn.au>
parents:
73
diff
changeset
|
15 import struct |
| 77 | 16 import binascii |
| 29 | 17 from colorsys import hls_to_rgb |
| 27 | 18 |
| 28 | 19 import config |
| 27 | 20 |
| 21 def sensor_rrd_path(s): | |
| 28 | 22 return '%s/sensor_%s.rrd' % (config.DATA_PATH, s) |
| 27 | 23 |
| 29 | 24 # returns (path, sensor_name) tuples |
| 25 def all_sensors(): | |
| 26 return [(r, os.path.basename(r[:-4])) | |
| 27 for r in glob.glob('%s/*.rrd' % config.DATA_PATH)] | |
| 28 | |
| 27 | 29 def create_rrd(sensor_id): |
|
43
ea99aae87884
Create with a start date so that it works
Matt Johnston <matt@ucc.asn.au>
parents:
40
diff
changeset
|
30 # start date of 10 seconds into 1970 is used so that we can |
|
ea99aae87884
Create with a start date so that it works
Matt Johnston <matt@ucc.asn.au>
parents:
40
diff
changeset
|
31 # update with prior values straight away. |
| 61 | 32 if 'voltage' in sensor_id: |
| 63 | 33 args = [ |
| 34 '--step', '3600', | |
| 35 'DS:temp:GAUGE:7200:1:10', | |
| 36 'RRA:AVERAGE:0.5:1:87600'] | |
| 61 | 37 else: |
| 63 | 38 args = [ |
| 39 '--step', '300', | |
|
88
6f4497a448e8
increase temperature logging range
Matt Johnston <matt@ucc.asn.au>
parents:
78
diff
changeset
|
40 'DS:temp:GAUGE:600:-100:500', |
| 61 | 41 'RRA:AVERAGE:0.5:1:1051200'] |
| 27 | 42 |
| 61 | 43 rrdtool.create(sensor_rrd_path(sensor_id), |
| 63 | 44 '--start', 'now-60d', |
| 61 | 45 *args) |
| 29 | 46 |
| 47 # stolen from viewmtn, stolen from monotone-viz | |
| 48 def colour_from_string(str): | |
| 49 def f(off): | |
| 50 return ord(hashval[off]) / 256.0 | |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
51 hashval = hashlib.sha1(str).digest() |
| 29 | 52 hue = f(5) |
| 53 li = f(1) * 0.15 + 0.55 | |
| 54 sat = f(2) * 0.5 + .5 | |
| 55 return ''.join(["%.2x" % int(x * 256) for x in hls_to_rgb(hue, li, sat)]) | |
| 56 | |
| 57 def graph_png(start, length): | |
|
104
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
58 os.environ['MATT_PNG_BODGE_COMPRESS'] = '4' |
|
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
59 os.environ['MATT_PNG_BODGE_FILTER'] = 'paeth' |
| 29 | 60 rrds = all_sensors() |
| 61 | |
| 62 graph_args = [] | |
| 61 | 63 have_volts = False |
| 29 | 64 for n, (rrdfile, sensor) in enumerate(rrds): |
| 61 | 65 if 'avrtemp' in sensor: |
| 66 continue | |
| 67 if 'voltage' in sensor: | |
| 68 have_volts = True | |
| 69 vname = 'scalevolts' | |
|
71
7d243ba2dd39
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
69
diff
changeset
|
70 graph_args.append('DEF:%(vname)s=%(rrdfile)s:temp:AVERAGE:step=3600' % locals()) |
| 78 | 71 unit = 'V' |
| 61 | 72 else: |
| 73 vname = 'temp%d' % n | |
|
71
7d243ba2dd39
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
69
diff
changeset
|
74 graph_args.append('DEF:raw%(vname)s=%(rrdfile)s:temp:AVERAGE' % locals()) |
| 101 | 75 # limit max temp to 50 |
| 76 graph_args.append('CDEF:%(vname)s=raw%(vname)s,35,GT,UNKN,raw%(vname)s,0.1,*,2,+,IF' % locals()) | |
| 78 | 77 unit = '<span face="Liberation Serif">º</span>C' |
| 78 | |
| 79 last_value = float(rrdtool.info(rrdfile)['ds[temp].last_ds']) | |
| 29 | 80 width = config.LINE_WIDTH |
| 81 legend = config.SENSOR_NAMES.get(sensor, sensor) | |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
82 colour = config.SENSOR_COLOURS.get(legend, colour_from_string(sensor)) |
| 78 | 83 format_last_value = ('%f' % last_value).rstrip('0') + unit |
| 84 print_legend = '%s (%s)' % (legend, format_last_value) | |
| 85 graph_args.append('LINE%(width)f:%(vname)s#%(colour)s:%(print_legend)s' % locals()) | |
| 29 | 86 |
| 67 | 87 end = int(start+length) |
| 88 start = int(start) | |
| 89 | |
| 29 | 90 tempf = tempfile.NamedTemporaryFile() |
| 67 | 91 dateformat = '%H:%M:%S %Y-%m-%d' |
| 92 watermark = ("Now %s\t" | |
| 93 "Start %s\t" | |
| 94 "End %s" % ( | |
| 95 datetime.datetime.now().strftime(dateformat), | |
| 96 datetime.datetime.fromtimestamp(start).strftime(dateformat), | |
| 97 datetime.datetime.fromtimestamp(end).strftime(dateformat) )) | |
| 98 | |
| 99 args = [tempf.name, '-s', str(start), | |
| 100 '-e', str(end), | |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
101 '-w', str(config.GRAPH_WIDTH), |
|
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
102 '-h', str(config.GRAPH_HEIGHT), |
| 29 | 103 '--slope-mode', |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
104 '--border', '0', |
|
73
0a8639039453
Get rid of axes labels, tidy html
Matt Johnston <matt@ucc.asn.au>
parents:
71
diff
changeset
|
105 # '--vertical-label', 'Voltage', |
|
71
7d243ba2dd39
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
69
diff
changeset
|
106 '--y-grid', '0.1:1', |
|
7d243ba2dd39
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
69
diff
changeset
|
107 '--dynamic-labels', |
| 56 | 108 '--grid-dash', '1:0', |
| 77 | 109 '--zoom', str(config.ZOOM), |
| 56 | 110 '--color', 'GRID#00000000', |
| 111 '--color', 'MGRID#aaaaaa', | |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
112 '--color', 'BACK#ffffff', |
| 67 | 113 '--disable-rrdtool-tag', |
| 78 | 114 '--pango-markup', |
| 67 | 115 '--watermark', watermark, |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
116 '--imgformat', 'PNG'] \ |
| 29 | 117 + graph_args |
| 67 | 118 args += ['--font', 'DEFAULT:12:%s' % config.GRAPH_FONT] |
| 119 args += ['--font', 'WATERMARK:10:%s' % config.GRAPH_FONT] | |
| 61 | 120 if have_volts: |
|
71
7d243ba2dd39
swap the left and right scales, kind of clunky
Matt Johnston <matt@ucc.asn.au>
parents:
69
diff
changeset
|
121 args += ['--right-axis', '10:-20', # matches the scalevolts CDEF above |
|
73
0a8639039453
Get rid of axes labels, tidy html
Matt Johnston <matt@ucc.asn.au>
parents:
71
diff
changeset
|
122 '--right-axis-format', '%.0lf', |
|
0a8639039453
Get rid of axes labels, tidy html
Matt Johnston <matt@ucc.asn.au>
parents:
71
diff
changeset
|
123 # '--right-axis-label', 'Temperature' |
|
0a8639039453
Get rid of axes labels, tidy html
Matt Johnston <matt@ucc.asn.au>
parents:
71
diff
changeset
|
124 ] |
| 61 | 125 |
|
104
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
126 print>>sys.stderr, ' '.join("'%s'" % s for s in args) |
| 29 | 127 rrdtool.graph(*args) |
|
104
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
128 #return tempf |
| 29 | 129 return tempf.read() |
| 130 | |
| 27 | 131 def sensor_update(sensor_id, measurements, first_real_time, time_step): |
| 132 try: | |
| 133 open(sensor_rrd_path(sensor_id)) | |
| 134 except IOError, e: | |
| 135 create_rrd(sensor_id) | |
| 136 | |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
137 if measurements: |
|
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
138 values = ['%d:%f' % p for p in |
|
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
139 zip((first_real_time + time_step*t for t in xrange(len(measurements))), |
|
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
140 measurements)] |
| 27 | 141 |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
142 rrdfile = sensor_rrd_path(sensor_id) |
| 40 | 143 # XXX what to do here when it fails... |
|
49
206294a354a7
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
45
diff
changeset
|
144 for v in values: |
|
206294a354a7
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
45
diff
changeset
|
145 try: |
|
206294a354a7
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
45
diff
changeset
|
146 rrdtool.update(rrdfile, v) |
| 63 | 147 except rrdtool.error, e: |
|
104
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
148 print>>sys.stderr, "Bad rrdtool update '%s': %s" % (v, str(e)) |
|
49
206294a354a7
update values one at a time, ignore failure
Matt Johnston <matt@ucc.asn.au>
parents:
45
diff
changeset
|
149 traceback.print_exc(file=sys.stderr) |
| 33 | 150 |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
151 # be paranoid |
|
104
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
152 #f = file(rrdfile) |
|
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
153 #os.fsync(f.fileno()) |
| 27 | 154 |
| 103 | 155 def debug_file(mode='r'): |
| 156 return open('%s/debug.log' % config.DATA_PATH, mode) | |
| 157 | |
| 33 | 158 def record_debug(lines): |
| 103 | 159 f = debug_file('a+') |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
160 f.write('===== %s =====\n' % time.strftime('%a, %d %b %Y %H:%M:%S')) |
| 33 | 161 f.writelines(('%s\n' % s for s in lines)) |
| 162 f.flush() | |
| 163 return f | |
| 164 | |
| 103 | 165 |
| 166 def tail_debug_log(): | |
| 167 f = debug_file() | |
| 168 f.seek(0, 2) | |
| 169 size = f.tell() | |
| 170 f.seek(max(0, size-30000)) | |
| 171 return '\n'.join(l.strip() for l in f.readlines()[-400:]) | |
| 172 | |
|
75
ca08442635ca
report raw ds18b20 values instead
Matt Johnston <matt@ucc.asn.au>
parents:
73
diff
changeset
|
173 def convert_ds18b20_12bit(reading): |
|
ca08442635ca
report raw ds18b20 values instead
Matt Johnston <matt@ucc.asn.au>
parents:
73
diff
changeset
|
174 value = struct.unpack('>h', binascii.unhexlify(reading))[0] |
|
ca08442635ca
report raw ds18b20 values instead
Matt Johnston <matt@ucc.asn.au>
parents:
73
diff
changeset
|
175 return value * 0.0625 |
|
ca08442635ca
report raw ds18b20 values instead
Matt Johnston <matt@ucc.asn.au>
parents:
73
diff
changeset
|
176 |
| 94 | 177 def time_rem(name, entries): |
| 178 val_ticks = int(entries[name]) | |
| 95 | 179 val_rem = int(entries['%s_rem' % name]) |
|
99
1a88bb989afb
fix off-by-one in remainder code
Matt Johnston <matt@ucc.asn.au>
parents:
95
diff
changeset
|
180 tick_wake = int(entries['tick_wake']) + 1 |
| 94 | 181 tick_secs = int(entries['tick_secs']) |
| 182 return val_ticks + float(val_rem) * tick_secs / tick_wake | |
| 183 | |
| 27 | 184 def parse(lines): |
|
104
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
185 |
|
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
186 start_time = time.time() |
| 33 | 187 |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
188 debugf = record_debug(lines) |
| 33 | 189 |
| 27 | 190 entries = dict(l.split('=', 1) for l in lines) |
|
31
5e75e08d20ac
- Various fixes for web server, kind of works
Matt Johnston <matt@ucc.asn.au>
parents:
29
diff
changeset
|
191 if len(entries) != len(lines): |
| 27 | 192 raise Exception("Keys are not unique") |
| 193 | |
| 194 num_sensors = int(entries['sensors']) | |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
195 num_measurements = int(entries['measurements']) |
| 27 | 196 |
|
37
8da0fdadc8d7
- Getting there, update has problems
Matt Johnston <matt@ucc.asn.au>
parents:
33
diff
changeset
|
197 sensors = [entries['sensor_id%d' % n] for n in xrange(num_sensors)] |
| 27 | 198 |
| 199 meas = [] | |
| 200 for s in sensors: | |
| 201 meas.append([]) | |
| 202 | |
| 203 for n in xrange(num_measurements): | |
| 77 | 204 vals = [convert_ds18b20_12bit(x) for x in entries["meas%d" % n].strip().split()] |
| 27 | 205 if len(vals) != num_sensors: |
| 206 raise Exception("Wrong number of sensors for measurement %d" % n) | |
| 207 # we make an array of values for each sensor | |
| 208 for s in xrange(num_sensors): | |
| 209 meas[s].append(vals[s]) | |
| 210 | |
| 94 | 211 avr_now = time_rem('now', entries) |
| 212 avr_first_time = time_rem('first_time', entries) | |
| 213 avr_comms_time = time_rem('comms_time', entries) | |
| 27 | 214 time_step = float(entries['time_step']) |
| 215 | |
| 94 | 216 debugf.write('now %f, comms_time %f, first_time %f, delta %f\n' % |
| 217 (avr_now, avr_comms_time, avr_first_time, avr_now - avr_comms_time)) | |
| 218 | |
| 61 | 219 if 'avrtemp' in entries: |
| 220 avrtemp = val_scale(int(entries['avrtemp'])) | |
| 221 sensor_update('avrtemp', [avrtemp], time.time(), 1) | |
| 222 | |
| 223 if 'voltage' in entries: | |
| 224 voltage = 0.001 * int(entries['voltage']) | |
| 225 sensor_update('voltage', [voltage], time.time(), 1) | |
| 226 | |
| 40 | 227 #sqlite |
| 228 # - time | |
| 229 # - voltage | |
| 230 # - boot time | |
| 231 | |
| 27 | 232 first_real_time = time.time() - (avr_now - avr_first_time) |
| 233 | |
| 234 for sensor_id, measurements in zip(sensors, meas): | |
| 40 | 235 # XXX sqlite add |
| 27 | 236 sensor_update(sensor_id, measurements, first_real_time, time_step) |
| 33 | 237 |
|
104
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
238 timedelta = time.time() - start_time |
|
7f3fc0980df1
- few more web tweaks. don't fsync, it's slow.
Matt Johnston <matt@ucc.asn.au>
parents:
103
diff
changeset
|
239 debugf.write("Updated %d sensors in %.2f secs\n" % (len(sensors), timedelta)) |
| 33 | 240 debugf.flush() |