annotate service/rdf_to_mqtt/rdf_to_mqtt.py @ 1563:71eec31da919

more theater output controls Ignore-this: 34e105c3760b9df22d63eb0035aca19d darcs-hash:095f5932e595b89d909c85018999fa926503abcd
author drewp <drewp@bigasterisk.com>
date Fri, 14 Feb 2020 10:21:24 -0800
parents d786df082a73
children ffbebd7902ee
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
1563
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
17 from cycloneerr import PrettyErrorHandler
1539
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
18
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
19 ROOM = Namespace('http://projects.bigasterisk.com/room/')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
20
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
21 devs = {
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
22 ROOM['kitchenLight']: {
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
23 'root': 'h801_skylight',
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 },
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
28 ROOM['livingLampShelf']: {
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
29 'root': 'sonoff_0/switch/sonoff_basic_relay/command',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
30 'values': 'binary',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
31 },
1540
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
32 ROOM['livingLamp1']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
33 'root': 'sonoff_1/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
34 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
35 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
36 ROOM['livingLamp2']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
37 'root': 'sonoff_2/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
38 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
39 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
40 ROOM['livingLamp3']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
41 'root': 'sonoff_3/switch/sonoff_basic_relay/command',
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['livingLamp4']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
45 'root': 'sonoff_4/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
46 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
47 },
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
48 ROOM['livingLamp5']: {
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
49 'root': 'sonoff_5/switch/sonoff_basic_relay/command',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
50 'values': 'binary',
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
51 },
1563
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
52 ROOM['theater']: {
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
53 'root': 'theater_blaster/ir_out',
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
54 'values': 'theaterOutputs',
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
55 },
1540
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
56 #-t theater_blaster/ir_out -m 'input_game'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
57 #-t theater_blaster/ir_out -m 'input_bd'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
58 #-t theater_blaster/ir_out -m 'input_cbl'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
59 #-t theater_blaster/ir_out -m 'input_pc'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
60 #-t theater_blaster/ir_out/volume_up -m '{"times":1}'
0c8aa432ec9d more devices
drewp <drewp@bigasterisk.com>
parents: 1539
diff changeset
61 #-t theater_blaster/ir_out/volume_down -m '{"times":1}'
1539
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
62 }
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
63
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
64
1563
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
65 class OutputPage(PrettyErrorHandler, cyclone.web.RequestHandler):
1539
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
66 def put(self):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
67 for stmt in rdf_over_http.rdfStatementsFromRequest(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
68 self.request.arguments,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
69 self.request.body,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
70 self.request.headers):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
71 self._onStatement(stmt)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
72
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
73 def _onStatement(self, stmt):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
74 log.info(f'incoming statement: {stmt}')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
75 ignored = True
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
76 for dev, attrs in devs.items():
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
77 if stmt[0] == ROOM['frontWindow']:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
78 ignored = ignored and self._publishFrontScreenText(stmt)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
79 if stmt[0:2] == (dev, ROOM['brightness']):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
80 log.info(f'brightness request: {stmt}')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
81 brightness = stmt[2].toPython()
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 attrs.get('values', '') == 'binary':
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
84 self._publishOnOff(attrs, brightness)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
85 else:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
86 self._publishRgbw(attrs, brightness)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
87 ignored = False
1563
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
88 if stmt[0:2] == (dev, ROOM['inputSelector']):
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
89 self._publish(topic=attrs['root'],
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
90 message='input_'+str(stmt[2].toPython()))
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
91 ignored = False
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
92 if stmt[0:2] == (dev, ROOM['volumeChange']):
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
93 delta = int(stmt[2].toPython())
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
94 which = 'up' if delta > 0 else 'down'
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
95 self._publish(topic=f'theater_blaster/ir_out/volume_{which}',
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
96 message=json.dumps({'timed': abs(delta)}))
71eec31da919 more theater output controls
drewp <drewp@bigasterisk.com>
parents: 1541
diff changeset
97 ignored = False
1539
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
98 if ignored:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
99 log.warn("ignoring %s", stmt)
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 _publishOnOff(self, attrs, brightness):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
102 msg = 'OFF'
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
103 if brightness > 0:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
104 msg = 'ON'
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
105 self._publish(topic=attrs['root'], message=msg)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
106
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
107 def _publishRgbw(self, attrs, brightness):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
108 for chan, scale in [('w1', 1),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
109 ('r', 1),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
110 ('g', .8),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
111 ('b', .8)]:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
112 self._publish(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
113 topic=f"{attrs['root']}/light/kit_{chan}/command",
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
114 messageJson={
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
115 'state': 'ON',
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
116 'brightness': int(brightness * 255)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
117 })
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
118
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
119 def _publishFrontScreenText(self, stmt):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
120 ignored = True
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
121 for line in ['line1', 'line2', 'line3', 'line4']:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
122 if stmt[1] == ROOM[line]:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
123 ignored = False
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
124 self.settings.mqtt.publish(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
125 b'frontwindow/%s' % line.encode('ascii'),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
126 stmt[2].toPython())
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
127 return ignored
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
128
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
129 def _publish(self, topic: str, messageJson: object=None,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
130 message: str=None):
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
131 if messageJson is not None:
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
132 message = json.dumps(messageJson)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
133 self.settings.mqtt.publish(
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
134 topic.encode('ascii'),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
135 message.encode('ascii'))
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
136
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 if __name__ == '__main__':
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
139 arg = docopt("""
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
140 Usage: rdf_to_mqtt.py [options]
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 -v Verbose
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
143 """)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
144 verboseLogging(arg['-v'])
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
145
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
146 mqtt = MqttClient(clientId='rdf_to_mqtt', brokerPort=1883)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
147
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
148 port = 10008
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
149 reactor.listenTCP(port, cyclone.web.Application([
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
150 (r"/()", cyclone.web.StaticFileHandler,
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
151 {"path": ".", "default_filename": "index.html"}),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
152 (r'/output', OutputPage),
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
153 ], mqtt=mqtt, debug=arg['-v']), interface='::')
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
154 log.warn('serving on %s', port)
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
155
8c410b493da4 more renaming, build updates
drewp <drewp@bigasterisk.com>
parents:
diff changeset
156 reactor.run()