annotate service/rdf_to_mqtt/rdf_to_mqtt.py @ 1541:d786df082a73

stop updating a graph (mqtt is our output) Ignore-this: 764732cb27bde75d7425a4691a4faaf0 darcs-hash:e172f313c0671c368d3e60b2d5d92a75885e105d
author drewp <drewp@bigasterisk.com>
date Mon, 10 Feb 2020 23:58:57 -0800
parents 0c8aa432ec9d
children 71eec31da919
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1539
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
1 """
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
2 We get output statements that are like light9's deviceAttrs (:dev1 :color "#ff0000"),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
3 convert those to outputAttrs (:dev1 :red 255; :green 0; :blue 0) and post them to mqtt.
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
4
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
5 This is like light9/bin/collector.
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
6 """
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
7 import json
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
8 from mqtt_client import MqttClient
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
9 from docopt import docopt
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
10 from rdflib import Namespace, Literal
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
11 from twisted.internet import reactor
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
12 import cyclone.web
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
13
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
14 from patchablegraph import PatchableGraph, CycloneGraphHandler, CycloneGraphEventsHandler
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
15 from standardservice.logsetup import log, verboseLogging
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
16 import rdf_over_http
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
17
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
18 ROOM = Namespace('http://projects.bigasterisk.com/room/')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
19
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
20 devs = {
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
21 ROOM['kitchenLight']: {
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
22 'root': 'h801_skylight',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
23 'ctx': ROOM['kitchenH801']
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
24 },
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
25 ROOM['kitchenCounterLight']: {
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
26 'root': 'h801_counter',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
27 'ctx': ROOM['kitchenH801']
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
28 },
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
29 ROOM['livingLampShelf']: {
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
30 'root': 'sonoff_0/switch/sonoff_basic_relay/command',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
31 'ctx': ROOM['sonoff_0'],
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
32 'values': 'binary',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
33 },
1540
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
34 ROOM['livingLamp1']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
35 'root': 'sonoff_1/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
36 'ctx': ROOM['sonoff_1'],
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
37 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
38 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
39 ROOM['livingLamp2']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
40 'root': 'sonoff_2/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
41 'ctx': ROOM['sonoff_2'],
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
42 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
43 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
44 ROOM['livingLamp3']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
45 'root': 'sonoff_3/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
46 'ctx': ROOM['sonoff_3'],
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
47 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
48 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
49 ROOM['livingLamp4']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
50 'root': 'sonoff_4/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
51 'ctx': ROOM['sonoff_4'],
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
52 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
53 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
54 ROOM['livingLamp5']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
55 'root': 'sonoff_5/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
56 'ctx': ROOM['sonoff_5'],
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
57 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
58 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
59 #-t theater_blaster/ir_out -m 'input_game'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
60 #-t theater_blaster/ir_out -m 'input_bd'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
61 #-t theater_blaster/ir_out -m 'input_cbl'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
62 #-t theater_blaster/ir_out -m 'input_pc'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
63 #-t theater_blaster/ir_out/volume_up -m '{"times":1}'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
64 #-t theater_blaster/ir_out/volume_down -m '{"times":1}'
1539
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
65 }
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
66
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
67
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
68 class OutputPage(cyclone.web.RequestHandler):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
69 def put(self):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
70 for stmt in rdf_over_http.rdfStatementsFromRequest(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
71 self.request.arguments,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
72 self.request.body,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
73 self.request.headers):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
74 self._onStatement(stmt)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
75
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
76 def _onStatement(self, stmt):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
77 log.info(f'incoming statement: {stmt}')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
78 ignored = True
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
79 for dev, attrs in devs.items():
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
80 if stmt[0] == ROOM['frontWindow']:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
81 ignored = ignored and self._publishFrontScreenText(stmt)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
82
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
83 if stmt[0:2] == (dev, ROOM['brightness']):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
84 log.info(f'brightness request: {stmt}')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
85 brightness = stmt[2].toPython()
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
86
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
87 if attrs.get('values', '') == 'binary':
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
88 self._publishOnOff(attrs, brightness)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
89 else:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
90 self._publishRgbw(attrs, brightness)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
91 ignored = False
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
92 if ignored:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
93 log.warn("ignoring %s", stmt)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
94
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
95 def _publishOnOff(self, attrs, brightness):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
96 msg = 'OFF'
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
97 if brightness > 0:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
98 msg = 'ON'
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
99 self._publish(topic=attrs['root'], message=msg)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
100
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
101 def _publishRgbw(self, attrs, brightness):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
102 for chan, scale in [('w1', 1),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
103 ('r', 1),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
104 ('g', .8),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
105 ('b', .8)]:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
106 self._publish(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
107 topic=f"{attrs['root']}/light/kit_{chan}/command",
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
108 messageJson={
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
109 'state': 'ON',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
110 'brightness': int(brightness * 255)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
111 })
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
112
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
113 def _publishFrontScreenText(self, stmt):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
114 ignored = True
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
115 for line in ['line1', 'line2', 'line3', 'line4']:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
116 if stmt[1] == ROOM[line]:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
117 ignored = False
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
118 self.settings.mqtt.publish(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
119 b'frontwindow/%s' % line.encode('ascii'),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
120 stmt[2].toPython())
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
121 return ignored
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
122
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
123 def _publish(self, topic: str, messageJson: object=None,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
124 message: str=None):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
125 if messageJson is not None:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
126 message = json.dumps(messageJson)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
127 self.settings.mqtt.publish(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
128 topic.encode('ascii'),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
129 message.encode('ascii'))
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
130
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
131
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
132 if __name__ == '__main__':
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
133 arg = docopt("""
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
134 Usage: rdf_to_mqtt.py [options]
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
135
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
136 -v Verbose
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
137 """)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
138 verboseLogging(arg['-v'])
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
139
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
140 mqtt = MqttClient(clientId='rdf_to_mqtt', brokerPort=1883)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
141
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
142 port = 10008
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
143 reactor.listenTCP(port, cyclone.web.Application([
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
144 (r"/()", cyclone.web.StaticFileHandler,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
145 {"path": ".", "default_filename": "index.html"}),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
146 (r'/output', OutputPage),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
147 ], mqtt=mqtt, debug=arg['-v']), interface='::')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
148 log.warn('serving on %s', port)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
149
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
150 reactor.run()