use ETH_RPC

.env.sample

@@ -3,6 +3,9 @@ DUNE_USER=
 DUNE_PASSWORD=
 DUNE_QUERY_ID=
 
+GNOSIS_RPC=
+ETH_RPC=
+
 # Official Dune API
 DUNE_API_KEY=
 

src/constants.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 
 
@@ -17,3 +18,6 @@
         "type": "function",
     }
 ]
+
+ETH_RPC = os.environ.get("ETH_RPC", "https://rpc.ankr.com/eth")
+GNOSIS_RPC = os.environ.get("ETH_RPC", "https://rpc.gnosischain.com")

src/gas_saved.py

@@ -5,6 +5,7 @@
 from web3 import Web3
 from web3.types import TxReceipt
 
+from src.constants import ETH_RPC
 from src.db.pg_client import pg_engine
 
 
@@ -34,7 +35,7 @@ def main(batch_tx_hash: str) -> int:
     # Ref: https://github.com/cowprotocol/services/blob/fd5f7cf47a6afdff89b310b60b869dfc577ac7a7/crates/shared/src/price_estimation/gas.rs#L37
     df_quotes["gas_amount"] = df_quotes["gas_amount"].apply(lambda x: x - 106391)
     load_dotenv()
-    w3 = Web3(Web3.HTTPProvider("https://rpc.ankr.com/eth"))
+    w3 = Web3(Web3.HTTPProvider(ETH_RPC))
     tx: TxReceipt = w3.eth.get_transaction_receipt(HexStr(batch_tx_hash))
     gas_used = tx["gasUsed"]
     print(

src/missing_tokens.py

@@ -12,6 +12,8 @@
 
 from web3 import Web3
 
+from src.constants import ETH_RPC, GNOSIS_RPC
+
 
 class Network(Enum):
     """
@@ -28,8 +30,8 @@ def as_dune_v2_repr(self) -> str:
     def node_url(self) -> str:
         """Returns URL to Node for Network"""
         return {
-            Network.MAINNET: "https://rpc.ankr.com/eth",
-            Network.GNOSIS: "https://rpc.gnosischain.com",
+            Network.MAINNET: ETH_RPC,
+            Network.GNOSIS: GNOSIS_RPC,
         }[self]
 
     @property

src/subgraph/ens_data.py

@@ -6,7 +6,7 @@
 from dotenv import load_dotenv
 from web3 import Web3
 
-from src.constants import PUBLIC_RESOLVER_ABI
+from src.constants import PUBLIC_RESOLVER_ABI, ETH_RPC
 from src.subgraph.fetch import execute_subgraph_query
 from src.utils import partition_array
 
@@ -23,7 +23,7 @@
 }
 
 load_dotenv()
-w3 = Web3(Web3.HTTPProvider("https://rpc.ankr.com/eth"))
+w3 = Web3(Web3.HTTPProvider(ETH_RPC))
 
 
 def read_ens_text(resolver: str, node: str, key: str) -> str:

tests/test_utils.py

@@ -46,9 +46,6 @@ def test_network_class(self):
         self.assertEqual(gnosis.chain_id, 100)
         self.assertEqual(mainnet.chain_id, 1)
 
-        self.assertEqual(gnosis.node_url(), "https://rpc.gnosischain.com")
-        self.assertEqual(mainnet.node_url(), "https://rpc.ankr.com/eth")
-
         self.assertEqual(gnosis.as_dune_v2_repr(), "gnosis")
         self.assertEqual(mainnet.as_dune_v2_repr(), "ethereum")