small fixes

Author: harisang

src/fetch/orderbook.py

@@ -45,7 +45,9 @@ class OrderbookFetcher:
     def _pg_engine(db_env: OrderbookEnv) -> Engine:
         """Returns a connection to postgres database"""
         load_dotenv()
-        db_url = os.environ[f"{db_env}_DB_URL"] + "/" + os.environ["NETWORK"]
+        db_url = (
+            os.environ[f"{db_env}_DB_URL"] + "/" + os.environ.get("NETWORK", "mainnet")
+        )
         db_string = f"postgresql+psycopg2://{db_url}"
         return create_engine(db_string)
 

src/main.py

@@ -64,7 +64,16 @@ def main() -> None:
         request_timeout=float(os.environ.get("DUNE_API_REQUEST_TIMEOUT", 10)),
     )
     orderbook = OrderbookFetcher()
-    web3 = Web3(Web3.HTTPProvider(os.environ.get("NODE_URL")))
+    network = os.environ.get("NETWORK", "mainnet")
+    if network == "mainnet":
+        node_suffix = "MAINNET"
+    else:
+        if network == "xdai":
+            node_suffix = "GNOSIS"
+        else:
+            if network == "arbitrum-one":
+                node_suffix = "ARBITRUM"
+    web3 = Web3(Web3.HTTPProvider(os.environ.get("NODE_URL" + "_" + node_suffix)))
 
     if args.sync_table == SyncTable.APP_DATA:
         table = os.environ["APP_DATA_TARGET_TABLE"]