aboutsummaryrefslogtreecommitdiffstats
path: root/mqtt-client/mqtt-client.py
diff options
context:
space:
mode:
Diffstat (limited to 'mqtt-client/mqtt-client.py')
-rw-r--r--mqtt-client/mqtt-client.py63
1 files changed, 0 insertions, 63 deletions
diff --git a/mqtt-client/mqtt-client.py b/mqtt-client/mqtt-client.py
deleted file mode 100644
index 248e5af..0000000
--- a/mqtt-client/mqtt-client.py
+++ /dev/null
@@ -1,63 +0,0 @@
-import paho.mqtt.client as paho
-from influxdb_client import InfluxDBClient, Point, WritePrecision
-from influxdb_client.client.write_api import SYNCHRONOUS
-import json
-import os
-
-class MQTTClient:
- def __init__(self):
- self.influxc = InfluxDBClient(
- url = "http://%s:8086" % INFLUXDB_HOST,
- token = os.environ["DOCKER_INFLUXDB_INIT_ADMIN_TOKEN"],
- org = os.environ["DOCKER_INFLUXDB_INIT_ORG"]
- )
- self.influxc.ping()
-
- self.mqttc = paho.Client('power-listener', clean_session = True)
- self.mqttc.on_connect = self._on_connect_cb
- self.mqttc.on_message = self._on_message_cb
-
- self.mqttc.username_pw_set(os.environ["MQTT_USER"], password = os.environ["MQTT_PASSWD"])
- self.mqttc.connect(MQTT_HOST, 1883, 60)
- self.mqttc.loop_forever()
-
- def _on_connect_cb(self, mqtt, userdata, flags, rc):
- print("Connected to broker")
- self.mqttc.subscribe("tele/+/SENSOR")
-
- def _on_message_cb(self, mqtt, userdata, msg):
- print('Topic: {0} | Message: {1}'.format(msg.topic, msg.payload))
-
- if "Tasmota" in msg.topic:
- self.handle_tasmota(msg)
-
- def handle_tasmota(self, msg):
- from_ = msg.topic.split("/")[1]
- msg_j = json.loads(msg.payload.decode())
- #print(from_)
- fields = {k: v for k, v in msg_j["ENERGY"].items() if k not in {"TotalStartTime"}}
- points = [{"measurement": "tasmota_power", "tags": {"plug": from_}, "fields": fields}]
- write_api = self.influxc.write_api(write_options = SYNCHRONOUS)
- write_api.write(
- os.environ["DOCKER_INFLUXDB_INIT_BUCKET"],
- os.environ["DOCKER_INFLUXDB_INIT_ORG"],
- points,
- write_precision = WritePrecision.S
- )
-
-if __name__ == "__main__":
- env_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "config.env")
- if os.path.exists(env_path):
- import dotenv
- dotenv.load_dotenv(dotenv_path = env_path)
- INFLUXDB_HOST = "localhost"
- MQTT_HOST = "localhost"
- else:
- INFLUXDB_HOST = "influxdb"
- MQTT_HOST = "mqtt"
-
- mqtt_client = MQTTClient()
-
-
-
-