Mercurial > templog
annotate py/sensor_ds18b20.py @ 228:d9e81a563923
porting to asyncio
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Fri, 20 Mar 2015 20:12:25 +0800 |
| parents | 16a83e2c97a0 |
| children | 185621f47040 |
| rev | line source |
|---|---|
| 141 | 1 #!/usr/bin/env python2.7 |
| 2 | |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
3 import os |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
4 import re |
| 228 | 5 import asyncio |
| 6 import concurrent.futures | |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
7 |
| 144 | 8 import config |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
9 from utils import D,L,W,E,EX |
| 144 | 10 |
| 228 | 11 class DS18B20s(object): |
| 144 | 12 |
| 13 THERM_RE = re.compile('.* YES\n.*t=(.*)\n', re.MULTILINE) | |
| 141 | 14 |
| 144 | 15 def __init__(self, server): |
| 16 self.server = server | |
| 228 | 17 self.readthread = concurrent.futures.ThreadPoolExecutor(max_workers=1) |
| 148 | 18 self.master_dir = config.SENSOR_BASE_DIR |
| 144 | 19 |
| 228 | 20 @asyncio.coroutine |
|
151
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
21 def do(self): |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
22 vals = {} |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
23 for n in self.sensor_names(): |
| 228 | 24 value = yield from self.do_sensor(n) |
|
151
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
25 if value is not None: |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
26 vals[n] = value |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
27 |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
28 itemp = self.do_internal() |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
29 if itemp: |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
30 vals['internal'] = itemp |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
31 |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
32 self.server.add_reading(vals) |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
33 |
| 228 | 34 @asyncio.coroutine |
| 35 def run(self): | |
| 144 | 36 while True: |
| 228 | 37 yield from self.do() |
| 38 yield from self.server.sleep(config.SENSOR_SLEEP) | |
| 144 | 39 |
| 228 | 40 @asyncio.coroutine |
| 148 | 41 def read_wait(self, f): |
| 228 | 42 # handles a blocking file read with a threadpool. A |
| 43 # real python thread performs the read while other | |
| 44 # asyncio tasks keep running. | |
| 148 | 45 # the ds18b20 takes ~750ms to read, which is noticable |
| 46 # interactively. | |
| 228 | 47 loop = asyncio.get_event_loop() |
| 48 yield from loop.run_in_executor(self.readthread, f.read) | |
| 141 | 49 |
| 228 | 50 @asyncio.coroutine |
| 148 | 51 def do_sensor(self, s, contents = None): |
| 52 """ contents can be set by the caller for testing """ | |
| 144 | 53 try: |
| 54 if contents is None: | |
| 148 | 55 fn = os.path.join(self.master_dir, s, 'w1_slave') |
| 144 | 56 f = open(fn, 'r') |
| 228 | 57 contents = yield from self.read_wait(f) |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
58 |
| 144 | 59 match = self.THERM_RE.match(contents) |
| 60 if match is None: | |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
61 D("no match") |
| 144 | 62 return None |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
63 temp = int(match.groups(1)[0]) / 1000.0 |
| 175 | 64 if temp > 80: |
| 65 E("Problem reading sensor '%s': %f" % (s, temp)) | |
| 66 return None | |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
67 return temp |
| 144 | 68 except Exception, e: |
| 69 EX("Problem reading sensor '%s': %s" % (s, str(e))) | |
| 70 return None | |
| 141 | 71 |
|
151
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
72 def do_internal(self): |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
73 try: |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
74 return int(open(config.INTERNAL_TEMPERATURE, 'r').read()) / 1000.0 |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
75 except Exception, e: |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
76 EX("Problem reading internal sensor: %s" % str(e)) |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
77 return None |
|
e114b38c8a55
internal temperature sensor
Matt Johnston <matt@ucc.asn.au>
parents:
149
diff
changeset
|
78 |
| 144 | 79 |
| 80 def sensor_names(self): | |
| 81 """ Returns a sequence of sensorname """ | |
| 148 | 82 slaves_path = os.path.join(self.master_dir, "w1_master_slaves") |
| 178 | 83 contents = open(slaves_path, 'r').read() |
| 84 if 'not found' in contents: | |
| 85 E("No W1 sensors found") | |
| 86 return [] | |
| 87 names = contents.split() | |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
88 return names |
| 141 | 89 |
| 90 def wort_name(self): | |
| 144 | 91 return config.WORT_NAME |
| 141 | 92 |
| 93 def fridge_name(self): | |
| 144 | 94 return config.FRIDGE_NAME |