setup first bit of rule based stuff
This commit is contained in:
parent
4b7b8e2cdc
commit
88ec6110c6
4
.gitignore
vendored
4
.gitignore
vendored
@ -1,3 +1,5 @@
|
||||
secrets.h
|
||||
arduino/MQTT/secret.h
|
||||
arduino/MQTT/secrets.h
|
||||
arduino/MQTT/secrets.h
|
||||
venv
|
||||
rule-listener/secrets.py
|
93
rule-listener/script.py
Normal file
93
rule-listener/script.py
Normal file
@ -0,0 +1,93 @@
|
||||
import paho.mqtt.client as mqtt
|
||||
import sched, time
|
||||
import secrets
|
||||
|
||||
s = sched.scheduler(time.time, time.sleep)
|
||||
|
||||
state = {}
|
||||
|
||||
rules = {
|
||||
"main/fan/1": [
|
||||
{ # after the fan has been on 2 times, another fan should be turned on
|
||||
"count": 2, # the amount of times
|
||||
"state": "on", # it has had this state
|
||||
"doTopic": "main/fan/2", # i will send to topic
|
||||
"doMessage": "on", # this message
|
||||
"delay": 3, # after this amount of seconds
|
||||
"reset": True # and i should reset my count afterwards
|
||||
},
|
||||
{ # 30 seconds after the fan has been turned on, turn it off again.
|
||||
"count": 1, # the amount of times
|
||||
"state": "on", # it has had this state
|
||||
"doTopic": "main/fan/11", # i will send to topic
|
||||
"doMessage": "off", # this message
|
||||
"delay": 30, # after this amount of seconds
|
||||
"reset": False # and i should reset my count afterwards
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
def on_connect(client, userdata, flags, reason_code, properties):
|
||||
print(f"Connected with result code {reason_code}")
|
||||
client.subscribe("main/#")
|
||||
|
||||
def publish_later(topic, msg, delay):
|
||||
print(f"TODO: make this run {delay} seconds later")
|
||||
# s.enter(delay, 1, publish, kwargs={'topic': topic, 'msg': msg})
|
||||
# s.run()
|
||||
publish(topic, msg)
|
||||
|
||||
def publish(topic, msg):
|
||||
result = mqttc.publish(topic, msg)
|
||||
|
||||
status = result[0]
|
||||
|
||||
if not status == 0:
|
||||
print(f"Failed to send message to topic {topic}")
|
||||
|
||||
def check_against_rules(msg):
|
||||
if msg.topic in rules:
|
||||
print(f"Found {len(rules[msg.topic])} rules on this topic")
|
||||
for rule in rules[msg.topic]:
|
||||
print(f"Rule is met when the count is above {rule['count']}. Current count is {state[msg.topic]['count']}")
|
||||
|
||||
if state[msg.topic]['count'] > rule['count']:
|
||||
print(f"i matched a rule")
|
||||
if(rule['reset']):
|
||||
state[msg.topic]['count'] = 0
|
||||
|
||||
if rule['delay']:
|
||||
publish_later(rule['doMessage'], rule['doMessage'], rule['delay'])
|
||||
else:
|
||||
publish(rule['doTopic'], rule['doMessage'])
|
||||
else:
|
||||
print("i have no rules for this topic")
|
||||
|
||||
print("finished rule checking")
|
||||
|
||||
# The callback for when a PUBLISH message is received from the server.
|
||||
def on_message(client, userdata, msg):
|
||||
if msg.retain:
|
||||
return False
|
||||
|
||||
if msg.topic in state:
|
||||
print("i already exist")
|
||||
state[msg.topic]["count"] += 1
|
||||
else:
|
||||
print("i do not")
|
||||
state[msg.topic] = {}
|
||||
state[msg.topic]["count"] = 0
|
||||
|
||||
check_against_rules(msg)
|
||||
|
||||
|
||||
|
||||
mqttc = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2)
|
||||
mqttc.on_connect = on_connect
|
||||
mqttc.on_message = on_message
|
||||
|
||||
mqttc.username_pw_set(secrets.mqtt_username, secrets.mqtt_password)
|
||||
|
||||
mqttc.connect("mqtt.klank.school", 7000, 60)
|
||||
|
||||
mqttc.loop_forever()
|
Loading…
Reference in New Issue
Block a user