Defillama api key support (#84)

* Defillama api key support * fix: env * fix: remove dotenv --------- Co-authored-by: 0xevolve <[email protected]>
astraly-labs · Feb 27, 2024 · 35ebdec · 35ebdec
1 parent 49065d4
commit 35ebdec
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/pragma/publisher/fetchers/defillama.py b/pragma/publisher/fetchers/defillama.py
@@ -1,5 +1,6 @@
 import asyncio
 import logging
+import os
 from typing import Dict, List
 
 import requests
@@ -46,15 +47,18 @@ class DefillamaFetcher(PublisherInterfaceT):
     )
 
     SOURCE: str = "DEFILLAMA"
-    headers = {
-        "Accepts": "application/json",
-    }
+    api_key: str
+
+    headers: dict
 
     publisher: str
 
-    def __init__(self, assets: List[PragmaAsset], publisher):
+    def __init__(self, assets: List[PragmaAsset], publisher, api_key=None):
         self.assets = assets
         self.publisher = publisher
+        self.headers = {"Accepts": "application/json"}
+        if api_key:
+            self.headers["X-Api-Key"] = api_key
 
     async def _fetch_pair(
         self, asset: PragmaSpotAsset, session: ClientSession