feat: implement sensors

- Also added mspconfig and telemetry type hints to all equipment classes

Author: cryptk

pyomnilogic_local/backyard.py

@@ -20,6 +20,8 @@
 class Backyard(OmniEquipment[MSPBackyard, TelemetryBackyard]):
     """Represents the backyard equipment in the OmniLogic system."""
 
+    mspconfig: MSPBackyard
+    telemetry: TelemetryBackyard
     bow: EquipmentDict[Bow] = EquipmentDict()
     lights: EquipmentDict[ColorLogicLight] = EquipmentDict()
     relays: EquipmentDict[Relay] = EquipmentDict()

pyomnilogic_local/bow.py

@@ -20,6 +20,8 @@
 class Bow(OmniEquipment[MSPBoW, TelemetryBoW]):
     """Represents a bow in the OmniLogic system."""
 
+    mspconfig: MSPBoW
+    telemetry: TelemetryBoW
     filters: EquipmentDict[Filter] = EquipmentDict()
     heater: Heater | None = None
     relays: EquipmentDict[Relay] = EquipmentDict()

pyomnilogic_local/chlorinator.py

@@ -5,3 +5,6 @@
 
 class Chlorinator(OmniEquipment[MSPChlorinator, TelemetryChlorinator]):
     """Represents a chlorinator in the OmniLogic system."""
+
+    mspconfig: MSPChlorinator
+    telemetry: TelemetryChlorinator

pyomnilogic_local/colorlogiclight.py

@@ -26,6 +26,9 @@
 class ColorLogicLight(OmniEquipment[MSPColorLogicLight, TelemetryColorLogicLight]):
     """Represents a color logic light."""
 
+    mspconfig: MSPColorLogicLight
+    telemetry: TelemetryColorLogicLight
+
     def __init__(self, omni: "OmniLogic", mspconfig: MSPColorLogicLight, telemetry: Telemetry) -> None:
         super().__init__(omni, mspconfig, telemetry)
 

pyomnilogic_local/csad.py

@@ -5,3 +5,6 @@
 
 class CSAD(OmniEquipment[MSPCSAD, TelemetryCSAD]):
     """Represents a CSAD in the OmniLogic system."""
+
+    mspconfig: MSPCSAD
+    telemetry: TelemetryCSAD

pyomnilogic_local/filter.py

@@ -9,6 +9,9 @@
 class Filter(OmniEquipment[MSPFilter, TelemetryFilter]):
     """Represents a filter in the OmniLogic system."""
 
+    mspconfig: MSPFilter
+    telemetry: TelemetryFilter
+
     # Expose MSPConfig attributes
     @property
     def equip_type(self) -> str:

pyomnilogic_local/heater.py

@@ -22,6 +22,8 @@ class Heater(OmniEquipment[MSPVirtualHeater, TelemetryVirtualHeater]):
     determine if conversion to Celsius should be performed for display.
     """
 
+    mspconfig: MSPVirtualHeater
+    telemetry: TelemetryVirtualHeater
     heater_equipment: EquipmentDict[HeaterEquipment] = EquipmentDict()
 
     def __init__(self, omni: "OmniLogic", mspconfig: MSPVirtualHeater, telemetry: Telemetry) -> None:

pyomnilogic_local/heater_equip.py

@@ -20,11 +20,14 @@ class HeaterEquipment(OmniEquipment[MSPHeaterEquip, TelemetryHeater]):
     Note: Temperature is always in Fahrenheit internally.
     """
 
+    mspconfig: MSPHeaterEquip
+    telemetry: TelemetryHeater
+
     def __init__(self, omni: "OmniLogic", mspconfig: MSPHeaterEquip, telemetry: Telemetry) -> None:
         super().__init__(omni, mspconfig, telemetry)
 
     @property
-    def heater_type(self) -> HeaterType | str:
+    def heater_type(self) -> HeaterType:
         """Returns the type of heater (GAS, HEAT_PUMP, SOLAR, etc.)."""
         return self.mspconfig.heater_type
 

pyomnilogic_local/pump.py

@@ -9,6 +9,9 @@
 class Pump(OmniEquipment[MSPPump, TelemetryPump]):
     """Represents a pump in the OmniLogic system."""
 
+    mspconfig: MSPPump
+    telemetry: TelemetryPump
+
     # Expose MSPConfig attributes
     @property
     def equip_type(self) -> str:

pyomnilogic_local/relay.py

@@ -14,6 +14,9 @@
 class Relay(OmniEquipment[MSPRelay, TelemetryRelay]):
     """Represents a relay in the OmniLogic system."""
 
+    mspconfig: MSPRelay
+    telemetry: TelemetryRelay
+
     def __init__(self, omni: "OmniLogic", mspconfig: MSPRelay, telemetry: Telemetry) -> None:
         super().__init__(omni, mspconfig, telemetry)