Add receive-gridpool-trades command to CLI tool

Similarly to list-gridpool-orders this list trades for a given gridpool
ID and optionally contract delivery start.

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

Author: cwasicki

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

@@ -16,6 +16,9 @@
 from frequenz.client.electricity_trading.cli.etrading import (
     create_order as run_create_order,
 )
+from frequenz.client.electricity_trading.cli.etrading import (
+    list_gridpool_trades as run_list_gridpool_trades,
+)
 from frequenz.client.electricity_trading.cli.etrading import (
     list_orders as run_list_orders,
 )
@@ -50,6 +53,18 @@ def receive_trades(url: str, key: str, *, start: datetime) -> None:
     asyncio.run(run_list_trades(url=url, key=key, delivery_start=start))
 
 
+@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("--start", default=None, type=iso)
+def receive_gridpool_trades(url: str, key: str, gid: int, *, start: datetime) -> None:
+    """List and/or stream orders."""
+    asyncio.run(
+        run_list_gridpool_trades(url=url, key=key, gid=gid, delivery_start=start)
+    )
+
+
 @cli.command()
 @click.option("--url", required=True, type=str)
 @click.option("--key", required=True, type=str)

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

@@ -21,6 +21,7 @@
     Power,
     Price,
     PublicTrade,
+    Trade,
 )
 
 
@@ -77,6 +78,44 @@ async def list_trades(url: str, key: str, *, delivery_start: datetime) -> None:
         print_public_trade(trade)
 
 
+async def list_gridpool_trades(
+    url: str, key: str, gid: int, *, delivery_start: datetime
+) -> None:
+    """List gridpool trades and stream new gridpool trades.
+
+    Optionally a delivery_start can be provided to filter the trades by delivery period.
+
+    Args:
+        url: URL of the trading API.
+        key: API key.
+        gid: Gridpool ID.
+        delivery_start: Start of the delivery period or None.
+    """
+    client = Client(server_url=url, auth_key=key)
+
+    print_trade_header()
+
+    delivery_period = None
+    # If delivery period is selected, list historical trades also
+    if delivery_start is not None:
+        check_delivery_start(delivery_start)
+        delivery_period = DeliveryPeriod(
+            start=delivery_start,
+            duration=timedelta(minutes=15),
+        )
+    lst = client.list_gridpool_trades(gid, delivery_period=delivery_period)
+
+    async for trade in lst:
+        print_trade(trade)
+
+    if delivery_start and delivery_start <= datetime.now(timezone.utc):
+        return
+
+    stream = await client.stream_gridpool_trades(gid, delivery_period=delivery_period)
+    async for trade in stream:
+        print_trade(trade)
+
+
 async def list_orders(
     url: str, key: str, *, delivery_start: datetime, gid: int
 ) -> None:
@@ -235,6 +274,44 @@ def print_public_trade(trade: PublicTrade) -> None:
     print(",".join(v.name if isinstance(v, Enum) else str(v) for v in values))
 
 
+def print_trade_header() -> None:
+    """Print trade header in CSV format."""
+    header = (
+        "trade_id,"
+        "order_id,"
+        "execution_time,"
+        "delivery_period_start,"
+        "delivery_period_duration,"
+        "delivery_area_code,"
+        "delivery_area_code_type,"
+        "side,"
+        "quantity_mw,"
+        "currency,"
+        "price,"
+        "state "
+    )
+    print(header)
+
+
+def print_trade(trade: Trade) -> None:
+    """Print trade details to stdout in CSV format."""
+    values = (
+        trade.id,
+        trade.order_id,
+        trade.execution_time.isoformat(),
+        trade.delivery_period.start.isoformat(),
+        trade.delivery_period.duration,
+        trade.delivery_area.code,
+        trade.delivery_area.code_type,
+        trade.side,
+        trade.quantity.mw,
+        trade.price.currency,
+        trade.price.amount,
+        trade.state,
+    )
+    print(",".join(v.name if isinstance(v, Enum) else str(v) for v in values))
+
+
 def print_order_header() -> None:
     """Print order header in CSV format."""
     header = (