Mercurial > code > home > repos > homeauto

annotate service/mqtt_graph_bridge/mqtt_graph_bridge.py @ 1389:928b1833de0f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
fix str/bytes to mqtt client Ignore-this: 6840d55dfdf0408cee1b4fb39a4b3c7c darcs-hash:cb29ba8c66c2736c730a5ba8f9fa0258e0f0e98b
author drewp <drewp@bigasterisk.com>
date Fri, 10 May 2019 02:28:14 -0700
parents a29a55f3429c
children e1ee6661329a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1384
a29a55f3429c mqtt_graph_bridge to new build rules and to py3
drewp <drewp@bigasterisk.com>
parents: 1263
diff changeset
1 import json
a29a55f3429c mqtt_graph_bridge to new build rules and to py3
drewp <drewp@bigasterisk.com>
parents: 1263
diff changeset
2
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
3 from docopt import docopt
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
4 from rdflib import Namespace, URIRef, Literal, Graph
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
5 from rdflib.parser import StringInputSource
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
6 from twisted.internet import reactor
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
7 import cyclone.web
1384
a29a55f3429c mqtt_graph_bridge to new build rules and to py3
drewp <drewp@bigasterisk.com>
parents: 1263
diff changeset
8
1183
6561367aa60a factor common mqtt code out of mqtt_graph_bridge
drewp <drewp@bigasterisk.com>
parents: 1178
diff changeset
9 from mqtt_client import MqttClient
1384
a29a55f3429c mqtt_graph_bridge to new build rules and to py3
drewp <drewp@bigasterisk.com>
parents: 1263
diff changeset
10 from patchablegraph import PatchableGraph, CycloneGraphHandler, CycloneGraphEventsHandler
a29a55f3429c mqtt_graph_bridge to new build rules and to py3
drewp <drewp@bigasterisk.com>
parents: 1263
diff changeset
11 from standardservice.logsetup import log, verboseLogging
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
12
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
13 ROOM = Namespace('http://projects.bigasterisk.com/room/')
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
14
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
15 devs = {
1263
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
16 ROOM['kitchenLight']: {
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
17 'root': 'h801_skylight',
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
18 'ctx': ROOM['kitchenH801']
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
19 },
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
20 ROOM['kitchenCounterLight']: {
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
21 'root': 'h801_counter',
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
22 'ctx': ROOM['kitchenH801']
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
23 },
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
24 }
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
25
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
26 def rdfGraphBody(body, headers):
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
27 g = Graph()
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
28 g.parse(StringInputSource(body), format='nt')
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
29 return g
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
30
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
31 class OutputPage(cyclone.web.RequestHandler):
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
32 def put(self):
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
33 arg = self.request.arguments
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
34 if arg.get('s') and arg.get('p'):
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
35 subj = URIRef(arg['s'][-1])
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
36 pred = URIRef(arg['p'][-1])
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
37 turtleLiteral = self.request.body
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
38 try:
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
39 obj = Literal(float(turtleLiteral))
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
40 except ValueError:
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
41 obj = Literal(turtleLiteral)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
42 stmt = (subj, pred, obj)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
43 else:
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
44 g = rdfGraphBody(self.request.body, self.request.headers)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
45 assert len(g) == 1, len(g)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
46 stmt = g.triples((None, None, None)).next()
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
47 self._onStatement(stmt)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
48
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
49 def _onStatement(self, stmt):
1197
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
50 ignored = True
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
51 for dev, attrs in devs.items():
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
52 if stmt[0:2] == (dev, ROOM['brightness']):
1263
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
53 for chan, scale in [('w1', 1),
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
54 ('r', 1),
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
55 ('g', .8),
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
56 ('b', .8)]:
1389
928b1833de0f fix str/bytes to mqtt client
drewp <drewp@bigasterisk.com>
parents: 1384
diff changeset
57 out = stmt[2].toPython() * scale
928b1833de0f fix str/bytes to mqtt client
drewp <drewp@bigasterisk.com>
parents: 1384
diff changeset
58 topic = f"{attrs['root']}/light/kit_{chan}/command"
1263
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
59 self.settings.mqtt.publish(
1389
928b1833de0f fix str/bytes to mqtt client
drewp <drewp@bigasterisk.com>
parents: 1384
diff changeset
60 topic.encode('ascii'),
1263
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
61 json.dumps({
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
62 'state': 'ON',
1389
928b1833de0f fix str/bytes to mqtt client
drewp <drewp@bigasterisk.com>
parents: 1384
diff changeset
63 'brightness': int(out * 255)}).encode('ascii'))
1263
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
64 self.settings.masterGraph.patchObject(
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
65 attrs['ctx'],
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
66 stmt[0], stmt[1], stmt[2])
1197
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
67 ignored = False
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
68 if ignored:
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
69 log.warn("ignoring %s", stmt)
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
70
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
71 if __name__ == '__main__':
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
72 arg = docopt("""
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
73 Usage: mqtt_graph_bridge.py [options]
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
74
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
75 -v Verbose
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
76 """)
1384
a29a55f3429c mqtt_graph_bridge to new build rules and to py3
drewp <drewp@bigasterisk.com>
parents: 1263
diff changeset
77 verboseLogging(arg['-v'])
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
78
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
79 masterGraph = PatchableGraph()
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
80
1389
928b1833de0f fix str/bytes to mqtt client
drewp <drewp@bigasterisk.com>
parents: 1384
diff changeset
81 mqtt = MqttClient(clientId='mqtt_graph_bridge', brokerPort=1883)
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
82
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
83 port = 10008
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
84 reactor.listenTCP(port, cyclone.web.Application([
1263
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
85 (r"/()", cyclone.web.StaticFileHandler,
82fe44eadf00 build updates. hack in r/g/b and some fixed multipliers
drewp <drewp@bigasterisk.com>
parents: 1197
diff changeset
86 {"path": ".", "default_filename": "index.html"}),
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
87 (r"/graph", CycloneGraphHandler, {'masterGraph': masterGraph}),
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
88 (r"/graph/events", CycloneGraphEventsHandler,
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
89 {'masterGraph': masterGraph}),
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
90 (r'/output', OutputPage),
1183
6561367aa60a factor common mqtt code out of mqtt_graph_bridge
drewp <drewp@bigasterisk.com>
parents: 1178
diff changeset
91 ], mqtt=mqtt, masterGraph=masterGraph, debug=arg['-v']), interface='::')
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
92 log.warn('serving on %s', port)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
93
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
94 for dev, attrs in devs.items():
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
95 masterGraph.patchObject(attrs['ctx'],
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
96 dev, ROOM['brightness'], Literal(0.0))
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
97
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
98 reactor.run()