api: select price source via query param and ticker renaming (#1307)

interlay · Jun 19, 2023 · 4e1c721 · 4e1c721 · vercel · Jun 19, 2023
1 parent 4a19226
commit 4e1c721
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/api/market_data.py b/api/market_data.py
@@ -8,6 +8,9 @@
 
 api_key = os.environ.get("CG_API_KEY")
 
+tickers = {
+  "tether usd": "tether",
+}
 
 @app.after_request
 def add_header(response):
@@ -47,8 +50,11 @@ def dia(asset):
     resp = requests.get(url, headers=headers_dict)
     data = resp.json()
 
+    # optionally rename the ticker
+    ticker = tickers.get(data["Name"], data["Name"]).lower()
+
     return {
-      data["Name"].lower(): {
+      ticker: {
         "usd": data["Price"],
       }
     }
@@ -58,7 +64,7 @@ def dia(asset):
 def get_price():
     args = request.args
 
-    price_source = request.headers.get('x-price-source')
+    price_source = args.get('price-source')
 
     data = {}