fix: update the mech subgraph's url

Author: Adamantios

scripts/mech_events.py

@@ -30,6 +30,7 @@
 from typing import Any, ClassVar, Dict
 
 import requests
+from dotenv import dotenv_values
 from gql import Client, gql
 from gql.transport.requests import RequestsHTTPTransport
 from tqdm import tqdm
@@ -38,7 +39,9 @@
 
 SCRIPT_PATH = Path(__file__).resolve().parent
 STORE_PATH = Path(SCRIPT_PATH, "..", ".trader_runner")
-MECH_EVENTS_JSON_PATH = Path(STORE_PATH, "mech_events.json")
+ENV_FILENAME = ".env"
+DOTENV_PATH = STORE_PATH / ENV_FILENAME
+MECH_EVENTS_JSON_PATH = STORE_PATH / "mech_events.json"
 HTTP = "http://"
 HTTPS = HTTP[:4] + "s" + HTTP[4:]
 CID_PREFIX = "f01701220"
@@ -47,7 +50,9 @@
 DEFAULT_MECH_FEE = 10000000000000000
 DEFAULT_FROM_TIMESTAMP = 0
 DEFAULT_TO_TIMESTAMP = 2147483647
-MECH_SUBGRAPH_URL = "https://api.studio.thegraph.com/query/57238/mech/0.0.2"
+MECH_SUBGRAPH_URL_TEMPLATE = Template(
+    "https://gateway.thegraph.com/api/${SUBGRAPH_API_KEY}/subgraphs/id/4YGoX3iXUni1NBhWJS5xyKcntrAzssfytJK7PQxxQk5g"
+)
 SUBGRAPH_HEADERS = {
     "Accept": "application/json, multipart/mixed",
     "Content-Type": "application/json",
@@ -74,6 +79,7 @@
     """
 )
 
+
 @dataclass
 class MechBaseEvent:  # pylint: disable=too-many-instance-attributes
     """Base class for mech's on-chain event representation."""
@@ -159,7 +165,7 @@ def _read_mech_events_data_from_file() -> Dict[str, Any]:
         if mech_events_data.get("db_version", 0) < MECH_EVENTS_DB_VERSION:
             current_time = time.strftime("%Y-%m-%d_%H-%M-%S")
             old_db_filename = f"mech_events.{current_time}.old.json"
-            os.rename(MECH_EVENTS_JSON_PATH, Path(STORE_PATH, old_db_filename))
+            os.rename(MECH_EVENTS_JSON_PATH, STORE_PATH / old_db_filename)
             mech_events_data = {}
             mech_events_data["db_version"] = MECH_EVENTS_DB_VERSION
     except FileNotFoundError:
@@ -190,17 +196,26 @@ def _write_mech_events_data_to_file(
         last_write_time = now
 
 
+def get_mech_subgraph_url() -> str:
+    """Get the mech subgraph's URL."""
+    env_file_vars = dotenv_values(DOTENV_PATH)
+    return MECH_SUBGRAPH_URL_TEMPLATE.substitute(env_file_vars)
+
+
 def _query_mech_events_subgraph(
     sender: str, event_cls: type[MechBaseEvent]
 ) -> dict[str, Any]:
     """Query the subgraph."""
 
-    transport = RequestsHTTPTransport(url=MECH_SUBGRAPH_URL)
+    mech_subgraph_url = get_mech_subgraph_url()
+    transport = RequestsHTTPTransport(mech_subgraph_url)
     client = Client(transport=transport, fetch_schema_from_transport=True)
 
     subgraph_event_set_name = f"{event_cls.subgraph_event_name}s"
     all_results: dict[str, Any] = {"data": {subgraph_event_set_name: []}}
-    query = MECH_EVENTS_SUBGRAPH_QUERY_TEMPLATE.safe_substitute(subgraph_event_set_name=subgraph_event_set_name)
+    query = MECH_EVENTS_SUBGRAPH_QUERY_TEMPLATE.safe_substitute(
+        subgraph_event_set_name=subgraph_event_set_name
+    )
     id_gt = ""
     while True:
         variables = {