diff --git a/src/deploy_prefect/flows.py b/src/deploy_prefect/flows.py index 4406c97..62b70a6 100644 --- a/src/deploy_prefect/flows.py +++ b/src/deploy_prefect/flows.py @@ -33,7 +33,5 @@ def prod_order_rewards() -> None: blockrange = get_block_range() orderbook = fetch_orderbook(blockrange) data = cast_orderbook_to_dune_string(orderbook) - table_name = upload_data_to_dune( - data, blockrange.block_from, blockrange.block_to, config - ) + table_name = upload_data_to_dune(data, blockrange.block_from, blockrange.block_to) update_aggregate_query(table_name, config) diff --git a/src/deploy_prefect/tasks.py b/src/deploy_prefect/tasks.py index 17df5ef..0896b15 100644 --- a/src/deploy_prefect/tasks.py +++ b/src/deploy_prefect/tasks.py @@ -2,6 +2,7 @@ import os from io import StringIO +from typing import Optional from datetime import datetime, timezone import requests @@ -77,13 +78,16 @@ def cast_orderbook_to_dune_string(orderbook: pd.DataFrame) -> str: @task # type: ignore[misc] def upload_data_to_dune( - data: str, block_start: int, block_end: int, config: Config + data: str, block_start: int, block_end: int, config: Optional[Config] = None ) -> str: """ Uploads the order rewards data to Dune, either creating a new query or updating an existing one """ - table_name = f"order_rewards_{config.env}_{block_start}" + if config: + table_name = f"order_rewards_{config.env}_{block_start}" + else: + table_name = f"order_rewards_{block_start}" dune = DuneClient.from_env() dune.upload_csv( # type: ignore[attr-defined] data=data,