Mercurial > code > home > repos > homeauto

annotate service/frontDoorLock/front_door_lock.py @ 1184:c87e5c8eb8ab

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
frontdoor autolock. logging improvements. use simpler mqtt interface. Ignore-this: e2bf5262a89ebb898108a634679fdec7 darcs-hash:4864716cd06f14c5ad5ef7e0e7bc51458603012b
author drewp <drewp@bigasterisk.com>
date Wed, 12 Dec 2018 01:11:54 -0800
parents bd215f18e715
children d495d4382a07
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
1 from docopt import docopt
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
2 from patchablegraph import PatchableGraph, CycloneGraphHandler, CycloneGraphEventsHandler
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
3 from rdflib import Namespace, URIRef, Literal, Graph
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
4 from rdflib.parser import StringInputSource
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
5 from twisted.internet import reactor, task
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
6 import cyclone.web
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
7 import sys, logging, time
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
8 from mqtt_client import MqttClient
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
9 from logsetup import log, enableTwistedLog
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
10
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
11 ROOM = Namespace('http://projects.bigasterisk.com/room/')
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
12
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
13 ctx = ROOM['frontDoorControl']
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
14
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
15 def rdfGraphBody(body, headers):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
16 g = Graph()
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
17 g.parse(StringInputSource(body), format='nt')
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
18 return g
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
19
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
20 def mqttMessageFromState(state):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
21 return {
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
22 ROOM['locked']: b'OFF',
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
23 ROOM['unlocked']: b'ON',
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
24 }[state]
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
25
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
26 def stateFromMqtt(msg):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
27 return {
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
28 'OFF': ROOM['locked'],
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
29 'ON': ROOM['unlocked'],
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
30 }[msg.decode('ascii')]
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
31
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
32 class OutputPage(cyclone.web.RequestHandler):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
33 def put(self):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
34 arg = self.request.arguments
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
35 if arg.get('s') and arg.get('p'):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
36 subj = URIRef(arg['s'][-1])
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
37 pred = URIRef(arg['p'][-1])
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
38 obj = URIRef(self.request.body)
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
39 stmt = (subj, pred, obj)
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
40 else:
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
41 g = rdfGraphBody(self.request.body, self.request.headers)
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
42 assert len(g) == 1, len(g)
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
43 stmt = g.triples((None, None, None)).next()
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
44 self._onStatement(stmt)
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
45 post = put
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
46
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
47 def _onStatement(self, stmt):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
48 if stmt[0:2] == (ROOM['frontDoorLock'], ROOM['state']):
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
49 self.settings.mqtt.publish("frontdoor/switch/strike/command",
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
50 mqttMessageFromState(stmt[2]))
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
51 return
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
52 log.warn("ignoring %s", stmt)
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
53
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
54
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
55 class AutoLock(object):
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
56 def __init__(self, masterGraph, mqtt):
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
57 self.masterGraph = masterGraph
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
58 self.mqtt = mqtt
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
59 self.timeUnlocked = None
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
60 self.autoLockSec = 5
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
61 self.subj = ROOM['frontDoorLock']
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
62 task.LoopingCall(self.check).start(1)
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
63
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
64 def check(self):
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
65 now = time.time()
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
66 state = self.masterGraph._graph.value(self.subj, ROOM['state'])
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
67 if state == ROOM['unlocked']:
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
68 if self.timeUnlocked is None:
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
69 self.timeUnlocked = now
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
70 unlockedFor = now - self.timeUnlocked
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
71 self.masterGraph.patchObject(ctx, self.subj, ROOM['unlockedForSec'],
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
72 Literal(int(unlockedFor)))
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
73 self.masterGraph.patchObject(ctx, self.subj, ROOM['autoLockInSec'],
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
74 Literal(self.autoLockSec - int(unlockedFor)))
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
75 if unlockedFor > self.autoLockSec:
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
76 self.mqtt.publish("frontdoor/switch/strike/command",
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
77 mqttMessageFromState(ROOM['locked']))
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
78 else:
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
79 self.timeUnlocked = None
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
80 self.masterGraph.patchObject(ctx, self.subj, ROOM['unlockedForSec'], None)
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
81 self.masterGraph.patchObject(ctx, self.subj, ROOM['autoLockInSec'], None)
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
82
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
83
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
84 if __name__ == '__main__':
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
85 arg = docopt("""
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
86 Usage: front_door_lock.py [options]
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
87
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
88 -v Verbose
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
89 """)
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
90 log.setLevel(logging.INFO)
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
91 if arg['-v']:
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
92 enableTwistedLog()
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
93 log.setLevel(logging.DEBUG)
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
94
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
95 masterGraph = PatchableGraph()
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
96 mqtt = MqttClient(brokerPort=10010)
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
97 autoclose = AutoLock(masterGraph, mqtt)
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
98
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
99 def toGraph(payload):
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
100 log.info('mqtt->graph %r', payload)
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
101 masterGraph.patchObject(ctx, ROOM['frontDoorLock'], ROOM['state'],
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
102 stateFromMqtt(payload))
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
103
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
104 mqtt.subscribe("frontdoor/switch/strike/state").subscribe(on_next=toGraph)
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
105 port = 10011
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
106 reactor.listenTCP(port, cyclone.web.Application([
1184
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
107 (r"/()", cyclone.web.StaticFileHandler,
c87e5c8eb8ab frontdoor autolock. logging improvements. use simpler mqtt interface.
drewp <drewp@bigasterisk.com>
parents: 1182
diff changeset
108 {"path": ".", "default_filename": "index.html"}),
1182
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
109 (r"/graph", CycloneGraphHandler, {'masterGraph': masterGraph}),
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
110 (r"/graph/events", CycloneGraphEventsHandler,
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
111 {'masterGraph': masterGraph}),
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
112 (r'/output', OutputPage),
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
113 ], mqtt=mqtt, masterGraph=masterGraph, debug=arg['-v']), interface='::')
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
114 log.warn('serving on %s', port)
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
115
bd215f18e715 docker/etc initial version of front door mqtt<->rdf using some rx
drewp <drewp@bigasterisk.com>
parents:
diff changeset
116 reactor.run()