smart-energy-monitor/box-pc/plc-connector/middlewares/time_correlation.py

import logging
from datetime import datetime, timedelta
from dataclasses import dataclass, field
from structures.common import BaseMeasurement

logger = logging.getLogger(__name__)

@dataclass(frozen=True)
class CorrelatedMeasurements(BaseMeasurement):
  series: str = field(default="correlated", init=False)
  measurement_24v: BaseMeasurement
  measurement_480v: BaseMeasurement
  measurement_plant: BaseMeasurement

class TimeCorrelation:
  def __init__(self, parent):
    self.state = {}
    self.timestamp = None

  def execute(self, values: list):
    results = []
    for i, measurement in enumerate(values):
      self.state[type(measurement).__name__] = measurement

      if self.timestamp and self.timestamp > measurement.timestamp + timedelta(milliseconds=100):
        logger.error(f"Timestamps are not in order: {measurement.series} is {self.timestamp - measurement.timestamp} to late")
      
      if len(values) > i+1 and values[i+1].timestamp == measurement.timestamp:
        continue

      self.timestamp = measurement.timestamp
      results.append(CorrelatedMeasurements(
        timestamp = measurement.timestamp,
        source = ','.join([x.source for x in self.state.values()]),
        measurement_24v = self.state.get("Measurement24v", None),
        measurement_480v = self.state.get("Measurement480v", None),
        measurement_plant = self.state.get("CompactLogixState", None) or self.state.get("S7State", None)
      ))
    return results