Renamed the scripts to adhere to python conventionsHEADmaster

1 files changed, 104 insertions, 0 deletions

diff --git a/tasmotaMQTTClient.py b/tasmotaMQTTClient.py
new file mode 100644
index 0000000..e8a43b8
--- /dev/null
+++ b/tasmotaMQTTClient.py
@@ -0,0 +1,104 @@
+import paho.mqtt.client as paho
+import threading
+import argparse
+import getpass
+import docker_net
+import time
+import json
+
+SENT_EVENT = threading.Event()
+
+class MQTTClient:
+
+    switch_energy = None
+    switch_power = None
+
+    def __init__(self, host, friendlyname, username, password, message = None):
+        self.host = host
+        self.friendlyname = friendlyname
+        self.username = username
+        self.password = password
+        self.message = message
+
+        # print("Instantiated. Message:", message)
+
+        self.mqtt_c = paho.Client("tasmota-cli", clean_session = True)
+
+        self.mqtt_c.on_connect = self._on_connect_cb
+        if message is None:
+            self.mqtt_c.on_message = self._on_message_cb
+
+        self.mqtt_c.username_pw_set(username = username, password = password)
+        self.mqtt_c.connect(self.host, 1883, 60)
+        self.mqtt_c.loop_forever()
+    
+    def _on_connect_cb(self, mqtt, userdata, flags, rc):
+        # print("Connected to broker")
+        topic = "tele/TasmotaPlug/%s/+" % self.friendlyname
+        if self.message is None:
+            print("Waiting for '%s'" % topic)
+            self.mqtt_c.subscribe(topic)
+
+        else:
+            self.mqtt_c.publish("cmnd/TasmotaPlug/%s/Power" % self.friendlyname, payload = self.message)
+            print("Sent message '%s' to topic 'cmnd/TasmotaPlug/%s/Power'" % (self.message, self.friendlyname))
+            SENT_EVENT.set()
+            self.mqtt_c.disconnect()
+
+    def _on_message_cb(self, mqtt, userdata, msg):
+        # print('Topic: {0} | Message: {1}'.format(msg.topic, msg.payload))
+
+        if msg.topic.split("/")[2] == self.friendlyname:
+            if msg.topic.split("/")[3] == "SENSOR":
+                self.switch_energy = json.loads(msg.payload.decode())["ENERGY"]
+            elif msg.topic.split("/")[3] == "STATE":
+                self.switch_power = json.loads(msg.payload.decode())["POWER"]
+        
+        if self.switch_power is not None and self.switch_energy is not None:
+            self.mqtt_c.disconnect()
+
+
+
+parser = argparse.ArgumentParser()
+parser.add_argument(
+    "-m", "--mqtt-host",
+    type = str,
+    help = "MQTT Server",
+    default = "<docker>"
+)
+parser.add_argument(
+    "-u", "--user",
+    type = str,
+    help = "Username to login with",
+    required = True
+)
+parser.add_argument(
+    "-t", "--toggle",
+    action = "store_true",
+    help = "Toggle current power status"
+)
+parser.add_argument(
+    "-n", "--friendlyname",
+    help = "Device friendly name",
+    type = str,
+    required = True
+)
+
+if __name__ == "__main__":
+    args = vars(parser.parse_args())
+    if args["mqtt_host"] == "<docker>":
+       args["mqtt_host"] = docker_net.get_mqtt_addr()[0]
+    if args["user"] is not None:
+        args["password"] = getpass.getpass("Input password for %s@%s: " % (args["user"], args["mqtt_host"]))
+    else:
+        args["password"] = None
+
+    if args["toggle"]:
+        client = MQTTClient(args["mqtt_host"], args["friendlyname"], args["user"], args["password"], "TOGGLE")    
+        # print("Waiting for event...")
+        SENT_EVENT.wait()
+        # print("Done waiting.")
+
+    client = MQTTClient(args["mqtt_host"], args["friendlyname"], args["user"], args["password"])
+    print(json.dumps(client.switch_energy, indent = 4))
+    print("'%s' is currently %s" % (client.friendlyname, client.switch_power))