Add CLI commands to cancel gridpool orders

Commands are added to cancel an individual order by its ID or all orders
for a particular gridpool.

Signed-off-by: cwasicki <[email protected]>

Author: cwasicki

src/frequenz/client/electricity_trading/cli/__main__.py

@@ -10,6 +10,9 @@
 import click
 
 from frequenz.client.electricity_trading.cli.day_ahead import list_day_ahead_prices
+from frequenz.client.electricity_trading.cli.etrading import (
+    cancel_order as run_cancel_order,
+)
 from frequenz.client.electricity_trading.cli.etrading import (
     create_order as run_create_order,
 )
@@ -102,6 +105,25 @@ def create_order(
     )
 
 
+@cli.command()
+@click.option("--url", required=True, type=str)
+@click.option("--key", required=True, type=str)
+@click.option("--gid", required=True, type=int)
+@click.option("--order", required=True, type=int)
+def cancel_order(url: str, key: str, gid: int, order: int) -> None:
+    """Cancel an order."""
+    asyncio.run(run_cancel_order(url=url, key=key, gridpool_id=gid, order_id=order))
+
+
+@cli.command()
+@click.option("--url", required=True, type=str)
+@click.option("--key", required=True, type=str)
+@click.option("--gid", required=True, type=int)
+def cancel_all_orders(url: str, key: str, gid: int) -> None:
+    """Cancel all orders for a gridpool ID."""
+    asyncio.run(run_cancel_order(url=url, key=key, gridpool_id=gid, order_id=None))
+
+
 @cli.command()
 @click.option("--entsoe-key", required=True, type=str)
 @click.option("--start", default=midnight(), type=iso)

src/frequenz/client/electricity_trading/cli/etrading.py

@@ -172,6 +172,26 @@ async def create_order(
     print_order(order)
 
 
+async def cancel_order(
+    url: str, key: str, *, gridpool_id: int, order_id: int | None
+) -> None:
+    """Cancel an order by order ID.
+
+    If order_id is None, cancel all orders in the gridpool.
+
+    Args:
+        url: URL of the trading API.
+        key: API key.
+        gridpool_id: Gridpool ID.
+        order_id: Order ID to cancel or None to cancel all orders.
+    """
+    client = Client(server_url=url, auth_key=key)
+    if order_id is None:
+        await client.cancel_all_gridpool_orders(gridpool_id)
+    else:
+        await client.cancel_gridpool_order(gridpool_id, order_id)
+
+
 def print_trade_header() -> None:
     """Print trade header in CSV format."""
     header = (