Mercurial > templog

annotate py/params.py @ 253:0a1b642e3086

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
long polling config updates
author Matt Johnston <matt@ucc.asn.au>
date Mon, 08 Jun 2015 22:29:46 +0800
parents 19569cb5ed46
children 26eee8591f61
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
1 # -*- coding: utf-8 -*-
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
2 import collections
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
3 import json
157
6d799bee3508 reload config on SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents: 155
diff changeset
4 import signal
230
185621f47040 run 2to3
Matt Johnston <matt@ucc.asn.au>
parents: 228
diff changeset
5 import io
253
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
6 import tempfile
157
6d799bee3508 reload config on SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents: 155
diff changeset
7
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
8 import config
148
b32e5a11a4cb few updates, seems to run
Matt Johnston <matt@ucc.asn.au>
parents: 145
diff changeset
9 from utils import W,L,E,EX
253
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
10 import utils
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
11
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
12 _FIELD_DEFAULTS = {
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
13 'fridge_setpoint': 16,
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
14 'fridge_difference': 0.2,
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
15 'overshoot_delay': 720, # 12 minutes
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
16 'overshoot_factor': 1, # ºC
162
d73077e8cd67 Add daemon mode with locking, add "disabled" parameter
Matt Johnston <matt@ucc.asn.au>
parents: 160
diff changeset
17 'disabled': False,
168
bfc3213edee4 add nowort mode
Matt Johnston <matt@ucc.asn.au>
parents: 166
diff changeset
18 'nowort': False,
170
78255c49bf9a make fridge air range a parameter
Matt Johnston <matt@ucc.asn.au>
parents: 168
diff changeset
19 'fridge_range_lower': 3,
78255c49bf9a make fridge air range a parameter
Matt Johnston <matt@ucc.asn.au>
parents: 168
diff changeset
20 'fridge_range_upper': 3,
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
21 }
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
22
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
23 class Params(dict):
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
24 class Error(Exception):
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
25 pass
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
26
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
27 def __init__(self):
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
28 self.update(_FIELD_DEFAULTS)
253
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
29 self._epoch = None
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
30
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
31 def __getattr__(self, k):
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
32 return self[k]
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
33
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
34 def __setattr__(self, k, v):
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
35 # fail if we set a bad value
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
36 self[k]
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
37 self[k] = v
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
38
233
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
39 def _do_load(self, f):
157
6d799bee3508 reload config on SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents: 155
diff changeset
40 try:
6d799bee3508 reload config on SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents: 155
diff changeset
41 u = json.load(f)
230
185621f47040 run 2to3
Matt Johnston <matt@ucc.asn.au>
parents: 228
diff changeset
42 except Exception as e:
157
6d799bee3508 reload config on SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents: 155
diff changeset
43 raise self.Error(e)
6d799bee3508 reload config on SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents: 155
diff changeset
44
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
45 for k in u:
166
5d5424acfed0 - use the fridge temperature for control too, keep it in a 6deg band
Matt Johnston <matt@ucc.asn.au>
parents: 162
diff changeset
46 if k.startswith('_'):
5d5424acfed0 - use the fridge temperature for control too, keep it in a 6deg band
Matt Johnston <matt@ucc.asn.au>
parents: 162
diff changeset
47 continue
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
48 if k not in self:
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
49 raise self.Error("Unknown parameter %s=%s in file '%s'" % (str(k), str(u[k]), getattr(f, 'name', '???')))
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
50 self.update(u)
253
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
51 self._epoch = utils.hexnonce()
145
6517ddee3187 few more bits
Matt Johnston <matt@ucc.asn.au>
parents:
diff changeset
52
160
256505f98c4d uploader works
Matt Johnston <matt@ucc.asn.au>
parents: 157
diff changeset
53 L("Loaded parameters")
256505f98c4d uploader works
Matt Johnston <matt@ucc.asn.au>
parents: 157
diff changeset
54 L(self.save_string())
256505f98c4d uploader works
Matt Johnston <matt@ucc.asn.au>
parents: 157
diff changeset
55
233
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
56 def load(self, f = None):
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
57 if f:
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
58 return self._do_load(f)
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
59 else:
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
60 with open(config.PARAMS_FILE, 'r') as f:
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
61 try:
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
62 return self._do_load(f)
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
63 except IOError as e:
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
64 W("Missing parameter file, using defaults. %s" % str(e))
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
65 return
160
256505f98c4d uploader works
Matt Johnston <matt@ucc.asn.au>
parents: 157
diff changeset
66
253
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
67 def get_epoch(self):
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
68 return self._epoch
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
69
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
70 def receive(self, params, epoch):
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
71 """ updates parameters from the server. does some validation,
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
72 writes config file to disk.
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
73 Returns True on success, False failure
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
74 """
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
75
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
76 if epoch != self._epoch:
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
77 return
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
78
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
79 def same_type(a, b):
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
80 ta = type(a)
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
81 tb = type(b)
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
82
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
83 if ta == int:
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
84 ta = float
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
85 if tb == int:
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
86 tb = float
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
87
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
88 return ta == tb
157
6d799bee3508 reload config on SIGHUP
Matt Johnston <matt@ucc.asn.au>
parents: 155
diff changeset
89
253
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
90 if self.keys() != new_params.keys():
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
91 diff = self.keys() ^ new_params.keys()
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
92 E("Mismatching params, %s" % str(diff))
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
93 return False
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
94
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
95 for k, v in new_params.items():
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
96 if not same_type(v, self[k]):
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
97 E("Bad type for %s" % k)
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
98 return False
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
99
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
100 dir = os.path.dirname(config.PARAMS_FILE)
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
101 try:
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
102 t = tempfile.NamedTemporaryFile(prefix='config',
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
103 dir = dir,
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
104 delete = False)
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
105
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
106 t.write(json.dumps(new_params, sort_keys=True, indent=4)+'\n')
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
107 name = t.name
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
108 t.close()
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
109
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
110 os.rename(name, config.PARAMS_FILE)
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
111 return True
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
112 except Exception as e:
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
113 E("Problem: %s" % e)
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
114 return False
233
19569cb5ed46 better arg parser. seems close to ready
Matt Johnston <matt@ucc.asn.au>
parents: 231
diff changeset
115
160
256505f98c4d uploader works
Matt Johnston <matt@ucc.asn.au>
parents: 157
diff changeset
116 def save_string(self):
253
0a1b642e3086 long polling config updates
Matt Johnston <matt@ucc.asn.au>
parents: 233
diff changeset
117 return json.dumps(self, sort_keys=True, indent=4)