remove optional env vars from critical path (#105)

cowprotocol · Jul 15, 2024 · a54f468 · a54f468
1 parent d3b054d
commit a54f468
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main.py b/src/main.py
@@ -47,9 +47,7 @@ def __init__(self) -> None:
 
 if __name__ == "__main__":
     load_dotenv()
-    volume_path = Path(os.environ["VOLUME_PATH"])
     args = ScriptArgs()
-    aws = AWSClient.new_from_environment()
     dune = DuneClient(os.environ["DUNE_API_KEY"])
     orderbook = OrderbookFetcher()
 
@@ -65,13 +63,17 @@ def __init__(self) -> None:
             )
         )
     elif args.sync_table == SyncTable.ORDER_REWARDS:
+        aws = AWSClient.new_from_environment()
+        volume_path = Path(os.environ["VOLUME_PATH"])
         sync_order_rewards(
             aws,
             config=SyncConfig(volume_path),
             fetcher=orderbook,
             dry_run=args.dry_run,
         )
     elif args.sync_table == SyncTable.BATCH_REWARDS:
+        aws = AWSClient.new_from_environment()
+        volume_path = Path(os.environ["VOLUME_PATH"])
         sync_batch_rewards(
             aws,
             config=SyncConfig(volume_path),