Add EVChargerPool.new_bounds_sender method for streaming bounds

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/timeseries/ev_charger_pool/__init__.py

@@ -4,9 +4,11 @@
 """Interactions with EV Chargers."""
 
 from ._ev_charger_pool import EVChargerData, EVChargerPool, EVChargerPoolError
+from ._set_current_bounds import ComponentCurrentLimit
 from ._state_tracker import EVChargerState
 
 __all__ = [
+    "ComponentCurrentLimit",
     "EVChargerPool",
     "EVChargerData",
     "EVChargerPoolError",

src/frequenz/sdk/timeseries/ev_charger_pool/_ev_charger_pool.py

@@ -25,7 +25,7 @@
     EVChargerPowerFormula,
     FormulaGeneratorConfig,
 )
-from ._set_current_bounds import BoundsSetter
+from ._set_current_bounds import BoundsSetter, ComponentCurrentLimit
 from ._state_tracker import EVChargerState, StateTracker
 
 logger = logging.getLogger(__name__)
@@ -179,6 +179,19 @@ async def set_bounds(self, component_id: int, max_amps: float) -> None:
             self._bounds_setter = BoundsSetter(self._repeat_interval)
         await self._bounds_setter.set(component_id, max_amps)
 
+    def new_bounds_sender(self) -> Sender[ComponentCurrentLimit]:
+        """Return a `Sender` for setting EV Charger current bounds with.
+
+        Bounds are used to limit the max current drawn by an EV, although the exact
+        value will be determined by the EV.
+
+        Returns:
+            A new `Sender`.
+        """
+        if not self._bounds_setter:
+            self._bounds_setter = BoundsSetter(self._repeat_interval)
+        return self._bounds_setter.new_bounds_sender()
+
     async def stop(self) -> None:
         """Stop all tasks and channels owned by the EVChargerPool."""
         if self._bounds_setter:

src/frequenz/sdk/timeseries/ev_charger_pool/_set_current_bounds.py

@@ -9,7 +9,7 @@
 from datetime import timedelta
 from typing import Dict
 
-from frequenz.channels import Broadcast
+from frequenz.channels import Broadcast, Sender
 from frequenz.channels.util import Select, Timer
 
 from ..._internal.asyncio import cancel_and_await
@@ -59,6 +59,14 @@ async def set(self, component_id: int, max_amps: float) -> None:
         """
         await self._bounds_tx.send(ComponentCurrentLimit(component_id, max_amps))
 
+    def new_bounds_sender(self) -> Sender[ComponentCurrentLimit]:
+        """Return a `Sender` for setting EV Charger current bounds with.
+
+        Returns:
+            A new `Sender`.
+        """
+        return self._bounds_chan.new_sender()
+
     async def stop(self) -> None:
         """Stop the BoundsSetter."""
         await self._bounds_chan.close()