diff --git a/packages/modules/devices/idm/__init__.py b/packages/modules/devices/idm/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/packages/modules/devices/idm/idm/__init__.py b/packages/modules/devices/idm/idm/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/packages/modules/devices/idm/idm/config.py b/packages/modules/devices/idm/idm/config.py new file mode 100644 index 0000000000..99edcdddc8 --- /dev/null +++ b/packages/modules/devices/idm/idm/config.py @@ -0,0 +1,42 @@ +from typing import Optional +from helpermodules.auto_str import auto_str +from modules.common.component_setup import ComponentSetup +from ..vendor import vendor_descriptor + + +@auto_str +class IDMConfiguration: + def __init__(self, ip_address: Optional[str] = None, port: int = 502, modbus_id: int = 1): + self.ip_address = ip_address + self.port = port + self.modbus_id = modbus_id + + +@auto_str +class IDM: + def __init__(self, + name: str = "IDM", + type: str = "idm", + id: int = 0, + configuration: IDMConfiguration = None) -> None: + self.name = name + self.type = type + self.vendor = vendor_descriptor.configuration_factory().type + self.id = id + self.configuration = configuration or IDMConfiguration() + + +@auto_str +class IDMCounterConfiguration: + def __init__(self): + pass + + +@auto_str +class IDMCounterSetup(ComponentSetup[IDMCounterConfiguration]): + def __init__(self, + name: str = "IDM Zähler", + type: str = "counter", + id: int = 0, + configuration: IDMCounterConfiguration = None) -> None: + super().__init__(name, type, id, configuration or IDMCounterConfiguration()) diff --git a/packages/modules/devices/idm/idm/counter.py b/packages/modules/devices/idm/idm/counter.py new file mode 100644 index 0000000000..44ff2ab5b9 --- /dev/null +++ b/packages/modules/devices/idm/idm/counter.py @@ -0,0 +1,48 @@ +#!/usr/bin/env python3 +from typing import TypedDict, Any +from pymodbus.constants import Endian + +from modules.common.abstract_device import AbstractCounter +from modules.common.component_state import CounterState +from modules.common.component_type import ComponentDescriptor +from modules.common.fault_state import ComponentInfo, FaultState +from modules.common.modbus import ModbusDataType, ModbusTcpClient_ +from modules.common.simcount import SimCounter +from modules.common.store import get_counter_value_store +from modules.devices.idm.idm.config import IDMCounterSetup + + +class KwargsDict(TypedDict): + device_id: int + client: ModbusTcpClient_ + modbus_id: int + + +class IDMCounter(AbstractCounter): + def __init__(self, component_config: IDMCounterSetup, **kwargs: Any) -> None: + self.component_config = component_config + self.kwargs: KwargsDict = kwargs + + def initialize(self) -> None: + self.__device_id: int = self.kwargs['device_id'] + self.client: ModbusTcpClient_ = self.kwargs['client'] + self.modbus_id: int = self.kwargs['modbus_id'] + self.sim_counter = SimCounter(self.__device_id, self.component_config.id, prefix="bezug") + self.store = get_counter_value_store(self.component_config.id) + self.fault_state = FaultState(ComponentInfo.from_component_config(self.component_config)) + + def update(self): + unit = self.modbus_id + power = self.client.read_input_registers(4122, ModbusDataType.FLOAT_32, + wordorder=Endian.Little, unit=unit) * 1000 + imported, exported = self.sim_counter.sim_count(power) + + counter_state = CounterState( + imported=imported, + exported=exported, + power=power, + ) + self.store.set(counter_state) + + +component_descriptor = ComponentDescriptor(configuration_factory=IDMCounterSetup) diff --git a/packages/modules/devices/idm/idm/device.py b/packages/modules/devices/idm/idm/device.py new file mode 100644 index 0000000000..c7c2983201 --- /dev/null +++ b/packages/modules/devices/idm/idm/device.py @@ -0,0 +1,45 @@ +#!/usr/bin/env python3 +import logging +from typing import Iterable + +from modules.common.abstract_device import DeviceDescriptor +from modules.common.component_context import SingleComponentUpdateContext +from modules.common.configurable_device import ConfigurableDevice, ComponentFactoryByType, MultiComponentUpdater +from modules.common.modbus import ModbusTcpClient_ +from modules.devices.idm.idm.config import IDM, IDMCounterSetup +from modules.devices.idm.idm.counter import IDMCounter + +log = logging.getLogger(__name__) + + +def create_device(device_config: IDM): + client = None + + def create_counter_component(component_config: IDMCounterSetup): + nonlocal client + return IDMCounter(component_config, + device_id=device_config.id, + client=client, + modbus_id=device_config.configuration.modbus_id) + + def update_components(components: Iterable[IDMCounter]): + with client: + for component in components: + with SingleComponentUpdateContext(component.fault_state): + component.update() + + def initializer(): + nonlocal client + client = ModbusTcpClient_(device_config.configuration.ip_address, device_config.configuration.port) + + return ConfigurableDevice( + device_config=device_config, + initializer=initializer, + component_factory=ComponentFactoryByType( + counter=create_counter_component, + ), + component_updater=MultiComponentUpdater(update_components) + ) + + +device_descriptor = DeviceDescriptor(configuration_factory=IDM) diff --git a/packages/modules/devices/idm/vendor.py b/packages/modules/devices/idm/vendor.py new file mode 100644 index 0000000000..e2d6437048 --- /dev/null +++ b/packages/modules/devices/idm/vendor.py @@ -0,0 +1,14 @@ +from pathlib import Path + +from modules.common.abstract_device import DeviceDescriptor +from modules.devices.vendors import VendorGroup + + +class Vendor: + def __init__(self): + self.type = Path(__file__).parent.name + self.vendor = "IDM" + self.group = VendorGroup.VENDORS.value + + +vendor_descriptor = DeviceDescriptor(configuration_factory=Vendor)