Mercurial > templog
annotate py/tempserver.py @ 231:e39ed85d87a5
fix a few python3 issues
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Sat, 11 Apr 2015 00:16:05 +0800 |
| parents | 185621f47040 |
| children | a01b7bccccd3 |
| rev | line source |
|---|---|
| 148 | 1 #!/home/matt/templog/venv/bin/python |
| 141 | 2 |
| 3 import sys | |
| 4 import os | |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
5 import logging |
| 198 | 6 import time |
| 7 import signal | |
| 228 | 8 import asyncio |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
9 |
| 172 | 10 import lockfile.pidlockfile |
|
162
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
11 import daemon |
| 141 | 12 |
| 13 import utils | |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
14 from utils import L,D,EX,W |
| 145 | 15 import fridge |
| 16 import config | |
| 231 | 17 import sensor |
| 145 | 18 import params |
| 160 | 19 import uploader |
| 145 | 20 |
| 141 | 21 |
| 22 class Tempserver(object): | |
| 23 def __init__(self): | |
| 24 self.readings = [] | |
| 145 | 25 self.current = (None, None) |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
26 self.fridge = None |
| 228 | 27 self._wakeup = asyncio.Event() |
| 141 | 28 |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
29 def __enter__(self): |
| 145 | 30 self.params = params.Params() |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
31 self.fridge = fridge.Fridge(self) |
| 160 | 32 self.uploader = uploader.Uploader(self) |
| 145 | 33 self.params.load() |
| 229 | 34 self.set_sensors(sensor.make_sensor(self)) |
| 228 | 35 asyncio.get_event_loop().add_signal_handler(signal.SIGHUP, self._reload_signal) |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
36 return self |
| 141 | 37 |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
38 def __exit__(self, exc_type, exc_value, traceback): |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
39 L("Exiting, cleanup handler"); |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
40 self.fridge.off() |
| 141 | 41 |
| 145 | 42 def run(self): |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
43 |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
44 if self.fridge is None: |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
45 raise Exception("Tempserver.run() must be within 'with Tempserver() as server'") |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
46 |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
47 # XXX do these go here or in __enter_() ? |
|
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
48 self.start_time = self.now() |
| 228 | 49 tasks = ( |
| 50 self.fridge.run(), | |
| 51 self.sensors.run(), | |
| 52 self.uploader.run(), | |
| 53 ) | |
| 54 loop = asyncio.get_event_loop() | |
| 55 try: | |
| 56 loop.run_until_complete(asyncio.wait(tasks)) | |
| 57 # not reached | |
| 58 except KeyboardInterrupt: | |
| 59 pass | |
| 145 | 60 |
| 61 def now(self): | |
| 62 return utils.monotonic_time() | |
| 63 | |
| 144 | 64 def set_sensors(self, sensors): |
| 148 | 65 if hasattr(self, 'sensors'): |
| 144 | 66 self.sensors.kill() |
| 67 self.sensors = sensors | |
| 68 self.wort_name = sensors.wort_name() | |
| 69 self.fridge_name = sensors.fridge_name() | |
| 141 | 70 |
| 71 def take_readings(self): | |
| 72 ret = self.readings | |
| 73 self.readings = [] | |
| 74 return ret | |
| 75 | |
| 76 def pushfront(self, readings): | |
| 77 """ used if a caller of take_readings() fails """ | |
| 160 | 78 self.readings = readings + self.readings |
| 141 | 79 |
| 144 | 80 # a reading is a map of {sensorname: value}. temperatures |
| 81 # are float degrees | |
| 141 | 82 def add_reading(self, reading): |
| 83 """ adds a reading at the current time """ | |
| 155 | 84 D("add_reading(%s)" % str(reading)) |
| 145 | 85 self.readings.append( (reading, self.now())) |
| 144 | 86 self.current = (reading.get(self.wort_name, None), |
| 87 reading.get(self.fridge_name, None)) | |
| 160 | 88 if len(self.readings) > config.MAX_READINGS: |
| 89 self.readings = self.readings[-config.MAX_READINGS:] | |
| 141 | 90 |
| 91 def current_temps(self): | |
| 144 | 92 """ returns (wort_temp, fridge_temp) tuple """ |
| 148 | 93 return self.current |
| 145 | 94 |
|
219
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
95 def sleep(self, timeout): |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
96 """ sleeps for timeout seconds, though wakes if the server's config is updated """ |
| 228 | 97 asyncio.wait_for(self._wakeup, timeout=timeout) |
|
219
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
98 |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
99 def _reload_signal(self): |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
100 try: |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
101 self.params.load() |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
102 L("Reloaded.") |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
103 self._wakeup.set() |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
104 self._wakeup.clear() |
| 229 | 105 except Error as e: |
|
219
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
106 W("Problem reloading: %s" % str(e)) |
|
16a83e2c97a0
sleep on a semaphore so it can start/stop immediately when there's a SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents:
217
diff
changeset
|
107 |
| 145 | 108 def setup_logging(): |
| 109 logging.basicConfig(format='%(asctime)s %(message)s', | |
| 110 datefmt='%m/%d/%Y %I:%M:%S %p', | |
| 163 | 111 level=logging.INFO) |
| 145 | 112 |
|
162
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
113 def start(): |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
114 with Tempserver() as server: |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
115 server.run() |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
116 |
| 145 | 117 def main(): |
|
149
d686b111dab4
working better. logging works properly, cleanup fridge.off() happens.
Matt Johnston <matt@ucc.asn.au>
parents:
148
diff
changeset
|
118 setup_logging() |
| 145 | 119 |
|
174
c49d87bb81b9
fix to absolute path for lockfile, --daemon does chdir("/")
Matt Johnston <matt@ucc.asn.au>
parents:
172
diff
changeset
|
120 heredir = os.path.abspath(os.path.dirname(__file__)) |
|
c49d87bb81b9
fix to absolute path for lockfile, --daemon does chdir("/")
Matt Johnston <matt@ucc.asn.au>
parents:
172
diff
changeset
|
121 pidpath = os.path.join(heredir, 'tempserver.pid') |
| 172 | 122 pidf = lockfile.pidlockfile.PIDLockFile(pidpath, threaded=False) |
| 199 | 123 do_hup = '--hup' in sys.argv |
| 172 | 124 try: |
| 217 | 125 pidf.acquire(1) |
| 172 | 126 pidf.release() |
| 229 | 127 except (lockfile.AlreadyLocked, lockfile.LockTimeout) as e: |
| 172 | 128 pid = pidf.read_pid() |
| 199 | 129 if do_hup: |
| 130 try: | |
| 131 os.kill(pid, signal.SIGHUP) | |
| 230 | 132 print("Sent SIGHUP to process %d" % pid, file=sys.stderr) |
| 199 | 133 sys.exit(0) |
| 134 except OSError: | |
| 230 | 135 print("Process %d isn't running?" % pid, file=sys.stderr) |
| 199 | 136 sys.exit(1) |
| 137 | |
| 230 | 138 print("Locked by PID %d" % pid, file=sys.stderr) |
| 199 | 139 |
| 198 | 140 stale = False |
| 172 | 141 if pid > 0: |
| 198 | 142 if '--new' in sys.argv: |
| 143 try: | |
| 144 os.kill(pid, 0) | |
| 145 except OSError: | |
| 146 stale = True | |
| 147 | |
| 148 if not stale: | |
| 230 | 149 print("Stopping old tempserver pid %d" % pid, file=sys.stderr) |
| 198 | 150 os.kill(pid, signal.SIGTERM) |
| 151 time.sleep(2) | |
| 152 pidf.acquire(0) | |
| 153 pidf.release() | |
| 154 else: | |
| 155 try: | |
| 156 os.kill(pid, 0) | |
| 157 # must still be running PID | |
| 158 raise e | |
| 159 except OSError: | |
| 160 stale = True | |
| 161 | |
| 162 if stale: | |
| 163 # isn't still running, steal the lock | |
| 230 | 164 print("Unlinking stale lockfile %s for pid %d" % (pidpath, pid), file=sys.stderr) |
| 198 | 165 pidf.break_lock() |
|
162
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
166 |
| 199 | 167 if do_hup: |
| 230 | 168 print("Doesn't seem to be running", file=sys.stderr) |
| 199 | 169 sys.exit(1) |
| 170 | |
| 145 | 171 if '--daemon' in sys.argv: |
|
162
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
172 logpath = os.path.join(os.path.dirname(__file__), 'tempserver.log') |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
173 logf = open(logpath, 'a+') |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
174 with daemon.DaemonContext(pidfile=pidf, stdout=logf, stderr = logf): |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
175 start() |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
176 else: |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
177 with pidf: |
|
d73077e8cd67
Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents:
160
diff
changeset
|
178 start() |
| 145 | 179 |
| 180 if __name__ == '__main__': | |
| 181 main() |