Update app/price_fetcher.py

app/price_fetcher.py

@@ -1,7 +1,7 @@
 """
-A professional-grade, multi-oracle price engine.
-This engine fetches prices from two independent, globally-accessible
-decentralized oracles (Pyth and Chainlink) to detect market dislocations.
+A professional-grade, multi-asset, multi-oracle price engine.
+This engine dynamically fetches prices for a configured list of assets
+from decentralized oracles to detect market dislocations.
 """
 import asyncio
 import logging
@@ -10,43 +10,80 @@ import httpx
 
 logger = logging.getLogger(__name__)
 
+# --- CONFIGURATION ---
+# Define the assets we want to track and their corresponding oracle IDs.
+# Find more IDs at https://pyth.network/developers/price-feeds
+ASSET_CONFIG = {
+    "BTC": {
+        "pyth_id": "e62df6c8b4a85fe1a67db44dc12de5db330f7ac66b72dc658afedf0f4a415B43",
+        "coingecko_id": "bitcoin",
+    },
+    "ETH": {
+        "pyth_id": "ff61491a931112ddf1bd8147cd1b641375f79f5825126d665480874634fd0ace",
+        "coingecko_id": "ethereum",
+    },
+    "SOL": {
+        "pyth_id": "ef0d8b6fda2ceba41da15d4095d1da392a0d2f8ed0c6c7bc0f4cfac8c280b56d",
+        "coingecko_id": "solana",
+    },
+}
+
 class PriceFetcher:
-    PYTH_URL = "https://hermes.pyth.network/v2/updates/price/latest?ids[]=e62df6c8b4a85fe1a67db44dc12de5db330f7ac66b72dc658afedf0f4a415B43"
-    CHAINLINK_URL = "https://api.coingecko.com/api/v3/simple/price?ids=bitcoin&vs_currencies=usd"
+    PYTH_URL = "https://hermes.pyth.network/v2/updates/price/latest"
+    AGGREGATOR_URL = "https://api.coingecko.com/api/v3/simple/price"
 
     def __init__(self, client: httpx.AsyncClient):
         self.client = client
-        self._prices: Dict[str, Optional[float]] = {"pyth": None, "chainlink_agg": None}
+        self._prices: Dict[str, Dict[str, Optional[float]]] = {}
         self._lock = asyncio.Lock()
 
-    async def _fetch_pyth(self) -> Optional[float]:
+    async def _fetch_pyth_prices(self) -> Dict[str, float]:
+        pyth_ids = [v['pyth_id'] for v in ASSET_CONFIG.values()]
+        params = [("ids[]", pid) for pid in pyth_ids]
         try:
-            resp = await self.client.get(self.PYTH_URL, timeout=5)
+            resp = await self.client.get(self.PYTH_URL, params=params, timeout=5)
             resp.raise_for_status()
-            price_data = resp.json()['parsed'][0]['price']
-            return int(price_data['price']) / (10 ** abs(int(price_data['expo'])))
+            parsed_data = resp.json().get('parsed', [])
+            
+            # Map the returned prices back to our asset symbols (BTC, ETH, etc.)
+            id_to_symbol = {v['pyth_id']: k for k, v in ASSET_CONFIG.items()}
+            return {
+                id_to_symbol[item['id']]: int(item['price']['price']) / (10 ** abs(int(item['price']['expo'])))
+                for item in parsed_data if item['id'] in id_to_symbol
+            }
         except Exception as e:
             logger.error(f"❌ Oracle Error (Pyth): {e}")
-            return None
+            return {}
 
-    async def _fetch_chainlink_agg(self) -> Optional[float]:
+    async def _fetch_aggregator_prices(self) -> Dict[str, float]:
+        coingecko_ids = ",".join([v['coingecko_id'] for v in ASSET_CONFIG.values()])
+        params = {"ids": coingecko_ids, "vs_currencies": "usd"}
         try:
-            resp = await self.client.get(self.CHAINLINK_URL, timeout=5)
+            resp = await self.client.get(self.AGGREGATOR_URL, params=params, timeout=5)
             resp.raise_for_status()
-            return float(resp.json()['bitcoin']['usd'])
+            data = resp.json()
+
+            # Map the returned prices back to our asset symbols
+            id_to_symbol = {v['coingecko_id']: k for k, v in ASSET_CONFIG.items()}
+            return {id_to_symbol[cg_id]: prices['usd'] for cg_id, prices in data.items()}
         except Exception as e:
-            logger.error(f"❌ Oracle Error (Chainlink Agg): {e}")
-            return None
+            logger.error(f"❌ Oracle Error (Aggregator): {e}")
+            return {}
 
     async def update_prices_async(self):
-        pyth_task = self._fetch_pyth()
-        chainlink_task = self._fetch_chainlink_agg()
-        pyth_price, chainlink_price = await asyncio.gather(pyth_task, chainlink_task)
+        pyth_task = self._fetch_pyth_prices()
+        agg_task = self._fetch_aggregator_prices()
+        pyth_prices, agg_prices = await asyncio.gather(pyth_task, agg_task)
         
         async with self._lock:
-            self._prices["pyth"] = pyth_price
-            self._prices["chainlink_agg"] = chainlink_price
-        logger.info(f"✅ Oracle Prices Updated: Pyth=${pyth_price}, Chainlink Agg=${chainlink_price}")
+            # Combine the results for each asset
+            for symbol in ASSET_CONFIG.keys():
+                self._prices[symbol] = {
+                    "pyth": pyth_prices.get(symbol),
+                    "chainlink_agg": agg_prices.get(symbol)
+                }
+        
+        logger.info(f"✅ Multi-Asset Prices Updated: {self._prices}")
 
-    def get_current_prices(self) -> Dict[str, Optional[float]]:
+    def get_all_prices(self) -> Dict[str, Dict[str, Optional[float]]]:
         return self._prices.copy()