nick
/
tasmota-power-exporter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated to match fork
Docker Push / build (push) Successful in 40s Details

This commit is contained in:
Nick Trimborn 2023-07-20 13:07:41 +02:00
parent 36c3489c55
commit 00f5029972
2 changed files with 8 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
metrics.py
View File

@ -5,14 +5,13 @@ from os import getenv
from time import sleep from time import sleep
from prometheus_client.core import GaugeMetricFamily, REGISTRY, CounterMetricFamily from prometheus_client.core import GaugeMetricFamily, REGISTRY, CounterMetricFamily
from prometheus_client import start_http_server from prometheus_client import start_http_server
from loguru import logger
class TasmotaCollector(object): class TasmotaCollector(object):
def __init__(self): def __init__(self):
self.ip = getenv('DEVICE_IP') self.ip = getenv('DEVICE_IP')
self.device_name = getenv("DEVICE_NAME").lower() self.device_name = getenv("DEVICE_NAME").lower()
if not self.ip:
self.ip = "192.168.4.1"
self.user = getenv('USER') self.user = getenv('USER')
self.password = getenv('PASSWORD') self.password = getenv('PASSWORD')
@ -37,7 +36,7 @@ class TasmotaCollector(object):
def fetch(self): def fetch(self):
url = 'http://' + self.ip + '/?m=1' url = 'http://' + self.ip + '/?m=1'
logger.info("Getting Metrics from: " + url)
session = requests.Session() session = requests.Session()
if self.user and self.password: if self.user and self.password:
@ -50,16 +49,9 @@ class TasmotaCollector(object):
string_values = str(page.text).split("{s}") string_values = str(page.text).split("{s}")
for i in range(1,len(string_values)): for i in range(1,len(string_values)):
try: label = string_values[i].split("{m}")[0]
label = string_values[i].split("{m}")[0] value = string_values[i].split("{m}")[1].split("{e}")[0]
value = string_values[i].split("{m}")[1].split("{e}")[0] values[label] = value
if "<td" in value:
value = value.replace("</td><td style='text-align:left'>", "")
value = value.replace("</td><td>&nbsp;</td><td>", "")
values[label] = value
except IndexError:
continue
return values return values
def signal_handler(signal, frame): def signal_handler(signal, frame):
@ -70,12 +62,9 @@ signal.signal(signal.SIGINT, signal_handler)
if __name__ == '__main__': if __name__ == '__main__':
port = getenv('EXPORTER_PORT') port = getenv('EXPORTER_PORT')
if not port: logger.info("Starting HTTP server on port: " + url)
port = 8000
start_http_server(int(port)) start_http_server(int(port))
REGISTRY.register(TasmotaCollector()) REGISTRY.register(TasmotaCollector())
while(True): while(True):
sleep(1) sleep(1)

3
requirements.txt
View File

@ -1,2 +1,3 @@
prometheus_client prometheus_client
requests requests
loguru