refactor: Improve error handling and retries for token price API request

app_value_locked.py

@@ -5,6 +5,10 @@ from datetime import datetime, timedelta
 from typing import List, Dict, Optional
 import pandas as pd
 import time 
+import requests
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
+import time
 
 ADDRESSES = {
     "optimism": {
@@ -238,37 +242,49 @@ def get_token_decimals(web3, token_address):
     ])
     return token_contract.functions.decimals().call()
 
-def get_token_price_usd(chain, token_address, retries=3, delay=5):
-    """Fetch the USD price for a given ERC-20 token using CoinGecko API with retries and error handling."""
+
+
+def requests_retry_session(
+    retries=3,
+    backoff_factor=0.3,
+    status_forcelist=(500, 502, 504),
+    session=None,
+):
+    session = session or requests.Session()
+    retry = Retry(
+        total=retries,
+        read=retries,
+        connect=retries,
+        backoff_factor=backoff_factor,
+        status_forcelist=status_forcelist,
+    )
+    adapter = HTTPAdapter(max_retries=retry)
+    session.mount('http://', adapter)
+    session.mount('https://', adapter)
+    return session
+
+def get_token_price_usd(chain, token_address):
     chain_dict = {"optimism": "optimistic-ethereum", "base": "base", "ethereum": "ethereum"}
-    chain_name = chain_dict.get(chain, chain)
-    
-    for attempt in range(retries):
-        try:
-            time.sleep(4)  # Initial delay to avoid hitting rate limits
-            url = f"https://api.coingecko.com/api/v3/simple/token_price/{chain_name}?contract_addresses={token_address}&vs_currencies=usd"
-
-            headers = {
-                "accept": "application/json",
-                "x-cg-api-key": "CG-mf5xZnGELpSXeSqmHDLY2nNU"
-            }
-
-            response = requests.get(url, headers=headers)
-            response.raise_for_status()  # Raise an HTTPError for bad responses
-
-            data = response.json()
-
-            # Extract USD price
-            key = token_address.lower()
-            return data.get(key, {}).get('usd', None)
-
-        except requests.exceptions.RequestException as e:
-            logging.error(f"Error fetching token price for {token_address} on attempt {attempt + 1}/{retries}: {e}")
-            if attempt < retries - 1:
-                time.sleep(delay)  # Wait before retry
-            else:
-                logging.error(f"Failed to fetch token price after {retries} attempts")
-                return None  # Return None if all retry attempts fail
+    chain_name = chain_dict.get(chain, chain) 
+    time.sleep(4)
+    url = f"https://api.coingecko.com/api/v3/simple/token_price/{chain_name}?contract_addresses={token_address}&vs_currencies=usd"
+
+    headers = {
+        "accept": "application/json",
+        "x-cg-api-key": "CG-mf5xZnGELpSXeSqmHDLY2nNU"
+    }
+
+    try:
+        response = requests_retry_session().get(url, headers=headers, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+
+        # Extract USD price
+        key = token_address.lower()
+        return data.get(key, {}).get('usd', None)
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+        return None
 
 def load_existing_transactions(file_path):
     """Load existing transactions from a CSV file."""