Migrate subgraphs for staging

KlimaDAO · Apr 24, 2024 · e2b28ed · e2b28ed
1 parent 9cd7c1c
commit e2b28ed
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/apps/treasury/util/constants.py b/src/apps/treasury/util/constants.py
@@ -1,5 +1,15 @@
+import os
+
 # Subgraphs
-KLIMA_PROTOCOL_SUBGRAPH = 'https://api.thegraph.com/subgraphs/name/klimadao/klimadao-protocol-metrics'
+GRAPH_API_KEY = os.environ.get('GRAPH_API_KEY')
+GRAPH_BASE_URL = f'https://gateway-arbitrum.network.thegraph.com/api/{GRAPH_API_KEY}/subgraphs/id/'
+GRAPH_DEV_BASE_URL = 'https://api.studio.thegraph.com/query/71975/'
+GRAPH_VERSION_SUFFIX = '/version/latest'
+
+if os.environ.get('ENV') == 'production':
+    KLIMA_PROTOCOL_SUBGRAPH = 'https://api.thegraph.com/subgraphs/name/klimadao/klimadao-protocol-metrics'
+else:
+    KLIMA_PROTOCOL_SUBGRAPH = GRAPH_DEV_BASE_URL + 'staging-klimadao-protocol-metrics' + GRAPH_VERSION_SUFFIX
 
 BCT_ERC20_CONTRACT = '0x2f800db0fdb5223b3c3f354886d907a671414a7f'
 NCT_ERC20_CONTRACT = '0xD838290e877E0188a4A44700463419ED96c16107'