Add metadata to Components

This commit introduces metadata to Components by incorporating
GridMetadata, which now includes the maximum grid current,
specified in Amperes.

Signed-off-by: Tiyash Basu <[email protected]>

Author: tiyash-basu-frequenz

src/frequenz/sdk/microgrid/client/_client.py

@@ -36,6 +36,7 @@
 )
 from ..component._component import (
     _component_category_from_protobuf,
+    _component_metadata_from_protobuf,
     _component_type_from_protobuf,
 )
 from ._connection import Connection
@@ -254,6 +255,7 @@ async def components(self) -> Iterable[Component]:
                 c.id,
                 _component_category_from_protobuf(c.category),
                 _component_type_from_protobuf(c.category, c.inverter),
+                _component_metadata_from_protobuf(c.category, c.grid),
             ),
             components_only,
         )

src/frequenz/sdk/microgrid/component/__init__.py

@@ -6,7 +6,14 @@
 This package provides classes to operate con microgrid components.
 """
 
-from ._component import Component, ComponentCategory, ComponentMetricId, InverterType
+from ._component import (
+    Component,
+    ComponentCategory,
+    ComponentMetadata,
+    ComponentMetricId,
+    GridMetadata,
+    InverterType,
+)
 from ._component_data import (
     BatteryData,
     ComponentData,
@@ -21,10 +28,12 @@
     "Component",
     "ComponentData",
     "ComponentCategory",
+    "ComponentMetadata",
     "ComponentMetricId",
     "EVChargerCableState",
     "EVChargerComponentState",
     "EVChargerData",
+    "GridMetadata",
     "InverterData",
     "InverterType",
     "MeterData",

src/frequenz/sdk/microgrid/component/_component.py

@@ -5,11 +5,13 @@
 
 from __future__ import annotations
 
+import typing
 from dataclasses import dataclass
 from enum import Enum
 from typing import Optional
 
 import frequenz.api.common.components_pb2 as components_pb
+import frequenz.api.microgrid.grid_pb2 as grid_pb
 import frequenz.api.microgrid.inverter_pb2 as inverter_pb
 
 
@@ -106,13 +108,64 @@ def _component_category_from_protobuf(
     return ComponentCategory(component_category)
 
 
+class ComponentMetadata:
+    """Base class for component metadata classes."""
+
+
+class GridMetadata(ComponentMetadata):
+    """Metadata for a grid connection point."""
+
+    def __init__(self, max_current: float) -> None:
+        """Create a new instance.
+
+        Args:
+            max_current: maximum current rating of the grid connection point in amps.
+        """
+        self._max_current = max_current
+
+    @property
+    def max_current(self) -> float:
+        """Get the maximum current rating of the grid connection point.
+
+        Returns:
+            Maximum current rating in amps.
+        """
+        return self._max_current
+
+    def __eq__(self, other: typing.Any) -> bool:
+        """Check if this instance is equal to another.
+
+        Args:
+            other: object to compare to.
+
+        Returns:
+            `True` if `other` is a `GridMetadata` instance and `max_current` is equal,
+                `False` otherwise.
+        """
+        if not isinstance(other, GridMetadata):
+            return False
+
+        return self.max_current == other.max_current
+
+
+def _component_metadata_from_protobuf(
+    component_category: components_pb.ComponentCategory.ValueType,
+    component_metadata: grid_pb.Metadata,
+) -> Optional[GridMetadata]:
+    if component_category == components_pb.ComponentCategory.COMPONENT_CATEGORY_GRID:
+        return GridMetadata(float(component_metadata.rated_fuse_current))
+
+    return None
+
+
 @dataclass(frozen=True)
 class Component:
     """Metadata for a single microgrid component."""
 
     component_id: int
     category: ComponentCategory
     type: Optional[ComponentType] = None
+    metadata: Optional[ComponentMetadata] = None
 
     def is_valid(self) -> bool:
         """Check if this instance contains valid data.

tests/microgrid/test_client.py

@@ -21,6 +21,7 @@
     Component,
     ComponentCategory,
     EVChargerData,
+    GridMetadata,
     InverterData,
     InverterType,
     MeterData,
@@ -117,9 +118,12 @@ async def test_components(self) -> None:
                     (999, components_pb.ComponentCategory.COMPONENT_CATEGORY_SENSOR),
                 ]
             )
+
             assert set(await microgrid.components()) == {
                 Component(100, ComponentCategory.NONE),
-                Component(101, ComponentCategory.GRID),
+                Component(
+                    101, ComponentCategory.GRID, None, GridMetadata(max_current=0.0)
+                ),
                 Component(104, ComponentCategory.METER),
                 Component(105, ComponentCategory.INVERTER, InverterType.NONE),
                 Component(106, ComponentCategory.BATTERY),

tests/microgrid/test_graph.py

@@ -18,7 +18,12 @@
 
 import frequenz.sdk.microgrid._graph as gr
 from frequenz.sdk.microgrid.client import Connection, MicrogridGrpcClient
-from frequenz.sdk.microgrid.component import Component, ComponentCategory, InverterType
+from frequenz.sdk.microgrid.component import (
+    Component,
+    ComponentCategory,
+    GridMetadata,
+    InverterType,
+)
 
 from .mock_api import MockGrpcServer, MockMicrogridServicer
 
@@ -817,7 +822,7 @@ async def test_refresh_from_api(self) -> None:
         await graph.refresh_from_api(client)
 
         expected = {
-            Component(101, ComponentCategory.GRID),
+            Component(101, ComponentCategory.GRID, None, GridMetadata(max_current=0.0)),
             Component(111, ComponentCategory.METER),
             Component(131, ComponentCategory.EV_CHARGER),
         }
@@ -843,7 +848,7 @@ async def test_refresh_from_api(self) -> None:
         servicer.set_connections([(707, 717), (717, 727), (727, 737), (717, 747)])
         await graph.refresh_from_api(client)
         expected = {
-            Component(707, ComponentCategory.GRID),
+            Component(707, ComponentCategory.GRID, None, GridMetadata(max_current=0.0)),
             Component(717, ComponentCategory.METER),
             Component(727, ComponentCategory.INVERTER, InverterType.NONE),
             Component(737, ComponentCategory.BATTERY),
@@ -1146,7 +1151,7 @@ def test__validate_grid_endpoint(self) -> None:
             gr.InvalidGraphError,
             match=r"Grid endpoint 1 has graph predecessors: \[Component"
             r"\(component_id=99, category=<ComponentCategory.METER: 2>, "
-            r"type=None\)\]",
+            r"type=None, metadata=None\)\]",
         ) as _err_predecessors:
             graph._validate_grid_endpoint()