Mercurial > ~darius > hgwebdir.cgi > beermon
annotate beermon.py @ 5:8d471840b153
Support overshooting on heating/cooling.
| author | darius |
|---|---|
| date | Sat, 29 Sep 2007 02:23:24 +0000 |
| parents | 32a56dd33e42 |
| children | 45d9895a5020 |
| rev | line source |
|---|---|
| 0 | 1 #!/usr/bin/env python |
| 2 | |
| 1 | 3 ############################################################################ |
| 4 # Monitor & control fermenter temperature | |
| 5 # v1.0 | |
| 6 # | |
| 5 | 7 # $Id: beermon.py,v 1.6 2007/09/29 02:23:24 darius Exp $ |
| 1 | 8 # |
| 9 # Depends on: Python 2.3 (I think) | |
| 10 # | |
| 11 ############################################################################ | |
| 12 # | |
| 13 # Copyright (C) 2007 Daniel O'Connor. All rights reserved. | |
| 14 # | |
| 15 # Redistribution and use in source and binary forms, with or without | |
| 16 # modification, are permitted provided that the following conditions | |
| 17 # are met: | |
| 18 # 1. Redistributions of source code must retain the above copyright | |
| 19 # notice, this list of conditions and the following disclaimer. | |
| 20 # 2. Redistributions in binary form must reproduce the above copyright | |
| 21 # notice, this list of conditions and the following disclaimer in the | |
| 22 # documentation and/or other materials provided with the distribution. | |
| 23 # | |
| 24 # THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND | |
| 25 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| 26 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
| 27 # ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE | |
| 28 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
| 29 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |
| 30 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
| 31 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |
| 32 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |
| 33 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |
| 34 # SUCH DAMAGE. | |
| 35 # | |
| 36 ############################################################################ | |
| 37 | |
| 38 | |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
39 import pexpect, re, threading, time, logging, sys, traceback |
| 0 | 40 from logging.handlers import RotatingFileHandler |
| 41 | |
| 42 class ROMReadError(Exception): | |
| 43 pass | |
| 44 | |
| 4 | 45 class ThreadDied(Exception): |
| 46 pass | |
| 47 | |
| 0 | 48 class Control(): |
| 49 targetTemp = 18 | |
| 4 | 50 hysteresis = 0.5 |
| 0 | 51 pollInterval = 30 |
| 4 | 52 staleDataTime = 30 |
| 0 | 53 |
| 1 | 54 def __init__(self, m, _log): |
| 0 | 55 self.m = m |
| 56 global log | |
| 1 | 57 log = _log |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
58 self.cv = threading.Condition() |
|
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
59 self.cv.acquire() |
| 1 | 60 |
| 0 | 61 def doit(self): |
| 62 log.debug("target temperature - %3.2f" % (self.targetTemp)) | |
| 63 log.debug("fermenterId - %s" % (self.m.fermenterId)) | |
| 64 log.debug("fridgeId - %s" % (self.m.fridgeId)) | |
| 65 log.debug("ambientId - %s" % (self.m.ambientId)) | |
| 66 log.debug("minCoolOnTime - %d, minCoolOffTime - %d" % (self.m.minCoolOnTime, self.m.minCoolOffTime)) | |
| 67 log.debug("minHeatOnTime - %d, minHeatOffTime - %d" % (self.m.minHeatOnTime, self.m.minHeatOffTime)) | |
| 68 log.debug("pollInterval - %d" % (self.pollInterval)) | |
| 69 | |
| 70 log.debug("=== Starting ===") | |
| 71 log.debug("Fermenter Fridge Ambient State New State") | |
| 72 while True: | |
| 4 | 73 # Check if our monitor thread has died |
| 74 if (not self.m.isAlive()): | |
| 75 raise ThreadDied, "Monitor thread has died" | |
| 76 | |
| 77 # Check for stale data | |
| 78 if (self.m.lastUpdate[self.m.fermenterId] + self.staleDataTime < time.time()): | |
| 79 log.debug("Stale data") | |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
80 self.cv.wait(self.pollInterval) |
| 0 | 81 self.m.setState('idle') |
| 82 continue | |
| 83 | |
| 4 | 84 # Work out what state we should go into |
| 0 | 85 nextState = "-" |
| 5 | 86 # Temperature diff, -ve => too cold, +ve => too warm |
| 0 | 87 diff = self.m.temps[self.m.fermenterId] - self.targetTemp |
| 88 if (self.m.currState == 'idle'): | |
| 89 # If we're idle then only heat or cool if the temperate difference is out of the | |
| 4 | 90 # hysteresis band |
| 0 | 91 if (abs(diff) > self.hysteresis): |
| 92 if (diff < 0 and self.m.minHeatOffTime + self.m.lastHeatOff < time.time()): | |
| 93 nextState = 'heat' | |
| 94 elif (diff > 0 and self.m.minHeatOffTime + self.m.lastHeatOff < time.time()): | |
| 95 nextState = 'cool' | |
| 96 elif (self.m.currState == 'cool'): | |
| 5 | 97 # Work out if we should go idle (based on min on time & overshoot) |
| 98 if (diff + self.m.minCoolOvershoot < 0 and self.m.minCoolOnTime + self.m.lastCoolOn < time.time()): | |
| 0 | 99 nextState = 'idle' |
| 100 elif (self.m.currState == 'heat'): | |
| 4 | 101 # Ditto |
| 5 | 102 if (diff - self.m.minHeatOvershoot > 0 and self.m.minHeatOnTime + self.m.lastHeatOn < time.time()): |
| 0 | 103 nextState = 'idle' |
| 104 else: | |
| 4 | 105 # Not possible.. |
| 0 | 106 raise KeyError |
| 107 | |
| 108 log.debug("%3.2f %3.2f %3.2f %s %s" % | |
| 109 (self.m.temps[self.m.fermenterId], | |
| 110 self.m.temps[self.m.fridgeId], self.m.temps[self.m.ambientId], | |
| 111 self.m.currState, nextState)) | |
| 4 | 112 |
| 0 | 113 if (nextState != "-"): |
| 114 self.m.setState(nextState) | |
| 4 | 115 |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
116 self.cv.wait(self.pollInterval) |
| 0 | 117 |
| 118 | |
| 119 class MonitorDev(threading.Thread): | |
| 120 # Match a ROM ID (eg 00:11:22:33:44:55:66:77) | |
| 121 romre = re.compile('([0-9a-f]{2}:){7}[0-9a-f]{2}') | |
| 122 # Match the prompt | |
| 123 promptre = re.compile('> ') | |
| 124 | |
| 125 coolRelay = 7 | |
| 126 heatRelay = 6 | |
| 127 | |
| 128 fermenterId = '10:eb:48:21:01:08:00:df' | |
| 129 fridgeId = '10:a6:2a:c4:00:08:00:11' | |
| 130 ambientId = '10:97:1b:fe:00:08:00:d1' | |
| 131 | |
| 132 # minimum time the cooler must spend on/off | |
| 133 minCoolOnTime = 10 * 60 | |
| 134 minCoolOffTime = 10 * 60 | |
| 135 | |
| 136 # minimum time the heater must spend on/off | |
| 137 minHeatOnTime = 60 | |
| 138 minHeatOffTime = 60 | |
| 139 | |
| 5 | 140 # minimum to overshoot on heating/cooling |
| 141 minHeatOvershoot = 1 | |
| 142 minCoolOvershoot = 0 | |
| 143 | |
| 4 | 144 # Dictionary of sensor IDs & temperatures |
| 0 | 145 temps = {} |
| 4 | 146 # Dictionary of sensor IDs & epoch times |
| 147 lastUpdate = {} | |
| 0 | 148 |
| 4 | 149 # List of all device IDs |
| 150 devs = [] | |
| 151 # List of temperature sensor IDs | |
| 152 tempdevs = [] | |
| 153 | |
| 0 | 154 # Lock to gate access to the comms |
| 155 commsLock = None | |
| 156 | |
| 157 currState = 'idle' | |
| 158 | |
| 159 lastHeatOn = 0 | |
| 160 lastHeatOff = 0 | |
| 161 lastCoolOn = 0 | |
| 162 lastCoolOff = 0 | |
| 163 | |
| 164 def __init__(self): | |
| 165 threading.Thread.__init__(self) | |
| 166 self.commsLock = threading.Lock() | |
| 167 self.p = pexpect.spawn('/usr/bin/ssh', ['-xt', '-enone', '-i', '/home/darius/.ssh/id_wrt', 'root@wrt', '(echo logged in; microcom -D/dev/cua/1)']) | |
| 168 assert(self.p.expect('logged in') == 0) | |
| 169 self.p.timeout = 3 | |
| 170 self.setspeed() | |
| 171 self.devs = self.find1wire() | |
| 4 | 172 self.tempdevs = filter(self.istemp, self.devs) |
| 0 | 173 |
| 174 self.start() | |
| 175 | |
| 176 def setspeed(self): | |
| 177 self.commsLock.acquire() | |
| 178 self.p.send('~') | |
| 179 assert(self.p.expect('t - set terminal') == 0) | |
| 180 self.p.send('t') | |
| 181 assert(self.p.expect('p - set speed') == 0) | |
| 182 self.p.send('p') | |
| 183 assert(self.p.expect('f - 38400') == 0) | |
| 184 self.p.send('f') | |
| 185 assert(self.p.expect('done!') == 0) | |
| 186 self.commsLock.release() | |
| 187 | |
| 188 def find1wire(self): | |
| 189 self.commsLock.acquire() | |
| 190 self.p.sendline('') | |
| 191 assert(self.p.expect('> ') == 0) | |
| 192 self.p.sendline('sr') | |
| 193 # Echo | |
| 194 assert(self.p.expect('sr') == 0) | |
| 195 | |
| 196 # Send a new line which will give us a command prompt to stop on | |
| 197 # later. We could use read() but that would make the code a lot | |
| 198 # uglier | |
| 199 self.p.sendline('') | |
| 200 | |
| 201 devlist = [] | |
| 202 | |
| 203 # Loop until we get the command prompt (> ) collecting ROM IDs | |
| 204 while True: | |
| 205 idx = self.p.expect([self.romre, self.promptre]) | |
| 206 if (idx == 0): | |
| 207 # Matched a ROM | |
| 208 #print "Found ROM " + self.p.match.group() | |
| 209 devlist.append(self.p.match.group(0)) | |
| 210 elif (idx == 1): | |
| 211 # Matched prompt, exit | |
| 212 break | |
| 213 else: | |
| 214 # Unpossible! | |
| 215 self.commsLock.release() | |
| 216 raise SystemError() | |
| 217 | |
| 218 self.commsLock.release() | |
| 219 | |
| 220 return(devlist) | |
| 221 | |
| 222 def istemp(self, id): | |
| 223 [family, a, b, c, d, e, f, g] = id.split(':') | |
| 224 if (family == '10'): | |
| 225 return True | |
| 226 else: | |
| 227 return False | |
| 228 | |
| 229 def updateTemps(self): | |
| 4 | 230 for i in self.tempdevs: |
| 231 try: | |
| 232 self.temps[i] = float(self.readTemp(i)) | |
| 233 self.lastUpdate[i] = time.time() | |
| 234 except ROMReadError: | |
| 235 # Ignore this - just results in no update reflected by lastUpdate | |
| 236 pass | |
| 237 | |
| 0 | 238 return(self.temps) |
| 239 | |
| 240 def readTemp(self, id): | |
| 241 self.commsLock.acquire() | |
| 242 cmd = 'te ' + id | |
| 243 self.p.sendline(cmd) | |
| 244 # Echo | |
| 245 assert(self.p.expect(cmd) == 0) | |
| 246 # Eat EOL left from expect | |
| 247 self.p.readline() | |
| 248 | |
| 249 line = self.p.readline().strip() | |
| 250 self.commsLock.release() | |
| 251 # 'CRC mismatch' can mean that we picked the wrong ROM.. | |
| 252 if (re.match('CRC mismatch', line) != None): | |
| 253 raise ROMReadError | |
| 254 | |
| 255 return(line) | |
| 256 | |
| 257 def setState(self, state): | |
| 258 if (state == 'cool'): | |
| 259 relay = 1 << self.coolRelay | |
| 260 elif (state == 'heat'): | |
| 261 relay = 1 << self.heatRelay | |
| 262 elif (state == 'idle'): | |
| 263 relay = 0 | |
| 264 else: | |
| 265 raise(ValueError) | |
| 266 | |
| 267 if (state == self.currState): | |
| 268 return | |
| 269 | |
| 270 # Keep track of when we last turned off or on | |
| 271 if (state == 'cool'): | |
| 272 if (self.currState == 'heat'): | |
| 273 self.lastHeatOff = time.time() | |
| 274 self.lastCoolOn = time.time() | |
| 275 elif (state == 'heat'): | |
| 276 if (self.currState == 'cool'): | |
| 277 self.lastCoolOff = time.time() | |
| 278 self.lastHeatOn = time.time() | |
| 279 else: | |
| 280 if (self.currState == 'cool'): | |
| 281 self.lastCoolOff = time.time() | |
| 282 if (self.currState == 'heat'): | |
| 283 self.lastHeatOff = time.time() | |
| 284 | |
| 285 self.currState = state | |
| 286 | |
| 287 self.commsLock.acquire() | |
| 288 # Need the extra spaces cause the parser in the micro is busted | |
| 289 cmd = 'out c %02x' % relay | |
| 290 self.p.sendline(cmd) | |
| 291 # Echo | |
| 292 assert(self.p.expect(cmd) == 0) | |
| 293 self.commsLock.release() | |
| 294 | |
| 295 def polltemps(self, temps): | |
| 296 while True: | |
| 297 for d in temps: | |
| 298 #print d | |
| 299 t = gettemp(p, d) | |
| 300 print "%s -> %s" % (d, t) | |
| 301 print | |
| 302 | |
| 303 def run(self): | |
| 304 while True: | |
| 305 self.updateTemps() | |
| 306 | |
| 1 | 307 def initLog(): |
| 308 # Init our logging | |
| 309 log = logging.getLogger("monitor") | |
| 310 | |
| 311 # Default to warts and all logging | |
| 312 log.setLevel(logging.DEBUG) | |
| 313 | |
| 314 # Log to this file | |
| 315 logfile = logging.handlers.RotatingFileHandler(filename = "/tmp/beermon.log", | |
| 316 maxBytes = 1000000, backupCount = 3) | |
| 317 | |
| 318 # And stderr | |
| 319 logstderr = logging.StreamHandler() | |
| 320 | |
| 321 # Format it nicely | |
| 322 formatter = logging.Formatter(fmt = "%(asctime)s: %(message)s", datefmt = "%Y/%m/%d %H:%M:%S") | |
| 323 | |
| 324 # Glue it all together | |
| 325 logfile.setFormatter(formatter) | |
| 326 logstderr.setFormatter(formatter) | |
| 327 log.addHandler(logfile) | |
| 328 log.addHandler(logstderr) | |
| 329 return(log) | |
| 330 | |
| 0 | 331 def main(): |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
332 import beermon |
| 1 | 333 |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
334 global log |
|
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
335 log = initLog() |
| 1 | 336 |
| 337 log.debug("=== Initing ===") | |
| 5 | 338 log.debug("$Id: beermon.py,v 1.6 2007/09/29 02:23:24 darius Exp $") |
| 0 | 339 |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
340 m = None |
|
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
341 exitCode = 0 |
| 0 | 342 try: |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
343 m = beermon.MonitorDev() |
|
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
344 |
| 1 | 345 c = beermon.Control(m, log) |
| 0 | 346 # Wait for the first temperature readings to come through, saves |
| 347 # getting an 'invalid data' message | |
| 348 time.sleep(3) | |
| 349 c.doit() | |
| 350 log.debug("doit exited") | |
| 351 | |
| 352 except KeyboardInterrupt: | |
| 353 log.debug("Exiting due to keyboard interrupt") | |
| 1 | 354 |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
355 except Exception, e: |
| 3 | 356 log.debug("Something went wrong, details below:") |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
357 log.debug(e) |
| 3 | 358 log.debug(reduce(lambda x, y: x + y, traceback.format_exception( |
| 359 sys.exc_type, sys.exc_value, sys.exc_traceback))) | |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
360 exitCode = 1 |
| 0 | 361 |
| 362 finally: | |
| 363 # Make sure we try and turn it off if something goes wrong | |
|
2
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
364 if (m != None): |
|
17bc5535bbb7
Wait on a condition var instead of using time.sleep() - this works
darius
parents:
1
diff
changeset
|
365 m.setState('idle') |
| 0 | 366 |
| 1 | 367 sys.exit(exitCode) |
| 0 | 368 |
| 369 if __name__ == "__main__": | |
| 370 main() |