juham-automation 0.0.19__py3-none-any.whl → 0.0.27__py3-none-any.whl

juham_automation/automation/powermeter_simulator.py

@@ -1,139 +1,139 @@
-import json
-from typing import Any, Dict, cast
-from typing_extensions import override
-
-from masterpiece.mqtt import MqttMsg
-from juham_core import Juham
-from juham_core.timeutils import timestamp
-from juham_core import MasterPieceThread, JuhamThread
-
-
-class PowerMeterSimulatorThread(MasterPieceThread):
-    """Thread simulating Energy Meter."""
-
-    _power: float = 1000.0  # W
-    _power_topic: str = "power"
-    _interval: float = 10  # 10 seconds
-
-    def __init__(self) -> None:
-        """Construct a thread for publishing power data.
-
-        Args:
-            topic (str, optional): MQTT topic to post the sensor readings. Defaults to None.
-            interval (float, optional): Interval specifying how often the sensor is read. Defaults to 60 seconds.
-        """
-        super().__init__(None)
-        self.current_ts: float = timestamp()
-
-    @classmethod
-    def initialize(cls, power_topic: str, power: float, interval: float) -> None:
-        """Initialize thread  class attributes.
-
-        Args:
-            power_topic (str): topic to publish the energy meter readings
-            power (float): power  to be simulated, the default is 1kW
-            interval (float): update interval, the default is 10s
-        """
-        cls._power = power
-        cls._interval = interval
-        cls._power_topic = power_topic
-
-    @override
-    def update_interval(self) -> float:
-        return self._interval
-
-    def publish_active_power(self, ts: float) -> None:
-        """Publish the active power, also known as real power. This is that
-        part of the  power that can be converted to useful work.
-
-        Args:
-            ts (str): time stamp of the event
-
-        """
-        dt = ts - self.current_ts
-        self.current_ts = ts
-
-        msg = {
-            "timestamp": ts,
-            "real_a": self._power * dt,
-            "real_b": self._power * dt,
-            "real_c": self._power * dt,
-            "real_total": 3 * self._power * dt,
-        }
-        self.publish(self._power_topic, json.dumps(msg), 1, True)
-
-    @override
-    def update(self) -> bool:
-        super().update()
-        self.publish_active_power(timestamp())
-        return True
-
-
-class PowerMeterSimulator(JuhamThread):
-    """Simulator energy meter sensor. Spawns a thread
-    to simulate Shelly PM mqtt messages"""
-
-    workerThreadId = PowerMeterSimulatorThread.get_class_id()
-    update_interval: float = 10
-    power: float = 1000.0
-
-    _POWERMETERSIMULATOR: str = "_powermetersimulator"
-
-    def __init__(
-        self,
-        name: str = "em",
-        interval: float = 0,
-    ) -> None:
-        """Create energy meter simulator.
-
-        Args:
-            name (str, optional): Name of the object. Defaults to 'em'.
-            topic (str, optional): MQTT topic to publish the energy meter reports. Defaults to None.
-            interval (float, optional): interval between events, in seconds. Defaults to None.
-        """
-        super().__init__(name)
-        self.update_ts: float = 0.0
-        if interval > 0.0:
-            self.update_interval = interval
-        self.power_topic = self.make_topic_name("powerconsumption")  # target topic
-
-    @override
-    def on_message(self, client: object, userdata: Any, msg: MqttMsg) -> None:
-        if msg.topic == self.power_topic:
-            em = json.loads(msg.payload.decode())
-            self.on_sensor(em)
-        else:
-            super().on_message(client, userdata, msg)
-
-    def on_sensor(self, em: dict[Any, Any]) -> None:
-        """Handle data coming from the energy meter.
-
-        Simply log the event to indicate the presense of simulated device.
-        Args:
-            em (dict): data from the sensor
-        """
-        self.debug(f"Simulated power meter sensor {em}")
-
-    @override
-    def run(self) -> None:
-        PowerMeterSimulatorThread.initialize(
-            self.power_topic, self.power, self.update_interval
-        )
-        self.worker = cast(
-            PowerMeterSimulatorThread,
-            Juham.instantiate(PowerMeterSimulatorThread.get_class_id()),
-        )
-        super().run()
-
-    @override
-    def to_dict(self) -> Dict[str, Any]:
-        data: Dict[str, Any] = super().to_dict()
-        data[self._POWERMETERSIMULATOR] = {"power_topic": self.power_topic}
-        return data
-
-    @override
-    def from_dict(self, data: Dict[str, Any]) -> None:
-        super().from_dict(data)
-        if self._POWERMETERSIMULATOR in data:
-            for key, value in data[self._POWERMETERSIMULATOR].items():
-                setattr(self, key, value)
+import json
+from typing import Any, Dict, cast
+from typing_extensions import override
+
+from masterpiece.mqtt import MqttMsg
+from juham_core import Juham
+from juham_core.timeutils import timestamp
+from juham_core import MasterPieceThread, JuhamThread
+
+
+class PowerMeterSimulatorThread(MasterPieceThread):
+    """Thread simulating Energy Meter."""
+
+    _power: float = 1000.0  # W
+    _power_topic: str = "power"
+    _interval: float = 10  # 10 seconds
+
+    def __init__(self) -> None:
+        """Construct a thread for publishing power data.
+
+        Args:
+            topic (str, optional): MQTT topic to post the sensor readings. Defaults to None.
+            interval (float, optional): Interval specifying how often the sensor is read. Defaults to 60 seconds.
+        """
+        super().__init__(None)
+        self.current_ts: float = timestamp()
+
+    @classmethod
+    def initialize(cls, power_topic: str, power: float, interval: float) -> None:
+        """Initialize thread  class attributes.
+
+        Args:
+            power_topic (str): topic to publish the energy meter readings
+            power (float): power  to be simulated, the default is 1kW
+            interval (float): update interval, the default is 10s
+        """
+        cls._power = power
+        cls._interval = interval
+        cls._power_topic = power_topic
+
+    @override
+    def update_interval(self) -> float:
+        return self._interval
+
+    def publish_active_power(self, ts: float) -> None:
+        """Publish the active power, also known as real power. This is that
+        part of the  power that can be converted to useful work.
+
+        Args:
+            ts (str): time stamp of the event
+
+        """
+        dt = ts - self.current_ts
+        self.current_ts = ts
+
+        msg = {
+            "timestamp": ts,
+            "real_a": self._power * dt,
+            "real_b": self._power * dt,
+            "real_c": self._power * dt,
+            "real_total": 3 * self._power * dt,
+        }
+        self.publish(self._power_topic, json.dumps(msg), 1, True)
+
+    @override
+    def update(self) -> bool:
+        super().update()
+        self.publish_active_power(timestamp())
+        return True
+
+
+class PowerMeterSimulator(JuhamThread):
+    """Simulator energy meter sensor. Spawns a thread
+    to simulate Shelly PM mqtt messages"""
+
+    workerThreadId = PowerMeterSimulatorThread.get_class_id()
+    update_interval: float = 10
+    power: float = 1000.0
+
+    _POWERMETERSIMULATOR: str = "_powermetersimulator"
+
+    def __init__(
+        self,
+        name: str = "em",
+        interval: float = 0,
+    ) -> None:
+        """Create energy meter simulator.
+
+        Args:
+            name (str, optional): Name of the object. Defaults to 'em'.
+            topic (str, optional): MQTT topic to publish the energy meter reports. Defaults to None.
+            interval (float, optional): interval between events, in seconds. Defaults to None.
+        """
+        super().__init__(name)
+        self.update_ts: float = 0.0
+        if interval > 0.0:
+            self.update_interval = interval
+        self.power_topic = self.make_topic_name("powerconsumption")  # target topic
+
+    @override
+    def on_message(self, client: object, userdata: Any, msg: MqttMsg) -> None:
+        if msg.topic == self.power_topic:
+            em = json.loads(msg.payload.decode())
+            self.on_sensor(em)
+        else:
+            super().on_message(client, userdata, msg)
+
+    def on_sensor(self, em: dict[Any, Any]) -> None:
+        """Handle data coming from the energy meter.
+
+        Simply log the event to indicate the presense of simulated device.
+        Args:
+            em (dict): data from the sensor
+        """
+        self.debug(f"Simulated power meter sensor {em}")
+
+    @override
+    def run(self) -> None:
+        PowerMeterSimulatorThread.initialize(
+            self.power_topic, self.power, self.update_interval
+        )
+        self.worker = cast(
+            PowerMeterSimulatorThread,
+            Juham.instantiate(PowerMeterSimulatorThread.get_class_id()),
+        )
+        super().run()
+
+    @override
+    def to_dict(self) -> Dict[str, Any]:
+        data: Dict[str, Any] = super().to_dict()
+        data[self._POWERMETERSIMULATOR] = {"power_topic": self.power_topic}
+        return data
+
+    @override
+    def from_dict(self, data: Dict[str, Any]) -> None:
+        super().from_dict(data)
+        if self._POWERMETERSIMULATOR in data:
+            for key, value in data[self._POWERMETERSIMULATOR].items():
+                setattr(self, key, value)

