setup first bit of rule based stuff

This commit is contained in:
vitrinekast 2024-10-28 17:45:30 +01:00
parent 4b7b8e2cdc
commit 88ec6110c6
2 changed files with 96 additions and 1 deletions

4
.gitignore vendored
View File

@ -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
View 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()