delete mqtt scripts
This commit is contained in:
Ingolf Wagner
parent
18cc3c5a7a
commit
0f1fb837b9
5 changed files with 0 additions and 340 deletions
2
mqtt/.gitignore
vendored
2
mqtt/.gitignore
vendored
|
|
@ -1,2 +0,0 @@
|
|||
venv
|
||||
**/__pycache__
|
||||
147
mqtt/fyrtur.py
147
mqtt/fyrtur.py
|
|
@ -1,147 +0,0 @@
|
|||
import time
|
||||
from enum import Enum
|
||||
from typing import Dict
|
||||
|
||||
import paho.mqtt.client as mqtt
|
||||
import json
|
||||
import threading
|
||||
|
||||
scene = "up-dark"
|
||||
|
||||
|
||||
class Position(Enum):
|
||||
UP = 1
|
||||
DOWN = 2
|
||||
HALF = 3
|
||||
|
||||
|
||||
class Fyrtur:
|
||||
def __init__(self, topic, top, bottom):
|
||||
self.topic = topic
|
||||
self.top = top
|
||||
self.bottom = bottom
|
||||
self.current_position = 100
|
||||
self.wanted_position = 100
|
||||
|
||||
def update_position(self, payload):
|
||||
self.current_position = payload["position"]
|
||||
|
||||
def needs_publish(self):
|
||||
return self.wanted_position != self.current_position
|
||||
|
||||
def topic_and_payload_for_set(self):
|
||||
payload = {"position": self.wanted_position}
|
||||
return ("%s/set" % self.topic), json.dumps(payload)
|
||||
|
||||
|
||||
class FyrturWatcher:
|
||||
def __init__(self, fyrturs: Dict[str, Fyrtur]):
|
||||
self.fyrturs = fyrturs
|
||||
|
||||
def get_topics(self):
|
||||
return [fyrtur.topic for fyrtur in self.fyrturs.values()]
|
||||
|
||||
def update_position(self, topic, payload):
|
||||
for fyrtur in self.fyrturs.values():
|
||||
if fyrtur.topic == topic:
|
||||
fyrtur.update_position(payload)
|
||||
return
|
||||
|
||||
def update(self, name, position: Position):
|
||||
fyrtur: Fyrtur = self.fyrturs.get(name)
|
||||
if position == Position.UP:
|
||||
fyrtur.wanted_position = fyrtur.top
|
||||
elif position == Position.DOWN:
|
||||
fyrtur.wanted_position = fyrtur.bottom
|
||||
elif position == Position.HALF:
|
||||
fyrtur.wanted_position = round(
|
||||
(fyrtur.top - fyrtur.bottom) / 2 + fyrtur.bottom
|
||||
)
|
||||
|
||||
def publish(self, client):
|
||||
for fyrtur in self.fyrturs.values():
|
||||
if fyrtur.needs_publish():
|
||||
topic, payload = fyrtur.topic_and_payload_for_set()
|
||||
client.publish(topic, payload)
|
||||
time.sleep(2)
|
||||
|
||||
|
||||
watcher = FyrturWatcher(
|
||||
{
|
||||
"office1": Fyrtur(topic="zigbee2mqtt/office_fyrtur_1", top=100, bottom=16),
|
||||
"office2": Fyrtur(topic="zigbee2mqtt/office_fyrtur_2", top=100, bottom=22),
|
||||
"bedroom": Fyrtur(topic="zigbee2mqtt/bedroom_fyrtur_1", top=100, bottom=16),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
# The callback for when the client receives a CONNACK response from the server.
|
||||
def on_connect(client, _userdata, _flags, rc):
|
||||
print("Connected with result code " + str(rc))
|
||||
|
||||
threading.Thread(target=loop_thread, args=(client,), daemon=True).start()
|
||||
|
||||
# Subscribing in on_connect() means that if we lose the connection and
|
||||
# reconnect then subscriptions will be renewed.
|
||||
client.subscribe("control/lights/set")
|
||||
for topic in watcher.get_topics():
|
||||
client.subscribe(topic)
|
||||
|
||||
|
||||
# The callback for when a PUBLISH message is received from the server.
|
||||
def on_message(client, _userdata, msg):
|
||||
global scene
|
||||
(topic, payload) = parse_message(msg)
|
||||
if topic == "control/lights/set":
|
||||
print("set scene %s -> %s" % (scene, payload["scene"]))
|
||||
scene = payload["scene"]
|
||||
update_scene(client)
|
||||
else:
|
||||
print("got %s" % topic)
|
||||
watcher.update_position(topic, payload)
|
||||
|
||||
|
||||
def parse_message(msg):
|
||||
m_decode = str(msg.payload.decode("utf-8", "ignore"))
|
||||
payload = json.loads(m_decode) # decode json data
|
||||
return msg.topic, payload
|
||||
|
||||
|
||||
def update_scene(client):
|
||||
if scene in ["night", "down"]:
|
||||
watcher.update("office1", Position.DOWN)
|
||||
watcher.update("office2", Position.DOWN)
|
||||
watcher.update("bedroom", Position.DOWN)
|
||||
elif scene in ["default", "up-bright", "up-dark", "outside"]:
|
||||
watcher.update("office1", Position.UP)
|
||||
watcher.update("office2", Position.UP)
|
||||
watcher.update("bedroom", Position.UP)
|
||||
elif scene in ["half"]:
|
||||
watcher.update("office1", Position.HALF)
|
||||
watcher.update("office2", Position.HALF)
|
||||
watcher.update("bedroom", Position.HALF)
|
||||
else:
|
||||
watcher.update("office1", Position.UP)
|
||||
watcher.update("office2", Position.UP)
|
||||
watcher.update("bedroom", Position.UP)
|
||||
|
||||
watcher.publish(client)
|
||||
|
||||
|
||||
def loop_thread(client):
|
||||
while True:
|
||||
watcher.publish(client)
|
||||
time.sleep(120)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
mqttClient = mqtt.Client()
|
||||
mqttClient.on_connect = on_connect
|
||||
mqttClient.on_message = on_message
|
||||
mqttClient.username_pw_set("homeassistant", password="password")
|
||||
mqttClient.connect("pepe.private", 1883, 60)
|
||||
# Blocking call that processes network traffic, dispatches callbacks and
|
||||
# handles reconnecting.
|
||||
# Other loop*() functions are available that give a threaded interface and a
|
||||
# manual interface.
|
||||
mqttClient.loop_forever()
|
||||
170
mqtt/heater.py
170
mqtt/heater.py
|
|
@ -1,170 +0,0 @@
|
|||
import json
|
||||
import paho.mqtt.client as mqtt
|
||||
import threading
|
||||
import time
|
||||
from typing import Dict
|
||||
|
||||
|
||||
class Heater:
|
||||
def __init__(self, topic):
|
||||
self.not_initialized_yet = True
|
||||
self.wanted_temperature = 14
|
||||
self.actual_temperature = 14
|
||||
self.topic = topic
|
||||
|
||||
def needs_publish(self):
|
||||
if self.not_initialized_yet:
|
||||
return True
|
||||
else:
|
||||
return self.wanted_temperature != self.actual_temperature
|
||||
|
||||
def update_actual_heating_point(self, payload):
|
||||
heating_setpoint = int(payload["current_heating_setpoint"])
|
||||
if self.not_initialized_yet:
|
||||
self.not_initialized_yet = False
|
||||
self.wanted_temperature = heating_setpoint
|
||||
print(
|
||||
"%s: update wanted temperature %d"
|
||||
% (self.topic, self.actual_temperature)
|
||||
)
|
||||
self.actual_temperature = heating_setpoint
|
||||
print(
|
||||
"%s: update actual temperature %d" % (self.topic, self.actual_temperature)
|
||||
)
|
||||
|
||||
def topic_and_payload_for_query(self):
|
||||
payload = {
|
||||
"current_heating_setpoint": "",
|
||||
"occupied_heating_setpoint": "",
|
||||
"unoccupied_heating_setpoint": "",
|
||||
"local_temperature": "",
|
||||
# "pi_heating_demand": "",
|
||||
# "system_mode": "",
|
||||
}
|
||||
return ("%s/get" % self.topic), json.dumps(payload)
|
||||
|
||||
def topic_and_payload_for_set(self):
|
||||
payload = {
|
||||
"system_mode": "auto",
|
||||
# "current_heating_setpoint": str(self.wanted_temperature),
|
||||
"occupied_heating_setpoint": str(self.wanted_temperature),
|
||||
"unoccupied_heating_setpoint": str(self.wanted_temperature),
|
||||
"eurotronic_host_flags": {"window_open": True},
|
||||
}
|
||||
return ("%s/set" % self.topic), json.dumps(payload)
|
||||
|
||||
|
||||
class Watcher:
|
||||
def __init__(self, heater: Dict[str, Heater]):
|
||||
self.heater = heater
|
||||
|
||||
def publish(self, client):
|
||||
for heater in self.heater.values():
|
||||
if heater.needs_publish():
|
||||
topic, payload = heater.topic_and_payload_for_set()
|
||||
client.publish(topic, payload)
|
||||
time.sleep(2)
|
||||
|
||||
def update(self, name, temperature):
|
||||
heater: Heater = self.heater.get(name)
|
||||
heater.wanted_temperature = temperature
|
||||
|
||||
def get_topics(self):
|
||||
return [heater.topic for heater in self.heater.values()]
|
||||
|
||||
def update_actual_heating_point_for_topic(self, topic, payload):
|
||||
for heater in self.heater.values():
|
||||
if heater.topic == topic:
|
||||
heater.update_actual_heating_point(payload)
|
||||
return
|
||||
|
||||
def pull_values(self, client):
|
||||
for heater in self.heater.values():
|
||||
topic, payload = heater.topic_and_payload_for_query()
|
||||
client.publish(topic, payload)
|
||||
|
||||
|
||||
scene = "default"
|
||||
|
||||
watcher = Watcher(
|
||||
{
|
||||
"office1": Heater(topic="zigbee2mqtt/office_heater_1"),
|
||||
"office2": Heater(topic="zigbee2mqtt/office_heater_2"),
|
||||
"bedroom": Heater(topic="zigbee2mqtt/bedroom_heater_1"),
|
||||
"storage": Heater(topic="zigbee2mqtt/storage_heater_1"),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
# The callback for when the client receives a CONNACK response from the server.
|
||||
def on_connect(client, _userdata, _flags, rc):
|
||||
print("Connected with result code " + str(rc))
|
||||
|
||||
threading.Thread(target=loop_thread, args=(client,), daemon=True).start()
|
||||
|
||||
# Subscribing in on_connect() means that if we lose the connection and
|
||||
# reconnect then subscriptions will be renewed.
|
||||
client.subscribe("control/lights/set")
|
||||
for topic in watcher.get_topics():
|
||||
client.subscribe(topic)
|
||||
watcher.pull_values(client)
|
||||
|
||||
|
||||
# The callback for when a PUBLISH message is received from the server.
|
||||
def on_message(client, _userdata, msg):
|
||||
global scene
|
||||
(topic, payload) = parse_message(msg)
|
||||
if topic == "control/lights/set":
|
||||
print("set scene %s -> %s" % (scene, payload["scene"]))
|
||||
scene = payload["scene"]
|
||||
update_scene(client)
|
||||
else:
|
||||
print("got %s" % topic)
|
||||
watcher.update_actual_heating_point_for_topic(topic, payload)
|
||||
|
||||
|
||||
def parse_message(msg):
|
||||
m_decode = str(msg.payload.decode("utf-8", "ignore"))
|
||||
payload = json.loads(m_decode) # decode json data
|
||||
return msg.topic, payload
|
||||
|
||||
|
||||
def update_scene(client):
|
||||
if scene in ["night", "outside"]:
|
||||
watcher.update("office1", 14)
|
||||
watcher.update("office2", 14)
|
||||
watcher.update("bedroom", 14)
|
||||
watcher.update("storage", 14)
|
||||
elif scene in ["default", "up-bright", "up-dark", "half", "down"]:
|
||||
watcher.update("office1", 25)
|
||||
watcher.update("office2", 25)
|
||||
watcher.update("bedroom", 18)
|
||||
watcher.update("storage", 18)
|
||||
else:
|
||||
watcher.update("office1", 14)
|
||||
watcher.update("office2", 14)
|
||||
watcher.update("bedroom", 14)
|
||||
watcher.update("storage", 14)
|
||||
|
||||
watcher.publish(client)
|
||||
|
||||
|
||||
def loop_thread(client):
|
||||
while True:
|
||||
watcher.publish(client)
|
||||
watcher.pull_values(client)
|
||||
time.sleep(120)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
mqttClient = mqtt.Client()
|
||||
mqttClient.on_connect = on_connect
|
||||
mqttClient.on_message = on_message
|
||||
mqttClient.username_pw_set("homeassistant", password="password")
|
||||
mqttClient.connect("pepe.private", 1883, 60)
|
||||
|
||||
# Blocking call that processes network traffic, dispatches callbacks and
|
||||
# handles reconnecting.
|
||||
# Other loop*() functions are available that give a threaded interface and a
|
||||
# manual interface.
|
||||
mqttClient.loop_forever()
|
||||
|
|
@ -1 +0,0 @@
|
|||
paho-mqtt
|
||||
|
|
@ -1,20 +0,0 @@
|
|||
{ pkgs ? import <nixpkgs> { } }:
|
||||
let
|
||||
|
||||
myPython = pkgs.python3.withPackages
|
||||
(python-packages: with python-packages; [ paho-mqtt ]);
|
||||
|
||||
startServer = pkgs.writers.writeBashBin "start-server" ''
|
||||
${myPython}/bin/python ./heater.py
|
||||
'';
|
||||
|
||||
reformat = pkgs.writers.writeBashBin "reformat" ''
|
||||
${pkgs.black}/bin/black --exclude venv ${toString ./.}
|
||||
'';
|
||||
|
||||
in
|
||||
pkgs.mkShell {
|
||||
|
||||
buildInputs = with pkgs; [ myPython startServer reformat ];
|
||||
|
||||
}
|
||||
Loading…
Reference in a new issue