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