Mercurial > code > home > repos > homeauto
annotate service/piNode/piNode.py @ 221:666e0e756ce6
piNode support for temp sensors. proper hostname lookup
Ignore-this: a68c3319bffb16c55cbb5f329118f0a4
| author | drewp@bigasterisk.com |
|---|---|
| date | Mon, 18 Jan 2016 22:43:22 -0800 |
| parents | fb23df31b642 |
| children | c5589f21d4a3 |
| rev | line source |
|---|---|
| 182 | 1 from __future__ import division |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
2 import sys, logging, socket, json, time |
| 182 | 3 import cyclone.web |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
4 from rdflib import Namespace, URIRef, Literal, Graph, RDF |
|
184
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
5 from rdflib.parser import StringInputSource |
| 182 | 6 from twisted.internet import reactor, task |
| 7 from docopt import docopt | |
| 8 logging.basicConfig(level=logging.DEBUG) | |
| 9 sys.path.append("/my/site/magma") | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
10 sys.path.append("../../../../site/magma") |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
11 |
| 182 | 12 from stategraph import StateGraph |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
13 sys.path.append('/opt/pigpio') |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
14 try: |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
15 import pigpio |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
16 except ImportError: |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
17 class pigpio(object): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
18 @staticmethod |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
19 def pi(): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
20 return None |
| 182 | 21 |
| 22 import devices | |
| 23 | |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
24 # from /my/proj/room |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
25 from carbondata import CarbonClient |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
26 |
| 182 | 27 log = logging.getLogger() |
| 28 logging.getLogger('serial').setLevel(logging.WARN) | |
| 29 ROOM = Namespace('http://projects.bigasterisk.com/room/') | |
| 30 HOST = Namespace('http://bigasterisk.com/ruler/host/') | |
| 31 | |
| 32 hostname = socket.gethostname() | |
| 33 | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
34 class Config(object): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
35 def __init__(self): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
36 self.graph = Graph() |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
37 log.info('read config') |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
38 self.graph.parse('config.n3', format='n3') |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
39 self.graph.bind('', ROOM) # maybe working |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
40 self.graph.bind('rdf', RDF) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
41 |
| 182 | 42 class GraphPage(cyclone.web.RequestHandler): |
| 43 def get(self): | |
| 44 g = StateGraph(ctx=ROOM['pi/%s' % hostname]) | |
| 45 | |
| 46 for stmt in self.settings.board.currentGraph(): | |
| 47 g.add(stmt) | |
| 48 | |
| 49 if self.get_argument('config', 'no') == 'yes': | |
| 50 for stmt in self.settings.config.graph: | |
| 51 g.add(stmt) | |
| 52 | |
| 53 self.set_header('Content-type', 'application/x-trig') | |
| 54 self.write(g.asTrig()) | |
| 55 | |
| 56 class Board(object): | |
| 57 """similar to arduinoNode.Board but without the communications stuff""" | |
| 58 def __init__(self, graph, uri, onChange): | |
| 59 self.graph, self.uri = graph, uri | |
| 60 self.pi = pigpio.pi() | |
| 61 self._devs = devices.makeDevices(graph, self.uri, self.pi) | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
62 log.debug('found %s devices', len(self._devs)) |
| 182 | 63 self._statementsFromInputs = {} # input device uri: latest statements |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
64 self._carbon = CarbonClient(serverHost='bang') |
| 182 | 65 |
| 66 def startPolling(self): | |
| 67 task.LoopingCall(self._poll).start(.5) | |
| 68 | |
| 69 def _poll(self): | |
| 70 for i in self._devs: | |
| 71 self._statementsFromInputs[i.uri] = i.poll() | |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
72 self._exportToGraphite() |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
73 |
| 182 | 74 def outputStatements(self, stmts): |
| 75 unused = set(stmts) | |
| 76 for dev in self._devs: | |
| 77 stmtsForDev = [] | |
| 78 for pat in dev.outputPatterns(): | |
| 79 if [term is None for term in pat] != [False, False, True]: | |
| 80 raise NotImplementedError | |
| 81 for stmt in stmts: | |
| 82 if stmt[:2] == pat[:2]: | |
| 83 stmtsForDev.append(stmt) | |
| 84 unused.discard(stmt) | |
| 85 if stmtsForDev: | |
| 86 log.info("output goes to action handler for %s" % dev.uri) | |
| 87 dev.sendOutput(stmtsForDev) | |
| 88 log.info("success") | |
| 89 if unused: | |
| 90 log.warn("No devices cared about these statements:") | |
| 91 for s in unused: | |
| 92 log.warn(repr(s)) | |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
93 |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
94 # needs merge with arduinoNode.py |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
95 def _exportToGraphite(self): |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
96 # note this is writing way too often- graphite is storing at a lower res |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
97 now = time.time() |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
98 # 20 sec is not precise; just trying to reduce wifi traffic |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
99 if getattr(self, 'lastGraphiteExport', 0) + 20 > now: |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
100 return |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
101 self.lastGraphiteExport = now |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
102 log.debug('graphite export:') |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
103 # objects of these statements are suitable as graphite values. |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
104 graphitePredicates = {ROOM['temperatureF']} |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
105 # bug: one sensor can have temp and humid- this will be ambiguous |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
106 for s, graphiteName in self.graph.subject_objects(ROOM['graphiteName']): |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
107 for group in self._statementsFromInputs.values(): |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
108 for stmt in group: |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
109 if stmt[0] == s and stmt[1] in graphitePredicates: |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
110 log.debug(' sending %s -> %s', stmt[0], graphiteName) |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
111 self._carbon.send(graphiteName, stmt[2].toPython(), now) |
| 182 | 112 |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
113 def currentGraph(self): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
114 g = Graph() |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
115 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
116 g.add((HOST[socket.gethostname()], ROOM['connectedTo'], self.uri)) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
117 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
118 for si in self._statementsFromInputs.values(): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
119 for s in si: |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
120 g.add(s) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
121 return g |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
122 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
123 def description(self): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
124 """for web page""" |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
125 return { |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
126 'uri': self.uri, |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
127 'devices': [d.description() for d in self._devs], |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
128 'graph': 'http://sticker:9059/graph', #todo |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
129 } |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
130 |
|
184
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
131 def rdfGraphBody(body, headers): |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
132 g = Graph() |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
133 g.parse(StringInputSource(body), format='nt') |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
134 return g |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
135 |
| 182 | 136 class OutputPage(cyclone.web.RequestHandler): |
| 137 def put(self): | |
|
184
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
138 arg = self.request.arguments |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
139 if arg.get('s') and arg.get('p'): |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
140 subj = URIRef(arg['s'][-1]) |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
141 pred = URIRef(arg['p'][-1]) |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
142 turtleLiteral = self.request.body |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
143 try: |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
144 obj = Literal(float(turtleLiteral)) |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
145 except ValueError: |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
146 obj = Literal(turtleLiteral) |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
147 stmt = (subj, pred, obj) |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
148 else: |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
149 g = rdfGraphBody(self.request.body, self.request.headers) |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
150 assert len(g) == 1, len(g) |
|
e052416a2290
piNode allow nt graphs as the body of a PUT /output
drewp@bigasterisk.com
parents:
183
diff
changeset
|
151 stmt = g.triples((None, None, None)).next() |
| 182 | 152 |
| 153 self.settings.board.outputStatements([stmt]) | |
| 154 | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
155 class Boards(cyclone.web.RequestHandler): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
156 def get(self): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
157 self.set_header('Content-type', 'application/json') |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
158 self.write(json.dumps({ |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
159 'boards': [self.settings.board.description()] |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
160 }, indent=2)) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
161 |
| 182 | 162 def main(): |
| 163 arg = docopt(""" | |
| 164 Usage: piNode.py [options] | |
| 165 | |
| 166 -v Verbose | |
| 167 """) | |
| 168 log.setLevel(logging.WARN) | |
| 169 if arg['-v']: | |
| 170 from twisted.python import log as twlog | |
| 171 twlog.startLogging(sys.stdout) | |
| 172 | |
| 173 log.setLevel(logging.DEBUG) | |
| 174 | |
| 175 config = Config() | |
| 176 | |
| 177 def onChange(): | |
| 178 # notify reasoning | |
| 179 pass | |
| 180 | |
|
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
181 thisHost = Literal(socket.gethostname()) |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
182 for row in config.graph.query( |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
183 'SELECT ?board WHERE { ?board a :PiBoard; :hostname ?h }', |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
184 initBindings=dict(h=thisHost)): |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
185 thisBoard = row.board |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
186 break |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
187 else: |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
188 raise ValueError("config had no board for :hostname %r" % thisHost) |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
189 |
|
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
190 log.info("found config for board %r" % thisBoard) |
| 182 | 191 board = Board(config.graph, thisBoard, onChange) |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
192 board.startPolling() |
| 182 | 193 |
| 194 reactor.listenTCP(9059, cyclone.web.Application([ | |
| 195 (r"/()", cyclone.web.StaticFileHandler, { | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
196 "path": "../arduinoNode/static", "default_filename": "index.html"}), |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
197 (r'/static/(.*)', cyclone.web.StaticFileHandler, {"path": "../arduinoNode/static"}), |
| 182 | 198 (r"/graph", GraphPage), |
| 199 (r'/output', OutputPage), | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
200 (r'/boards', Boards), |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
201 #(r'/dot', Dot), |
| 220 | 202 ], config=config, board=board, debug=arg['-v']), interface='::') |
| 182 | 203 reactor.run() |
| 204 | |
| 205 main() |