Move component status tracking into _component_status sub package

There's going to be status trackers for more component types. Those
can now go into the `_component_status` package.

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

Author: shsms

src/frequenz/sdk/actor/power_distributing/_component_managers/_battery_manager.py

@@ -19,9 +19,8 @@
 from ....microgrid import connection_manager
 from ....microgrid.component import BatteryData, ComponentCategory, InverterData
 from ....timeseries._quantities import Power
-from .._battery_status_tracker import BatteryStatusTracker
 from .._component_pool_status_tracker import ComponentPoolStatusTracker
-from .._component_status import ComponentPoolStatus
+from .._component_status import BatteryStatusTracker, ComponentPoolStatus
 from .._distribution_algorithm import (
     AggregatedBatteryData,
     BatteryDistributionAlgorithm,

src/frequenz/sdk/actor/power_distributing/_component_status/__init__.py

@@ -0,0 +1,22 @@
+# License: MIT
+# Copyright © 2024 Frequenz Energy-as-a-Service GmbH
+
+"""Status tracking for components."""
+
+from ._battery_status_tracker import BatteryStatusTracker
+from ._component_status import (
+    ComponentPoolStatus,
+    ComponentStatus,
+    ComponentStatusEnum,
+    ComponentStatusTracker,
+    SetPowerResult,
+)
+
+__all__ = [
+    "BatteryStatusTracker",
+    "ComponentPoolStatus",
+    "ComponentStatus",
+    "ComponentStatusEnum",
+    "ComponentStatusTracker",
+    "SetPowerResult",
+]

…_distributing/_battery_status_tracker.py …ponent_status/_battery_status_tracker.pysrc/frequenz/sdk/actor/power_distributing/_battery_status_tracker.py renamed to src/frequenz/sdk/actor/power_distributing/_component_status/_battery_status_tracker.py src/frequenz/sdk/actor/power_distributing/_battery_status_tracker.py renamed to src/frequenz/sdk/actor/power_distributing/_component_status/_battery_status_tracker.py

@@ -20,9 +20,9 @@
 from frequenz.channels.util import Timer, select, selected_from
 from typing_extensions import override
 
-from ..._internal._asyncio import cancel_and_await
-from ...microgrid import connection_manager
-from ...microgrid.component import (
+from ...._internal._asyncio import cancel_and_await
+from ....microgrid import connection_manager
+from ....microgrid.component import (
     BatteryData,
     ComponentCategory,
     ComponentData,

…/power_distributing/_component_status.py …g/_component_status/_component_status.pysrc/frequenz/sdk/actor/power_distributing/_component_status.py renamed to src/frequenz/sdk/actor/power_distributing/_component_status/_component_status.py src/frequenz/sdk/actor/power_distributing/_component_status.py renamed to src/frequenz/sdk/actor/power_distributing/_component_status/_component_status.py

@@ -7,13 +7,13 @@
 from frequenz.channels import Broadcast
 from pytest_mock import MockerFixture
 
-from frequenz.sdk.actor.power_distributing._battery_status_tracker import (
-    BatteryStatusTracker,
-)
 from frequenz.sdk.actor.power_distributing._component_pool_status_tracker import (
     ComponentPoolStatusTracker,
 )
-from frequenz.sdk.actor.power_distributing._component_status import ComponentPoolStatus
+from frequenz.sdk.actor.power_distributing._component_status import (
+    BatteryStatusTracker,
+    ComponentPoolStatus,
+)
 from frequenz.sdk.microgrid.component import ComponentCategory
 from tests.timeseries.mock_microgrid import MockMicrogrid
 

tests/actor/test_battery_pool_status.py

@@ -29,10 +29,8 @@
 from pytest_mock import MockerFixture
 from time_machine import TimeMachineFixture
 
-from frequenz.sdk.actor.power_distributing._battery_status_tracker import (
-    BatteryStatusTracker,
-)
 from frequenz.sdk.actor.power_distributing._component_status import (
+    BatteryStatusTracker,
     ComponentStatus,
     ComponentStatusEnum,
     SetPowerResult,