juham_automation/automation/spothintafi.py

@@ -1,140 +1,140 @@
-from datetime import datetime
-import time
-import json
-from typing import Any, Dict, Optional, cast
-from typing_extensions import override
-
-from masterpiece.mqtt import Mqtt, MqttMsg
-from juham_core import JuhamCloudThread, JuhamThread
-
-
-class SpotHintaFiThread(JuhamCloudThread):
-    """Thread running SpotHinta.fi.
-
-    Periodically fetches the spot electricity prices and publishes them
-    to 'spot' topic.
-    """
-
-    _spot_topic: str = ""
-    _url: str = ""
-    _interval: float = 12 * 3600
-    grid_cost_day: float = 0.0314
-    grid_cost_night: float = 0.0132
-    grid_cost_tax: float = 0.028272
-
-    def __init__(self, client: Optional[Mqtt] = None) -> None:
-        super().__init__(client)
-        self._interval = 60
-
-    def init(self, topic: str, url: str, interval: float) -> None:
-        self._spot_topic = topic
-        self._url = url
-        self._interval = interval
-
-    @override
-    def make_weburl(self) -> str:
-        return self._url
-
-    @override
-    def update_interval(self) -> float:
-        return self._interval
-
-    @override
-    def process_data(self, rawdata: Any) -> None:
-        """Publish electricity price message to Juham topic.
-
-        Args:
-            rawdata (dict): electricity prices
-        """
-
-        super().process_data(rawdata)
-        data = rawdata.json()
-
-        spot = []
-        for e in data:
-            dt = datetime.fromisoformat(e["DateTime"])  # Correct timezone handling
-            ts = int(dt.timestamp())  # Ensure integer timestamps like in the test
-
-            hour = dt.strftime("%H")  # Correctly extract hour
-
-            if 6 <= int(hour) < 22:
-                grid_cost = self.grid_cost_day
-            else:
-                grid_cost = self.grid_cost_night
-
-            total_price = round(e["PriceWithTax"] + grid_cost + self.grid_cost_tax, 6)
-            grid_cost_total = round(grid_cost + self.grid_cost_tax, 6)
-
-            h = {
-                "Timestamp": ts,
-                "hour": hour,
-                "Rank": e["Rank"],
-                "PriceWithTax": total_price,
-                "GridCost": grid_cost_total,
-            }
-            spot.append(h)
-
-        self.publish(self._spot_topic, json.dumps(spot), 1, True)
-        # self.info(f"Spot electricity prices published for the next {len(spot)} days")
-
-
-class SpotHintaFi(JuhamThread):
-    """Spot electricity price for reading hourly electricity prices from
-    https://api.spot-hinta.fi site.
-    """
-
-    _SPOTHINTAFI: str = "_spothintafi"
-    worker_thread_id = SpotHintaFiThread.get_class_id()
-    url = "https://api.spot-hinta.fi/TodayAndDayForward"
-    update_interval = 12 * 3600
-
-    def __init__(self, name: str = "rspothintafi") -> None:
-        super().__init__(name)
-        self.active_liter_lpm = -1
-        self.update_ts = None
-        self.spot_topic = self.make_topic_name("spot")
-
-    @override
-    def on_connect(self, client: object, userdata: Any, flags: int, rc: int) -> None:
-        super().on_connect(client, userdata, flags, rc)
-        if rc == 0:
-            self.subscribe(self.spot_topic)
-
-    @override
-    def on_message(self, client: object, userdata: Any, msg: MqttMsg) -> None:
-        if msg.topic == self.spot_topic:
-            em = json.loads(msg.payload.decode())
-            self.on_spot(em)
-        else:
-            super().on_message(client, userdata, msg)
-
-    def on_spot(self, m: dict[Any, Any]) -> None:
-        """Write hourly spot electricity prices to time series database.
-
-        Args:
-            m (dict): holding hourly spot electricity prices
-        """
-        pass
-
-    @override
-    def run(self) -> None:
-        self.worker = cast(SpotHintaFiThread, self.instantiate(self.worker_thread_id))
-        self.worker.init(self.spot_topic, self.url, self.update_interval)
-        super().run()
-
-    @override
-    def to_dict(self) -> Dict[str, Any]:
-        data: Dict[str, Any] = super().to_dict()
-        data[self._SPOTHINTAFI] = {
-            "topic": self.spot_topic,
-            "url": self.url,
-            "interval": self.update_interval,
-        }
-        return data
-
-    @override
-    def from_dict(self, data: Dict[str, Any]) -> None:
-        super().from_dict(data)
-        if self._SPOTHINTAFI in data:
-            for key, value in data[self._SPOTHINTAFI].items():
-                setattr(self, key, value)
+from datetime import datetime
+import time
+import json
+from typing import Any, Dict, Optional, cast
+from typing_extensions import override
+
+from masterpiece.mqtt import Mqtt, MqttMsg
+from juham_core import JuhamCloudThread, JuhamThread
+
+
+class SpotHintaFiThread(JuhamCloudThread):
+    """Thread running SpotHinta.fi.
+
+    Periodically fetches the spot electricity prices and publishes them
+    to 'spot' topic.
+    """
+
+    _spot_topic: str = ""
+    _url: str = ""
+    _interval: float = 12 * 3600
+    grid_cost_day: float = 0.0314
+    grid_cost_night: float = 0.0132
+    grid_cost_tax: float = 0.028272
+
+    def __init__(self, client: Optional[Mqtt] = None) -> None:
+        super().__init__(client)
+        self._interval = 60
+
+    def init(self, topic: str, url: str, interval: float) -> None:
+        self._spot_topic = topic
+        self._url = url
+        self._interval = interval
+
+    @override
+    def make_weburl(self) -> str:
+        return self._url
+
+    @override
+    def update_interval(self) -> float:
+        return self._interval
+
+    @override
+    def process_data(self, rawdata: Any) -> None:
+        """Publish electricity price message to Juham topic.
+
+        Args:
+            rawdata (dict): electricity prices
+        """
+
+        super().process_data(rawdata)
+        data = rawdata.json()
+
+        spot = []
+        for e in data:
+            dt = datetime.fromisoformat(e["DateTime"])  # Correct timezone handling
+            ts = int(dt.timestamp())  # Ensure integer timestamps like in the test
+
+            hour = dt.strftime("%H")  # Correctly extract hour
+
+            if 6 <= int(hour) < 22:
+                grid_cost = self.grid_cost_day
+            else:
+                grid_cost = self.grid_cost_night
+
+            total_price = round(e["PriceWithTax"] + grid_cost + self.grid_cost_tax, 6)
+            grid_cost_total = round(grid_cost + self.grid_cost_tax, 6)
+
+            h = {
+                "Timestamp": ts,
+                "hour": hour,
+                "Rank": e["Rank"],
+                "PriceWithTax": total_price,
+                "GridCost": grid_cost_total,
+            }
+            spot.append(h)
+
+        self.publish(self._spot_topic, json.dumps(spot), 1, True)
+        # self.info(f"Spot electricity prices published for the next {len(spot)} days")
+
+
+class SpotHintaFi(JuhamThread):
+    """Spot electricity price for reading hourly electricity prices from
+    https://api.spot-hinta.fi site.
+    """
+
+    _SPOTHINTAFI: str = "_spothintafi"
+    worker_thread_id = SpotHintaFiThread.get_class_id()
+    url = "https://api.spot-hinta.fi/TodayAndDayForward"
+    update_interval = 12 * 3600
+
+    def __init__(self, name: str = "rspothintafi") -> None:
+        super().__init__(name)
+        self.active_liter_lpm = -1
+        self.update_ts = None
+        self.spot_topic = self.make_topic_name("spot")
+
+    @override
+    def on_connect(self, client: object, userdata: Any, flags: int, rc: int) -> None:
+        super().on_connect(client, userdata, flags, rc)
+        if rc == 0:
+            self.subscribe(self.spot_topic)
+
+    @override
+    def on_message(self, client: object, userdata: Any, msg: MqttMsg) -> None:
+        if msg.topic == self.spot_topic:
+            em = json.loads(msg.payload.decode())
+            self.on_spot(em)
+        else:
+            super().on_message(client, userdata, msg)
+
+    def on_spot(self, m: dict[Any, Any]) -> None:
+        """Write hourly spot electricity prices to time series database.
+
+        Args:
+            m (dict): holding hourly spot electricity prices
+        """
+        pass
+
+    @override
+    def run(self) -> None:
+        self.worker = cast(SpotHintaFiThread, self.instantiate(self.worker_thread_id))
+        self.worker.init(self.spot_topic, self.url, self.update_interval)
+        super().run()
+
+    @override
+    def to_dict(self) -> Dict[str, Any]:
+        data: Dict[str, Any] = super().to_dict()
+        data[self._SPOTHINTAFI] = {
+            "topic": self.spot_topic,
+            "url": self.url,
+            "interval": self.update_interval,
+        }
+        return data
+
+    @override
+    def from_dict(self, data: Dict[str, Any]) -> None:
+        super().from_dict(data)
+        if self._SPOTHINTAFI in data:
+            for key, value in data[self._SPOTHINTAFI].items():
+                setattr(self, key, value)