Expose BatteryPool and PowerDistributingHandle from sdk.microgrid

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

Author: shsms

src/frequenz/sdk/microgrid/_data_pipeline.py

@@ -300,7 +300,7 @@ def ev_charger_pool(ev_charger_ids: set[int] | None = None) -> EVChargerPool:
     created and returned.
 
     Args:
-        ev_charger_ids: Optional set of IDs of EV Charger to be managed by the
+        ev_charger_ids: Optional set of IDs of EV Chargers to be managed by the
             EVChargerPool.
 
     Returns:
@@ -309,6 +309,31 @@ def ev_charger_pool(ev_charger_ids: set[int] | None = None) -> EVChargerPool:
     return _get().ev_charger_pool(ev_charger_ids)
 
 
+def battery_pool(battery_ids: abc.Set[int] | None = None) -> BatteryPool:
+    """Return the corresponding BatteryPool instance for the given ids.
+
+    If a BatteryPool instance for the given ids doesn't exist, a new one is
+    created and returned.
+
+    Args:
+        battery_ids: Optional set of IDs of batteries to be managed by the
+            BatteryPool.
+
+    Returns:
+        A BatteryPool instance.
+    """
+    return _get().battery_pool(battery_ids)
+
+
+def power_distributing_handle() -> Bidirectional.Handle[Request, Result]:
+    """Return the handle to the power distributing actor.
+
+    Returns:
+        A Bidirectional handle to communicate with the power distributing actor.
+    """
+    return _get().power_distributing_handle()
+
+
 def _get() -> _DataPipeline:
     if _DATA_PIPELINE is None:
         raise RuntimeError(