Refactor

Set chain IDs attribute in constructor (typ across project)
Remove ABI in Curve pool
Use batched requests in more places
Make web3 connection manager multi-chain
Make token and pool registry singletons
Move Aerodrome V3 tests
Linting and cleanup

Author: BowTiedDevil

src/degenbot/__init__.py

@@ -1,5 +1,4 @@
 # TODO:
-# - add all module folders to __all__
 # - AllTokens and AllPools improvements
 #   - singleton created at import
 #   - make getter/setter methods with chain ID
@@ -33,8 +32,8 @@
 from .logging import logger
 from .managers.erc20_token_manager import Erc20TokenManager
 from .pancakeswap.pools import PancakeV3Pool
-from .registry.all_pools import AllPools
-from .registry.all_tokens import AllTokens
+from .registry.all_pools import pool_registry
+from .registry.all_tokens import token_registry
 from .sushiswap.managers import SushiswapV2PoolManager, SushiswapV3PoolManager
 from .sushiswap.pools import SushiswapV2Pool, SushiswapV3Pool
 from .transaction.uniswap_transaction import UniswapTransaction
@@ -62,18 +61,18 @@
     "get_web3",
     "logger",
     "pancakeswap",
+    "pool_registry",
     "set_web3",
     "solidly",
     "sushiswap",
+    "token_registry",
     "uniswap",
     "AerodromeV2Pool",
     "AerodromeV2PoolManager",
     "AerodromeV2PoolState",
     "AerodromeV3Pool",
     "AerodromeV3PoolManager",
     "AerodromeV3PoolState",
-    "AllPools",
-    "AllTokens",
     "AnvilFork",
     "ArbitrageCalculationResult",
     "BuilderEndpoint",

src/degenbot/aerodrome/managers.py

@@ -8,10 +8,10 @@
 
 from degenbot.exceptions import AddressMismatch, DegenbotError, ManagerError, PoolNotAssociated
 from degenbot.logging import logger
-from degenbot.registry.all_pools import AllPools
+from degenbot.registry.all_pools import pool_registry
 
 from ..aerodrome.pools import AerodromeV2Pool, AerodromeV3Pool
-from ..config import get_web3
+from ..config import web3_connection_manager
 from ..functions import encode_function_calldata, get_number_for_block_identifier, raw_call
 from ..uniswap.managers import UniswapV2PoolManager, UniswapV3PoolManager
 from .functions import generate_aerodrome_v3_pool_address
@@ -36,7 +36,7 @@ def get_pair_from_factory(  # type: ignore[override]
                 function_arguments=[token0, token1, stable],
             ),
             return_types=["address"],
-            block_identifier=get_number_for_block_identifier(block_identifier),
+            block_identifier=get_number_for_block_identifier(block_identifier, w3),
         )
         return cast(str, pool_address)
 
@@ -54,7 +54,7 @@ def get_pool_from_tokens(  # type: ignore[override]
         pool = self._build_pool(
             pool_address=to_checksum_address(
                 self.get_pair_from_factory(
-                    w3=get_web3(),
+                    w3=web3_connection_manager.get_web3(self.chain_id),
                     token0=to_checksum_address(token_addresses[0]),
                     token1=to_checksum_address(token_addresses[1]),
                     stable=stable,
@@ -92,8 +92,12 @@ def _build_pool(
                 f"Pool address {pool_address} not associated with factory {self._factory_address}"
             )
 
-        # Check if the AllPools collection already has this pool
-        if (known_pool_helper := AllPools(self._chain_id).get(pool_address)) is not None:
+        # Check if the pool registry already has this pool
+        if (
+            known_pool_helper := pool_registry.get(
+                pool_address=pool_address, chain_id=self._chain_id
+            )
+        ) is not None:
             if TYPE_CHECKING:
                 assert isinstance(known_pool_helper, AerodromeV3Pool)
             if known_pool_helper.factory == self._factory_address: