80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
import time
|
|
from datetime import datetime
|
|
|
|
from energy_monitor import CONN_STR, DatabaseManager, PlcManager, root_logger
|
|
|
|
# Globals
|
|
DEVICE_DELAY = 1 # Delay between device data fetches
|
|
POOL_RATE = 2 # 2s polling rate reading of NextRead value from database
|
|
FALLBACK_INTERVAL = 3600 # 1h interval in case db read issue
|
|
|
|
|
|
def process_plc_devices(db_manager: DatabaseManager, plc_manager: PlcManager):
|
|
plc_configs = db_manager.get_plc_configs()
|
|
|
|
for plc_config in plc_configs:
|
|
try:
|
|
if not plc_manager.check_connection(plc_config.ip):
|
|
root_logger.error(
|
|
"PLC connection failed",
|
|
extra={
|
|
"PlcId": plc_config.id,
|
|
"PlcIp": plc_config.ip,
|
|
"timestamp": datetime.now().isoformat(),
|
|
},
|
|
)
|
|
continue
|
|
|
|
air_value, energy_value = plc_manager.read_plc_data(plc_config)
|
|
db_manager.save_energy_data(plc_config.id, energy_value, air_value, True)
|
|
|
|
root_logger.info(
|
|
"✅ Data successfully read and saved",
|
|
extra={
|
|
"PlcId": plc_config.id,
|
|
"PlcIp": plc_config.ip,
|
|
"energy_value": energy_value,
|
|
"air_value": air_value,
|
|
"timestamp": datetime.now().isoformat(),
|
|
},
|
|
)
|
|
|
|
time.sleep(DEVICE_DELAY) # n-second delay between devices
|
|
|
|
except Exception as e:
|
|
root_logger.error(
|
|
"Error processing PLC",
|
|
extra={
|
|
"PlcId": plc_config.id,
|
|
"PlcIp": plc_config.ip,
|
|
"error": str(e),
|
|
"timestamp": datetime.now().isoformat(),
|
|
},
|
|
)
|
|
|
|
|
|
def main():
|
|
db_manager = DatabaseManager(CONN_STR)
|
|
plc_manager = PlcManager()
|
|
|
|
while True:
|
|
try:
|
|
scheduler_config = db_manager.get_scheduler_config()
|
|
|
|
if datetime.now() >= scheduler_config.next_read:
|
|
process_plc_devices(db_manager, plc_manager)
|
|
db_manager.update_next_read(scheduler_config.interval)
|
|
|
|
time.sleep(POOL_RATE)
|
|
|
|
except Exception as e:
|
|
root_logger.error(
|
|
"Main loop error! Falling back to 1h interval.",
|
|
extra={"error": str(e), "timestamp": datetime.now().isoformat()},
|
|
)
|
|
time.sleep(FALLBACK_INTERVAL) # Default fallback interval
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|