diff options
| -rw-r--r-- | .gitignore | 2 | ||||
| -rw-r--r-- | database.py | 96 | ||||
| -rw-r--r-- | db.env.example | 2 | ||||
| -rw-r--r-- | devices.py | 31 | ||||
| -rw-r--r-- | requirements.txt | 3 |
5 files changed, 134 insertions, 0 deletions
@@ -1,3 +1,5 @@ +db.env + # Byte-compiled / optimized / DLL files __pycache__/ *.py[cod] diff --git a/database.py b/database.py new file mode 100644 index 0000000..51b3ca1 --- /dev/null +++ b/database.py @@ -0,0 +1,96 @@ +from dataclasses import dataclass +import pymysql +import os + +@dataclass +class PowerDatabase: + host: str = "db" + user: str = "root" + passwd: str = None + db: str = "power" + port: int = 3306 + + def __enter__(self): + if self.passwd is None: + self.passwd = os.environ["MYSQL_ROOT_PASSWORD"] + + try: + self.__connection = self.__get_connection() + except Exception as e: + print(e) + if e.args[0] == 1049: + self.__connection = self.__build_db() + return self + + def __exit__(self, type, value, traceback): + self.__connection.close() + + def __get_connection(self): + return pymysql.connect( + host = self.host, + port = self.port, + user = self.user, + passwd = self.passwd, + charset = "utf8mb4", + database = self.db + ) + + def __build_db(self): + print("Building database...") + self.__connection = pymysql.connect( + host = self.host, + port = self.port, + user = self.user, + passwd = self.passwd, + charset = "utf8mb4", + ) + with self.__connection.cursor() as cursor: + # unsafe: + cursor.execute("CREATE DATABASE %s" % self.db) + cursor.execute("USE %s" % self.db) + + cursor.execute(""" + CREATE TABLE tasmota_devices ( + host VARCHAR(25) NOT NULL PRIMARY KEY, + username VARCHAR(50) NOT NULL, + password VARCHAR(50) NOT NULL + ); + """) + + if "TASMOTA_DEVICES" in os.environ.keys(): + devices = os.environ["TASMOTA_DEVICES"].split(",") + for device in devices: + host, username, password = device.split(":") + cursor.execute("INSERT INTO tasmota_devices VALUES (%s, %s, %s);", (host, username, password)) + + cursor.execute(""" + CREATE TABLE watt_readings ( + host VARCHAR(25) NOT NULL, + `datetime` DATETIME DEFAULT NOW(), + reading DECIMAL(3,1) NOT NULL, + FOREIGN KEY (host) REFERENCES tasmota_devices (host), + PRIMARY KEY (host, `datetime`) + ); + """) + + self.__connection.commit() + return self.__connection + + def get_tasmota_devices(self): + with self.__connection.cursor() as cursor: + cursor.execute("SELECT * FROM tasmota_devices;") + return cursor.fetchall() + + def append_watt_readings(self, host, reading): + cursor.execute("INSERT INTO watt_readings (host, reading) VALUES (%s, %s);", (host, reading)) + +if __name__ == "__main__": + if not os.path.exists(".docker"): + import dotenv + dotenv.load_dotenv(dotenv_path = "db.env") + host = "srv.home" + else: + host = "db" + + with PowerDatabase(host = host) as db: + print(db.get_tasmota_devices()) diff --git a/db.env.example b/db.env.example new file mode 100644 index 0000000..b7dfa35 --- /dev/null +++ b/db.env.example @@ -0,0 +1,2 @@ +MYSQL_ROOT_PASSWORD=********** +TASMOTA_DEVICES=switch.plug:admin:***********,nas.plug:admin:***********,router.plug:admin:************ diff --git a/devices.py b/devices.py new file mode 100644 index 0000000..0b0ba2a --- /dev/null +++ b/devices.py @@ -0,0 +1,31 @@ +import tasmotadevicecontroller +import database +import asyncio +import json +import os + +COUNTER_NAMES = ["Total", "Today"] +GAUGE_NAMES = ["Power", "ApparentPower", "ReactivePower", "Factor", "Voltage", "Current"] +SUMMARY_NAMES = ["TotalStartTime", "Yesterday"] +BOOLEAN_ENUM_NAMES = ["Power"] + +async def get_energy_for(host, username = None, password = None): + device = await tasmotadevicecontroller.TasmotaDevice().connect(host, username, password) + energy = await device.sendRawRequest("Status 8") + power = await device.getPower() + # friendlyname = await device.getFriendlyName() + energy["StatusSNS"]["ENERGY"]["Power"] = power + return energy["StatusSNS"]["ENERGY"] + # return {"%s_%s" % (status["FriendlyName"], k): v for k, v in status.items()} + +async def log_energies_for(db: database.PowerDatabase, host, username, password): + pass + +if __name__ == "__main__": + loop = asyncio.new_event_loop() + asyncio.set_event_loop(loop) + try: + print(asyncio.run(get_energy_for("switch.plug", "admin", "securebackdoor"))) + # asyncio.run(get_all_plugs("4u.plug:admin:securebackdoor,switch.plug:admin:securebackdoor,router.plug:admin:securebackdoor,nas.plug:admin:securebackdoor")) + except KeyboardInterrupt: + pass diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..ec38b64 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,3 @@ +tasmotadevicecontroller==0.0.8 +aiohttp==3.8.3 +pymysql |