WIP

Author: llucax

src/frequenz/client/microgrid/_client.py

@@ -118,11 +118,12 @@ async def get_microgrid_info(  # noqa: DOC502 (raises ApiClientError indirectly)
                 most likely a subclass of
                 [GrpcError][frequenz.client.microgrid.GrpcError].
         """
+        print(f"CALLED get_microgrid_info(): {self.stub.GetMicrogridMetadata=}")
         microgrid = await client.call_stub_method(
             self,
             lambda: self.stub.GetMicrogridMetadata(
                 Empty(),
-                timeout=int(DEFAULT_GRPC_CALL_TIMEOUT),
+                timeout=DEFAULT_GRPC_CALL_TIMEOUT,
             ),
             method_name="GetMicrogridMetadata",
         )
@@ -177,7 +178,7 @@ async def list_components(  # noqa: DOC502 (raises ApiClientError indirectly)
                     component_ids=map(_get_component_id, components),
                     categories=map(_get_category_value, categories),
                 ),
-                timeout=int(DEFAULT_GRPC_CALL_TIMEOUT),
+                timeout=DEFAULT_GRPC_CALL_TIMEOUT,
             ),
             method_name="ListComponents",
         )
@@ -231,7 +232,7 @@ async def list_connections(  # noqa: DOC502 (raises ApiClientError indirectly)
                     starts=map(_get_component_id, sources),
                     ends=map(_get_component_id, destinations),
                 ),
-                timeout=int(DEFAULT_GRPC_CALL_TIMEOUT),
+                timeout=DEFAULT_GRPC_CALL_TIMEOUT,
             ),
             method_name="ListConnections",
         )
@@ -307,7 +308,7 @@ async def set_component_power_active(  # noqa: DOC502 (raises ApiClientError ind
                     power=power,
                     request_lifetime=lifetime_seconds,
                 ),
-                timeout=int(DEFAULT_GRPC_CALL_TIMEOUT),
+                timeout=DEFAULT_GRPC_CALL_TIMEOUT,
             ),
             method_name="SetComponentPowerActive",
         )
@@ -390,7 +391,7 @@ async def set_component_power_reactive(  # noqa: DOC502 (raises ApiClientError i
                     power=power,
                     request_lifetime=lifetime_seconds,
                 ),
-                timeout=int(DEFAULT_GRPC_CALL_TIMEOUT),
+                timeout=DEFAULT_GRPC_CALL_TIMEOUT,
             ),
             method_name="SetComponentPowerReactive",
         )
@@ -487,7 +488,7 @@ async def add_component_bounds(  # noqa: DOC502 (Raises ApiClientError indirectl
                     ),
                     **extra_args,
                 ),
-                timeout=int(DEFAULT_GRPC_CALL_TIMEOUT),
+                timeout=DEFAULT_GRPC_CALL_TIMEOUT,
             ),
             method_name="AddComponentBounds",
         )
@@ -498,7 +499,7 @@ async def add_component_bounds(  # noqa: DOC502 (Raises ApiClientError indirectl
         return None
 
     # noqa: DOC502 (Raises ApiClientError indirectly)
-    async def receive_component_data_samples_stream(
+    def receive_component_data_samples_stream(
         self,
         component: ComponentId | Component,
         metrics: Iterable[Metric | int],
@@ -532,6 +533,7 @@ async def receive_component_data_samples_stream(
         Returns:
             The data stream from the component.
         """
+        print(f"CALLED receive_component_data_samples_stream({component=}, {metrics=})")
         component_id = _get_component_id(component)
         metrics_set = frozenset([_get_metric_value(m) for m in metrics])
         key = f"{component_id}-{hash(metrics_set)}"
@@ -542,6 +544,23 @@ async def receive_component_data_samples_stream(
             create_filter = (
                 microgrid_pb2.ReceiveComponentDataStreamRequest.ComponentDataStreamFilter
             )
+            call = self.stub.ReceiveComponentDataStream
+            # res = self.stub.ReceiveComponentDataStream(
+            #     microgrid_pb2.ReceiveComponentDataStreamRequest(
+            #         component_id=_get_component_id(component),
+            #         filter=create_filter(metrics=metrics_set),
+            #     ),
+            #     timeout=DEFAULT_GRPC_CALL_TIMEOUT,
+            # )
+            print(f"Creating new broadcaster for {call=}")
+            # result = self.stub.ReceiveComponentDataStream(
+            #     microgrid_pb2.ReceiveComponentDataStreamRequest(
+            #         component_id=_get_component_id(component),
+            #         filter=create_filter(metrics=metrics_set),
+            #     ),
+            #     timeout=DEFAULT_GRPC_CALL_TIMEOUT,
+            # )
+            # print(f"STREAM RESULT >>>>>  {result=}")
             broadcaster = streaming.GrpcStreamBroadcaster(
                 stream_name,
                 lambda: aiter(
@@ -550,7 +569,7 @@ async def receive_component_data_samples_stream(
                             component_id=_get_component_id(component),
                             filter=create_filter(metrics=metrics_set),
                         ),
-                        timeout=int(DEFAULT_GRPC_CALL_TIMEOUT),
+                        timeout=DEFAULT_GRPC_CALL_TIMEOUT,
                     )
                 ),
                 lambda msg: component_data_samples_from_proto(msg.data),

src/frequenz/client/microgrid/component/_base.py

@@ -43,7 +43,7 @@ class Component:  # pylint: disable=too-many-instance-attributes
         and in case some low level code needs to know the category of a component.
         """
 
-    status: ComponentStatus | int = ComponentStatus.ACTIVE
+    status: ComponentStatus | int = ComponentStatus.UNSPECIFIED
     """The status of this component.
 
     Note: