Skip to content

Commit 8972624

Browse files
committed
Accept ElectricalComponentId in client methods, instead of ints
Signed-off-by: Sahas Subramanian <[email protected]>
1 parent 24da2a0 commit 8972624

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/frequenz/client/assets/_client.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from frequenz.client.base import channel
1616
from frequenz.client.base.client import BaseApiClient, call_stub_method
1717
from frequenz.client.common.microgrid import MicrogridId
18+
from frequenz.client.common.microgrid.electrical_components import ElectricalComponentId
1819

1920
from ._microgrid import Microgrid
2021
from ._microgrid_proto import microgrid_from_proto
@@ -144,8 +145,8 @@ async def list_microgrid_electrical_components(
144145
async def list_microgrid_electrical_component_connections(
145146
self,
146147
microgrid_id: MicrogridId,
147-
source_component_ids: Iterable[int] = (),
148-
destination_component_ids: Iterable[int] = (),
148+
source_component_ids: Iterable[ElectricalComponentId] = (),
149+
destination_component_ids: Iterable[ElectricalComponentId] = (),
149150
) -> list[ComponentConnection | None]:
150151
"""
151152
Get the electrical component connections of a microgrid.
@@ -163,8 +164,8 @@ async def list_microgrid_electrical_component_connections(
163164
"""
164165
request = assets_pb2.ListMicrogridElectricalComponentConnectionsRequest(
165166
microgrid_id=int(microgrid_id),
166-
source_component_ids=source_component_ids,
167-
destination_component_ids=destination_component_ids,
167+
source_component_ids=(int(c) for c in source_component_ids),
168+
destination_component_ids=(int(c) for c in destination_component_ids),
168169
)
169170

170171
response = await call_stub_method(

0 commit comments

Comments
 (0)