Obbart/edelweiss
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactoring funzioni di utility

Browse Source
This commit is contained in:
Emanuele Trabattoni
2025-06-01 10:14:46 +02:00
parent 676918e686
commit 021b7e70af
3 changed files with 91 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
upsmon/ups.py
View File

@@ -4,25 +4,15 @@ import time
import string
import serial
import logging
import signal
import json
from influxdb_client_3 import InfluxDBClient3, Point
from pyutils.utils import *
from influxdb_client_3 import InfluxDBClient3
# Get environment variables
env = dict(os.environ)
LOGGER: logging.Logger
class SignalHandler:
running: bool
def __init__(self):
self.running: bool = True
signal.signal(signal.SIGINT, self._handle_sigint)
signal.signal(signal.SIGTERM, self._handle_sigint)
def _handle_sigint(self, signum, frame):
self.running = False
GIT_HASH: str = "HHHHHHHHH"
def send(port: serial.Serial, d: str):
port.write((d+'\r').encode('ascii'))
@@ -49,43 +39,60 @@ def bruteforceCommands(port: serial.Serial):
send(port, d)
receive(port, d)
##################
###### MAIN ######
##################
def main():
INTERVAL = int(env['INTERVAL'])
LOGGER.debug(json.dumps(env, indent=2))
run: SignalHandler = SignalHandler()
port = serial.Serial(port=env['PORT'], baudrate=int(env['BAUD']), bytesize=8, parity='N', stopbits=1)
write_client = InfluxDBClient3(host=env['INFLUXDB_URL'],
token=env['INFLUXDB_TOKEN'],
database=env['INFLUXDB_DATABASE'])
UPS_COMMAND = "Q1"
try:
# Init InfluxDB-v3 Client
write_client = InfluxDBClient3(host=env['INFLUXDB_URL'],
token=env['INFLUXDB_TOKEN'],
database=env['INFLUXDB_DATABASE'])
# Init Serial port
port = serial.Serial(port=env['PORT'], baudrate=int(env['BAUD']), bytesize=8, parity='N', stopbits=1)
except Exception as e:
LOGGER.error(e)
return 0
finally:
LOGGER.info(f"InfluxDB Connected: [{env['INFLUXDB_URL']}/{env['INFLUXDB_DATABASE']}]")
LOGGER.info(f"Serial Port Open: [{env['PORT']}]")
##############################
########## MAIN LOOP #########
##############################
run: SignalHandler = SignalHandler(LOGGER)
while run.running:
try:
send(port, "Q1")
data = receive(port, "Q1").lstrip('(').split()
if len(data) < 8:
LOGGER.error(f"Incomplete data: {data}")
send(port, UPS_COMMAND)
raw_data = receive(port, UPS_COMMAND).lstrip('(').split()
if len(raw_data) < 8:
LOGGER.error(f"Incomplete data: {raw_data}")
continue
values = {
'inV': float(data[0]),
'outV': float(data[2]),
'loadPercent': float(data[3]),
'lineFreq': float(data[4]),
'timeLeft': float(data[5]),
'onBatt': True if str(data[7]).startswith('1') else False,
'onLine': True if str(data[7]).endswith('1') else False,
'inV': float(raw_data[0]),
'outV': float(raw_data[2]),
'loadPercent': float(raw_data[3]),
'lineFreq': float(raw_data[4]),
'timeLeft': float(raw_data[5]),
'onBatt': True if str(raw_data[7]).startswith('1') else False,
'onLine': True if str(raw_data[7]).endswith('1') else False,
}
LOGGER.debug(f"UPS Status: \n{json.dumps(values, indent=2)}")
if values['onBatt']:
LOGGER.info(f"OnBattery\n{json.dumps(values,indent=2)}")
p = Point('ups')
for k,v in values.items():
p.field(k,v)
write_client.write(record=p)
write_client.write(record=dict2Point('ups', values))
time.sleep(INTERVAL)
except Exception as e:
print(f"Unexpected exception: [{e}]")
return 1
##############################
###### END MAIN LOOP #########
##############################
port.close()
return 0
if __name__ == "__main__":
# Logger Constants
@@ -112,6 +119,7 @@ if __name__ == "__main__":
LOGGER.addHandler(cl)
LOGGER.warning(f"UPSmon started on: {time.asctime()}")
LOGGER.info(f"UPSmon BUILD: {env.get("VER", "Test")}")
while main():
LOGGER.error("Main thread exited unexpectedly")