Mercurial > code > home > repos > homeauto

annotate service/mqtt_graph_bridge/mqtt_graph_bridge.py @ 1197:d8acab2b01f5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
mqtt has two devices now. various older cleanups. Ignore-this: 67ca3acc5dc6aa672d0c896c9f5ae48e darcs-hash:964511475873219e6bc4725388a8d2e54a3c4e28
author drewp <drewp@bigasterisk.com>
date Sat, 19 Jan 2019 12:08:59 -0800
parents 6561367aa60a
children 7051b8b4766a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
1 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
2 from patchablegraph import PatchableGraph, CycloneGraphHandler, CycloneGraphEventsHandler
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
3 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
4 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
5 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
6 import cyclone.web
1197
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
7 import sys, logging, json
1183
6561367aa60a factor common mqtt code out of mqtt_graph_bridge
drewp <drewp@bigasterisk.com>
parents: 1178
diff changeset
8 from mqtt_client import MqttClient
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
9
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
10 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
11
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
12 devs = {
1197
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
13 ROOM['kitchenLight']: {'root': 'h801_skylight', 'ctx': ROOM['kitchenH801']},
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
14 ROOM['kitchenCounterLight']: {'root': 'h801_counter', 'ctx': ROOM['kitchenH801']},
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
15 }
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
16
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
17 logging.basicConfig()
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
18 log = logging.getLogger()
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
19
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
20 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
21 g = Graph()
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
22 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
23 return g
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 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
26 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
27 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
28 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
29 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
30 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
31 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
32 try:
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
33 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
34 except ValueError:
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
35 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
36 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
37 else:
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
38 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
39 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
40 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
41 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
42
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
43 def _onStatement(self, stmt):
1197
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
44 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
45 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
46 if stmt[0:2] == (dev, ROOM['brightness']):
1197
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
47 self.settings.mqtt.publish("%s/light/kit_w1/command" % attrs['root'],
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
48 json.dumps({'state': 'ON', 'brightness': int(stmt[2].toPython() * 255)}))
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
49 self.settings.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
50 stmt[0], stmt[1], stmt[2])
1197
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
51 ignored = False
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
52 if ignored:
d8acab2b01f5 mqtt has two devices now. various older cleanups.
drewp <drewp@bigasterisk.com>
parents: 1183
diff changeset
53 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
54
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
55 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
56 arg = docopt("""
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
57 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
58
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
59 -v Verbose
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
60 """)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
61 log.setLevel(logging.WARN)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
62 if arg['-v']:
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
63 from twisted.python import log as twlog
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
64 twlog.startLogging(sys.stdout)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
65 log.setLevel(logging.DEBUG)
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
66
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
67 masterGraph = PatchableGraph()
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
68
1183
6561367aa60a factor common mqtt code out of mqtt_graph_bridge
drewp <drewp@bigasterisk.com>
parents: 1178
diff changeset
69 mqtt = MqttClient(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
70
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
71 port = 10008
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
72 reactor.listenTCP(port, cyclone.web.Application([
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
73 (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
74 (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
75 {'masterGraph': masterGraph}),
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
76 (r'/output', OutputPage),
1183
6561367aa60a factor common mqtt code out of mqtt_graph_bridge
drewp <drewp@bigasterisk.com>
parents: 1178
diff changeset
77 ], 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
78 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
79
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
80 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
81 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
82 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
83
1183
6561367aa60a factor common mqtt code out of mqtt_graph_bridge
drewp <drewp@bigasterisk.com>
parents: 1178
diff changeset
84
1178
e3991af5bd39 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp <drewp@bigasterisk.com>
parents:
diff changeset
85 reactor.run()