refactor: use network singltone for get_network_height

subsquid · Feb 12, 2024 · cf133bf · cf133bf
1 parent 590401c
commit cf133bf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/ape_subsquid/__init__.py b/ape_subsquid/__init__.py
@@ -1,8 +1,8 @@
 from ape import plugins
 
-from ape_subsquid.query import SubsquidQueryEngine, get_gateway_height
+from ape_subsquid.query import SubsquidQueryEngine, get_network_height
 
-__all__ = ["exceptions", "get_gateway_height"]
+__all__ = ["exceptions", "get_network_height"]
 
 
 @plugins.register(plugins.QueryPlugin)

diff --git a/ape_subsquid/query.py b/ape_subsquid/query.py
@@ -1,5 +1,6 @@
 from typing import Iterator, Optional, Type, TypeVar, cast
 
+from ape import networks
 from ape.api import BlockAPI, ReceiptAPI
 from ape.api.query import (
     AccountTransactionQuery,
@@ -215,7 +216,7 @@ def gateway_ingest(gateway: SubsquidGateway, network: str, query: Query) -> Iter
         query["fromBlock"] = last_block + 1
 
 
-def get_gateway_height(network_manager) -> int:
-    network = get_network(network_manager)
+def get_network_height() -> int:
+    network = get_network(networks)
     height = gateway.get_height(network)
     return height