1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
import tasmotadevicecontroller
import database
import asyncio
import datetime
import json
import sys
import os
if not os.path.exists(os.path.join("/app", ".docker")):
import dotenv
dotenv.load_dotenv(dotenv_path = "db.env")
HOST = "srv.athome"
else:
HOST = None
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()
return energy["StatusSNS"]["ENERGY"]
# return {"%s_%s" % (status["FriendlyName"], k): v for k, v in status.items()}
async def poll_watt_for(db: database.PowerDatabase, host, username, password):
power = await get_energy_for(host, username, password)
power = float(power['Power'])
db.append_watt_readings(host, power)
print("'%s' is using %.1fW at %s" % (host, power, datetime.datetime.now()))
async def poll_yesterday_kwh_for(db: database.PowerDatabase, host, username, password):
power = await get_energy_for(host, username, password)
power = float(power['Yesterday'])
db.append_kwh_readings(host, power)
print("'%s' used %.1fkWh yesterday" % (host, power))
def poll_watt_all():
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
with database.PowerDatabase(host = HOST) as db:
for host, username, password in db.get_tasmota_devices():
try:
asyncio.run(poll_watt_for(db, host, username, password))
except:
print("Retrying %s..." % host)
asyncio.run(poll_watt_for(db, host, username, password))
def poll_kwh_all():
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
with database.PowerDatabase(host = HOST) as db:
for host, username, password in db.get_tasmota_devices():
try:
asyncio.run(poll_yesterday_kwh_for(db, host, username, password))
except:
print("Retrying %s..." % host)
asyncio.run(poll_yesterday_kwh_for(db, host, username, password))
if __name__ == "__main__":
if sys.argv[1] == "daily":
poll_kwh_all()
else:
poll_watt_all()
|
