93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
|
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()
|