Add support for duration=None

Signed-off-by: Mathias L. Baumann <[email protected]>

Author: Marenz

RELEASE_NOTES.md

@@ -10,7 +10,7 @@
 
 ## New Features
 
-<!-- Here goes the main new features and examples or instructions on how to use them -->
+* Added support for duration=None when creating a dispatch.
 
 ## Bug Fixes
 

src/frequenz/client/dispatch/__main__.py

@@ -239,9 +239,9 @@ def validate_reccurance(ctx: click.Context, param: click.Parameter, value: Any)
     required=True,
     type=str,
 )
-@click.argument("start-time", required=True, type=FuzzyDateTime())
-@click.argument("duration", required=True, type=FuzzyTimeDelta())
 @click.argument("selector", required=True, type=SelectorParamType())
+@click.argument("start-time", required=True, type=FuzzyDateTime())
+@click.argument("duration", required=False, type=FuzzyTimeDelta())
 @click.option("--active", "-a", type=bool, default=True)
 @click.option("--dry-run", "-d", type=bool, default=False)
 @click.option(
@@ -279,6 +279,7 @@ async def create(
 @click.argument("dispatch_id", type=int)
 @click.option("--start-time", type=FuzzyDateTime())
 @click.option("--duration", type=FuzzyTimeDelta())
+@click.option("--no-duration", is_flag=True)
 @click.option("--selector", type=SelectorParamType())
 @click.option("--active", type=bool)
 @click.option(
@@ -310,6 +311,13 @@ def skip_field(value: Any) -> bool:
     if len(new_fields) == 0:
         raise click.BadArgumentUsage("At least one field must be given to update.")
 
+    if new_fields.get("no_duration"):
+        if new_fields.get("duration"):
+            raise click.BadArgumentUsage("Cannot set both no_duration and duration.")
+        new_fields["duration"] = None  # type: ignore
+
+    new_fields.pop("no_duration")
+
     try:
         changed_dispatch = await ctx.obj["client"].update(
             microgrid_id=microgrid_id, dispatch_id=dispatch_id, new_fields=new_fields

src/frequenz/client/dispatch/_client.py

@@ -229,7 +229,7 @@ async def create(
         microgrid_id: int,
         type: str,  # pylint: disable=redefined-builtin
         start_time: datetime,
-        duration: timedelta,
+        duration: timedelta | None,
         selector: ComponentSelector,
         active: bool = True,
         dry_run: bool = False,
@@ -245,7 +245,8 @@ async def create(
             microgrid_id: The microgrid_id to create the dispatch for.
             type: User defined string to identify the dispatch type.
             start_time: The start time of the dispatch.
-            duration: The duration of the dispatch.
+            duration: The duration of the dispatch. Can be `None` for infinite
+                or no-duration dispatches (e.g. switching a component on).
             selector: The component selector for the dispatch.
             active: The active status of the dispatch.
             dry_run: The dry_run status of the dispatch.
@@ -325,7 +326,10 @@ async def update(
                 case "start_time":
                     msg.update.start_time.CopyFrom(to_timestamp(val))
                 case "duration":
-                    msg.update.duration = round(val.total_seconds())
+                    if val is None:
+                        msg.update.ClearField("duration")
+                    else:
+                        msg.update.duration = round(val.total_seconds())
                 case "selector":
                     msg.update.selector.CopyFrom(component_selector_to_protobuf(val))
                 case "is_active":

src/frequenz/client/dispatch/_internal_types.py

@@ -12,9 +12,9 @@
 from frequenz.api.dispatch.v1.dispatch_pb2 import (
     CreateMicrogridDispatchRequest as PBDispatchCreateRequest,
 )
-
-# pylint: enable=no-name-in-module
+from frequenz.api.dispatch.v1.dispatch_pb2 import DispatchData
 from google.protobuf.json_format import MessageToDict
+from google.protobuf.struct_pb2 import Struct
 
 from frequenz.client.base.conversion import to_datetime, to_timestamp
 
@@ -25,6 +25,8 @@
     component_selector_to_protobuf,
 )
 
+# pylint: enable=no-name-in-module
+
 
 # pylint: disable=too-many-instance-attributes
 @dataclass(kw_only=True)
@@ -42,8 +44,12 @@ class DispatchCreateRequest:
     start_time: datetime
     """The start time of the dispatch in UTC."""
 
-    duration: timedelta
-    """The duration of the dispatch, represented as a timedelta."""
+    duration: timedelta | None
+    """The duration of the dispatch, represented as a timedelta.
+
+    If None, the dispatch is considered to be "infinite" or "instantaneous",
+    like a command to turn on a component.
+    """
 
     selector: ComponentSelector
     """The component selector specifying which components the dispatch targets."""
@@ -78,13 +84,19 @@ def from_protobuf(
         Returns:
             The converted dispatch.
         """
+        duration = (
+            timedelta(seconds=pb_object.dispatch_data.duration)
+            if pb_object.dispatch_data.HasField("duration")
+            else None
+        )
+
         return DispatchCreateRequest(
             microgrid_id=pb_object.microgrid_id,
             type=pb_object.dispatch_data.type,
             start_time=rounded_start_time(
                 to_datetime(pb_object.dispatch_data.start_time)
             ),
-            duration=timedelta(seconds=pb_object.dispatch_data.duration),
+            duration=duration,
             selector=component_selector_from_protobuf(pb_object.dispatch_data.selector),
             active=pb_object.dispatch_data.is_active,
             dry_run=pb_object.dispatch_data.is_dry_run,
@@ -98,24 +110,24 @@ def to_protobuf(self) -> PBDispatchCreateRequest:
         Returns:
             The converted protobuf dispatch create request.
         """
-        pb_request = PBDispatchCreateRequest()
-
-        pb_request.microgrid_id = self.microgrid_id
-        pb_request.dispatch_data.type = self.type
-        pb_request.dispatch_data.start_time.CopyFrom(to_timestamp(self.start_time))
-        pb_request.dispatch_data.duration = round(self.duration.total_seconds())
-        pb_request.dispatch_data.selector.CopyFrom(
-            component_selector_to_protobuf(self.selector)
+        payload = Struct()
+        payload.update(self.payload)
+
+        return PBDispatchCreateRequest(
+            microgrid_id=self.microgrid_id,
+            dispatch_data=DispatchData(
+                type=self.type,
+                start_time=to_timestamp(self.start_time),
+                duration=(
+                    round(self.duration.total_seconds()) if self.duration else None
+                ),
+                selector=component_selector_to_protobuf(self.selector),
+                is_active=self.active,
+                is_dry_run=self.dry_run,
+                payload=payload,
+                recurrence=self.recurrence.to_protobuf() if self.recurrence else None,
+            ),
         )
-        pb_request.dispatch_data.is_active = self.active
-        pb_request.dispatch_data.is_dry_run = self.dry_run
-        pb_request.dispatch_data.payload.update(self.payload)
-        if self.recurrence:
-            pb_request.dispatch_data.recurrence.CopyFrom(self.recurrence.to_protobuf())
-        else:
-            pb_request.dispatch_data.ClearField("recurrence")
-
-        return pb_request
 
 
 def rounded_start_time(start_time: datetime) -> datetime:

src/frequenz/client/dispatch/test/_service.py

@@ -142,10 +142,16 @@ def _filter_dispatch(dispatch: Dispatch, request: PBDispatchListRequest) -> bool
                         return False
             if _filter.HasField("end_time_interval"):
                 if end_from := _filter.end_time_interval.__dict__["from"]:
-                    if dispatch.start_time + dispatch.duration < _to_dt(end_from):
+                    if (
+                        dispatch.duration
+                        and dispatch.start_time + dispatch.duration < _to_dt(end_from)
+                    ):
                         return False
                 if end_to := _filter.end_time_interval.to:
-                    if dispatch.start_time + dispatch.duration >= _to_dt(end_to):
+                    if (
+                        dispatch.duration
+                        and dispatch.start_time + dispatch.duration >= _to_dt(end_to)
+                    ):
                         return False
             if _filter.HasField("is_active"):
                 if dispatch.active != _filter.is_active:

src/frequenz/client/dispatch/test/generator.py

@@ -85,7 +85,12 @@ def generate_dispatch(self) -> Dispatch:
                 datetime.now(tz=timezone.utc)
                 + timedelta(seconds=self._rng.randint(0, 1000000))
             ),
-            duration=timedelta(seconds=self._rng.randint(0, 1000000)),
+            duration=self._rng.choice(
+                [
+                    None,
+                    timedelta(seconds=self._rng.randint(0, 1000000)),
+                ]
+            ),
             selector=self._rng.choice(  # type: ignore
                 [
                     [

src/frequenz/client/dispatch/types.py

@@ -14,9 +14,11 @@
     ComponentSelector as PBComponentSelector,
 )
 from frequenz.api.dispatch.v1.dispatch_pb2 import Dispatch as PBDispatch
+from frequenz.api.dispatch.v1.dispatch_pb2 import DispatchData, DispatchMetadata
 from frequenz.api.dispatch.v1.dispatch_pb2 import RecurrenceRule as PBRecurrenceRule
 from frequenz.api.dispatch.v1.dispatch_pb2 import StreamMicrogridDispatchesResponse
 from google.protobuf.json_format import MessageToDict
+from google.protobuf.struct_pb2 import Struct
 
 from frequenz.client.base.conversion import to_datetime, to_timestamp
 
@@ -272,7 +274,7 @@ class Dispatch:
     start_time: datetime
     """The start time of the dispatch in UTC."""
 
-    duration: timedelta
+    duration: timedelta | None
     """The duration of the dispatch, represented as a timedelta."""
 
     selector: ComponentSelector
@@ -318,7 +320,11 @@ def from_protobuf(cls, pb_object: PBDispatch) -> "Dispatch":
             create_time=to_datetime(pb_object.metadata.create_time),
             update_time=to_datetime(pb_object.metadata.modification_time),
             start_time=to_datetime(pb_object.data.start_time),
-            duration=timedelta(seconds=pb_object.data.duration),
+            duration=(
+                timedelta(seconds=pb_object.data.duration)
+                if pb_object.data.duration
+                else None
+            ),
             selector=component_selector_from_protobuf(pb_object.data.selector),
             active=pb_object.data.is_active,
             dry_run=pb_object.data.is_dry_run,
@@ -332,23 +338,28 @@ def to_protobuf(self) -> PBDispatch:
         Returns:
             The converted protobuf dispatch.
         """
-        pb_dispatch = PBDispatch()
-
-        pb_dispatch.metadata.dispatch_id = self.id
-        pb_dispatch.metadata.create_time.CopyFrom(to_timestamp(self.create_time))
-        pb_dispatch.metadata.modification_time.CopyFrom(to_timestamp(self.update_time))
-        pb_dispatch.data.type = self.type
-        pb_dispatch.data.start_time.CopyFrom(to_timestamp(self.start_time))
-        pb_dispatch.data.duration = int(self.duration.total_seconds())
-        pb_dispatch.data.selector.CopyFrom(
-            component_selector_to_protobuf(self.selector)
+        payload = Struct()
+        payload.update(self.payload)
+
+        return PBDispatch(
+            metadata=DispatchMetadata(
+                dispatch_id=self.id,
+                create_time=to_timestamp(self.create_time),
+                modification_time=to_timestamp(self.update_time),
+            ),
+            data=DispatchData(
+                type=self.type,
+                start_time=to_timestamp(self.start_time),
+                duration=(
+                    round(self.duration.total_seconds()) if self.duration else None
+                ),
+                selector=component_selector_to_protobuf(self.selector),
+                is_active=self.active,
+                is_dry_run=self.dry_run,
+                payload=payload,
+                recurrence=self.recurrence.to_protobuf(),
+            ),
         )
-        pb_dispatch.data.is_active = self.active
-        pb_dispatch.data.is_dry_run = self.dry_run
-        pb_dispatch.data.payload.update(self.payload)
-        pb_dispatch.data.recurrence.CopyFrom(self.recurrence.to_protobuf())
-
-        return pb_dispatch
 
 
 class Event(IntEnum):

tests/test_dispatch_cli.py

@@ -172,9 +172,9 @@ async def test_list_command(  # pylint: disable=too-many-arguments
                 "create",
                 "829",
                 "test",
+                "BATTERY",
                 "in 1 hour",
                 "1h",
-                "BATTERY",
                 "--active",
                 "False",
             ],
@@ -192,9 +192,9 @@ async def test_list_command(  # pylint: disable=too-many-arguments
                 "create",
                 "1",
                 "test",
+                "1,2,3",
                 "in 2 hours",
                 "1 hour",
-                "1,2,3",
                 "--dry-run",
                 "true",
             ],
@@ -223,9 +223,9 @@ async def test_list_command(  # pylint: disable=too-many-arguments
                 "create",
                 "1",
                 "test",
+                "CHP",
                 "in 1 hour",
                 "1h",
-                "CHP",
                 "--frequency",
                 "hourly",
                 "--interval",
@@ -274,9 +274,9 @@ async def test_list_command(  # pylint: disable=too-many-arguments
                 "create",
                 "50",
                 "test50",
+                "EV_CHARGER",
                 "in 5 hours",
                 "1h",
-                "EV_CHARGER",
                 "--frequency",
                 "daily",
                 "--until",
@@ -353,8 +353,9 @@ async def test_create_command(  # pylint: disable=too-many-arguments,too-many-lo
         (now + expected_start_time_delta).astimezone(timezone.utc).timestamp(),
         abs=2,
     )
-    assert created_dispatch.duration.total_seconds() == pytest.approx(
-        expected_duration.total_seconds(), abs=2
+    assert created_dispatch.duration and (
+        created_dispatch.duration.total_seconds()
+        == pytest.approx(expected_duration.total_seconds(), abs=2)
     )
     assert created_dispatch.selector == expected_selector
     assert created_dispatch.recurrence == expected_reccurence