Mercurial > code > home > repos > homeauto

annotate service/mqtt_graph_bridge/mqtt_graph_bridge.py @ 722:a93fbf0d0daa

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
dep updates; graph url renames; and other build updates Ignore-this: 4603ef3d8db650a13e543dad8580ade8
author drewp@bigasterisk.com
date Wed, 05 Feb 2020 00:23:06 -0800
parents c52b172c0824
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
597
e1ee6661329a adjust kitchen PWM freqs. add comments and proposed contents of n3 configs
drewp@bigasterisk.com
parents: 586
diff changeset
1 """
e1ee6661329a adjust kitchen PWM freqs. add comments and proposed contents of n3 configs
drewp@bigasterisk.com
parents: 586
diff changeset
2 We get output statements that are like light9's deviceAttrs (:dev1 :color "#ff0000"),
e1ee6661329a adjust kitchen PWM freqs. add comments and proposed contents of n3 configs
drewp@bigasterisk.com
parents: 586
diff changeset
3 convert those to outputAttrs (:dev1 :red 255; :green 0; :blue 0) and post them to mqtt.
e1ee6661329a adjust kitchen PWM freqs. add comments and proposed contents of n3 configs
drewp@bigasterisk.com
parents: 586
diff changeset
4
e1ee6661329a adjust kitchen PWM freqs. add comments and proposed contents of n3 configs
drewp@bigasterisk.com
parents: 586
diff changeset
5 This is like light9/bin/collector.
e1ee6661329a adjust kitchen PWM freqs. add comments and proposed contents of n3 configs
drewp@bigasterisk.com
parents: 586
diff changeset
6 """
581
30022797642e mqtt_graph_bridge to new build rules and to py3
drewp@bigasterisk.com
parents: 460
diff changeset
7 import json
30022797642e mqtt_graph_bridge to new build rules and to py3
drewp@bigasterisk.com
parents: 460
diff changeset
8
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
9 from docopt import docopt
694
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
10 from rdflib import Namespace, Literal
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
11 from twisted.internet import reactor
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
12 import cyclone.web
581
30022797642e mqtt_graph_bridge to new build rules and to py3
drewp@bigasterisk.com
parents: 460
diff changeset
13
378
b90d9321d2ce factor common mqtt code out of mqtt_graph_bridge
drewp@bigasterisk.com
parents: 373
diff changeset
14 from mqtt_client import MqttClient
581
30022797642e mqtt_graph_bridge to new build rules and to py3
drewp@bigasterisk.com
parents: 460
diff changeset
15 from patchablegraph import PatchableGraph, CycloneGraphHandler, CycloneGraphEventsHandler
30022797642e mqtt_graph_bridge to new build rules and to py3
drewp@bigasterisk.com
parents: 460
diff changeset
16 from standardservice.logsetup import log, verboseLogging
694
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
17 import rdf_over_http
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
18
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
19 ROOM = Namespace('http://projects.bigasterisk.com/room/')
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
20
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
21 devs = {
460
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
22 ROOM['kitchenLight']: {
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
23 'root': 'h801_skylight',
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
24 'ctx': ROOM['kitchenH801']
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
25 },
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
26 ROOM['kitchenCounterLight']: {
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
27 'root': 'h801_counter',
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
28 'ctx': ROOM['kitchenH801']
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
29 },
696
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
30 ROOM['livingLampShelf']: {
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
31 'root': 'sonoff_0/switch/sonoff_basic_relay/command',
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
32 'ctx': ROOM['sonoff_0'],
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
33 'values': 'binary',
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
34 },
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
35 }
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
36
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
37
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
38 class OutputPage(cyclone.web.RequestHandler):
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
39 def put(self):
694
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
40 for stmt in rdf_over_http.rdfStatementsFromRequest(
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
41 self.request.arguments,
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
42 self.request.body,
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
43 self.request.headers):
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
44 self._onStatement(stmt)
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
45
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
46 def _onStatement(self, stmt):
694
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
47 log.info(f'incoming statement: {stmt}')
392
79d041273e26 mqtt has two devices now. various older cleanups.
drewp@bigasterisk.com
parents: 378
diff changeset
48 ignored = True
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
49 for dev, attrs in devs.items():
696
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
50 if stmt[0] == ROOM['frontWindow']:
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
51 ignored = ignored and self._publishFrontScreenText(stmt)
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
52
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
53 if stmt[0:2] == (dev, ROOM['brightness']):
696
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
54 log.info(f'brightness request: {stmt}')
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
55 brightness = stmt[2].toPython()
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
56
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
57 if attrs.get('values', '') == 'binary':
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
58 self._publishOnOff(attrs, brightness)
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
59 else:
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
60 self._publishRgbw(attrs, brightness)
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
61 # try to stop saving this; let the device be the master usually
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
62 self.settings.masterGraph.patchObject(
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
63 attrs['ctx'],
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
64 stmt[0], stmt[1], stmt[2])
392
79d041273e26 mqtt has two devices now. various older cleanups.
drewp@bigasterisk.com
parents: 378
diff changeset
65 ignored = False
79d041273e26 mqtt has two devices now. various older cleanups.
drewp@bigasterisk.com
parents: 378
diff changeset
66 if ignored:
79d041273e26 mqtt has two devices now. various older cleanups.
drewp@bigasterisk.com
parents: 378
diff changeset
67 log.warn("ignoring %s", stmt)
696
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
68
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
69 def _publishOnOff(self, attrs, brightness):
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
70 msg = 'OFF'
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
71 if brightness > 0:
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
72 msg = 'ON'
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
73 self._publish(topic=attrs['root'], message=msg)
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
74
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
75 def _publishRgbw(self, attrs, brightness):
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
76 for chan, scale in [('w1', 1),
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
77 ('r', 1),
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
78 ('g', .8),
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
79 ('b', .8)]:
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
80 self._publish(
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
81 topic=f"{attrs['root']}/light/kit_{chan}/command",
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
82 messageJson={
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
83 'state': 'ON',
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
84 'brightness': int(brightness * 255)
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
85 })
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
86
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
87 def _publishFrontScreenText(self, stmt):
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
88 ignored = True
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
89 for line in ['line1', 'line2', 'line3', 'line4']:
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
90 if stmt[1] == ROOM[line]:
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
91 ignored = False
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
92 self.settings.mqtt.publish(
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
93 b'frontwindow/%s' % line.encode('ascii'),
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
94 stmt[2].toPython())
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
95 return ignored
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
96
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
97 def _publish(self, topic: str, messageJson: object=None,
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
98 message: str=None):
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
99 if messageJson is not None:
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
100 message = json.dumps(messageJson)
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
101 self.settings.mqtt.publish(
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
102 topic.encode('ascii'),
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
103 message.encode('ascii'))
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
104
c52b172c0824 add publish to ON/OFF messages. split up the main statement handler
drewp@bigasterisk.com
parents: 694
diff changeset
105
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
106 if __name__ == '__main__':
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
107 arg = docopt("""
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
108 Usage: mqtt_graph_bridge.py [options]
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
109
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
110 -v Verbose
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
111 """)
581
30022797642e mqtt_graph_bridge to new build rules and to py3
drewp@bigasterisk.com
parents: 460
diff changeset
112 verboseLogging(arg['-v'])
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
113
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
114 masterGraph = PatchableGraph()
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
115
586
b112da45e035 fix str/bytes to mqtt client
drewp@bigasterisk.com
parents: 581
diff changeset
116 mqtt = MqttClient(clientId='mqtt_graph_bridge', brokerPort=1883)
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
117
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
118 port = 10008
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
119 reactor.listenTCP(port, cyclone.web.Application([
460
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
120 (r"/()", cyclone.web.StaticFileHandler,
7051b8b4766a build updates. hack in r/g/b and some fixed multipliers
drewp@bigasterisk.com
parents: 392
diff changeset
121 {"path": ".", "default_filename": "index.html"}),
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
122 (r'/output', OutputPage),
722
a93fbf0d0daa dep updates; graph url renames; and other build updates
drewp@bigasterisk.com
parents: 696
diff changeset
123 ], mqtt=mqtt, debug=arg['-v']), interface='::')
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
124 log.warn('serving on %s', port)
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
125
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
126 for dev, attrs in devs.items():
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
127 masterGraph.patchObject(attrs['ctx'],
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
128 dev, ROOM['brightness'], Literal(0.0))
694
925bc4137c93 extract rdfStatementsFromRequest for sharing with other tools
drewp@bigasterisk.com
parents: 597
diff changeset
129
373
2158e7ad19b1 receive oneshot updates from reasoning; emit commands on MQTT to control H801 wifi dimmer
drewp@bigasterisk.com
parents:
diff changeset
130 reactor.run()