Upgrade client-base to v0.2.0

Signed-off-by: Leandro Lucarella <[email protected]>

Author: llucax

mkdocs.yml

@@ -118,7 +118,7 @@ plugins:
             - https://frequenz-floss.github.io/frequenz-api-common/v0.5/objects.inv
             - https://frequenz-floss.github.io/frequenz-api-microgrid/v0.15/objects.inv
             - https://frequenz-floss.github.io/frequenz-channels-python/v1.0-pre/objects.inv
-            - https://frequenz-floss.github.io/frequenz-client-base-python/v0.2-dev/objects.inv
+            - https://frequenz-floss.github.io/frequenz-client-base-python/v0.2/objects.inv
             - https://grpc.github.io/grpc/python/objects.inv
             - https://typing-extensions.readthedocs.io/en/stable/objects.inv
   # Note this plugin must be loaded after mkdocstrings to be able to use macros

pyproject.toml

@@ -38,7 +38,7 @@ requires-python = ">= 3.11, < 4"
 dependencies = [
   "frequenz-api-microgrid >= 0.15.3, < 0.16.0",
   "frequenz-channels == 1.0.0b2",
-  "frequenz-client-base @ git+https://github.com/frequenz-floss/frequenz-client-base-python.git@03f2ed2104efc1bdad05eed68db167e87eb8b1aa",
+  "frequenz-client-base >= 0.2.0, < 0.3.0",
   "grpcio >= 1.54.2, < 2",
   "protobuf >= 4.25.3, < 5",
   "timezonefinder >= 6.2.0, < 7",

src/frequenz/client/microgrid/__init__.py

@@ -7,12 +7,6 @@
 """
 
 
-from frequenz.client.base.retry_strategy import (
-    ExponentialBackoff,
-    LinearBackoff,
-    RetryStrategy,
-)
-
 from ._client import ApiClient
 from ._component import Component
 from ._component_data import (
@@ -35,11 +29,8 @@
     "EVChargerCableState",
     "EVChargerComponentState",
     "EVChargerData",
-    "ExponentialBackoff",
     "InverterData",
-    "LinearBackoff",
     "Location",
     "Metadata",
     "MeterData",
-    "RetryStrategy",
 ]

src/frequenz/client/microgrid/_client.py

@@ -30,8 +30,7 @@
 
 # pylint: enable=no-name-in-module
 from frequenz.channels import Receiver
-from frequenz.client.base.grpc_streaming_helper import GrpcStreamingHelper
-from frequenz.client.base.retry_strategy import LinearBackoff, RetryStrategy
+from frequenz.client.base import retry, streaming
 from google.protobuf.empty_pb2 import Empty  # pylint: disable=no-name-in-module
 
 from ._component import (
@@ -68,7 +67,7 @@ def __init__(
         self,
         grpc_channel: grpc.aio.Channel,
         target: str,
-        retry_spec: RetryStrategy = LinearBackoff(),
+        retry_spec: retry.Strategy = retry.LinearBackoff(),
     ) -> None:
         """Initialize the class instance.
 
@@ -85,7 +84,7 @@ def __init__(
         self.api = MicrogridStub(grpc_channel)
         """The gRPC stub for the microgrid API."""
 
-        self._broadcasters: dict[int, GrpcStreamingHelper[Any, Any]] = {}
+        self._broadcasters: dict[int, streaming.GrpcStreamBroadcaster[Any, Any]] = {}
         self._retry_spec = retry_spec
 
     async def components(self) -> Iterable[Component]:
@@ -264,7 +263,7 @@ async def _new_component_data_receiver(
 
         broadcaster = self._broadcasters.setdefault(
             component_id,
-            GrpcStreamingHelper(
+            streaming.GrpcStreamBroadcaster(
                 f"raw-component-data-{component_id}",
                 # We need to cast here because grpc says StreamComponentData is
                 # a grpc.CallIterator[PbComponentData], not a

tests/test_client.py

@@ -18,10 +18,11 @@
 from frequenz.api.microgrid.microgrid_pb2 import ConnectionFilter as PbConnectionFilter
 from frequenz.api.microgrid.microgrid_pb2 import ConnectionList as PbConnectionList
 from frequenz.api.microgrid.microgrid_pb2 import SetBoundsParam as PbSetBoundsParam
-from frequenz.client.base.retry_strategy import LinearBackoff
-from google.protobuf.empty_pb2 import Empty
 
 # pylint: enable=no-name-in-module
+from frequenz.client.base.retry import LinearBackoff
+from google.protobuf.empty_pb2 import Empty  # pylint: disable=no-name-in-module
+
 from frequenz.client.microgrid import _client as client
 from frequenz.client.microgrid._component import (
     Component,