perf: make plugin load faster (#116)

Author: antazoey

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
 -   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.6.0
+    rev: v5.0.0
     hooks:
     -   id: check-yaml
 
@@ -10,7 +10,7 @@ repos:
       - id: isort
 
 -   repo: https://github.com/psf/black
-    rev: 24.8.0
+    rev: 24.10.0
     hooks:
       - id: black
         name: black
@@ -22,13 +22,13 @@ repos:
         additional_dependencies: [flake8-breakpoint, flake8-print, flake8-pydantic]
 
 -   repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.11.1
+    rev: v1.13.0
     hooks:
     -   id: mypy
         additional_dependencies: [types-PyYAML, types-requests, types-setuptools, pydantic]
 
 -   repo: https://github.com/executablebooks/mdformat
-    rev: 0.7.17
+    rev: 0.7.18
     hooks:
     -   id: mdformat
         additional_dependencies: [mdformat-gfm, mdformat-frontmatter, mdformat-pyproject]

ape_foundry/__init__.py

@@ -4,25 +4,22 @@
 """
 
 from ape import plugins
-from ape.api.networks import LOCAL_NETWORK_NAME
-from ape_ethereum.ecosystem import NETWORKS
-
-from .provider import (
-    FoundryForkProvider,
-    FoundryNetworkConfig,
-    FoundryProvider,
-    FoundryProviderError,
-    FoundrySubprocessError,
-)
 
 
 @plugins.register(plugins.Config)
 def config_class():
+    from .provider import FoundryNetworkConfig
+
     return FoundryNetworkConfig
 
 
 @plugins.register(plugins.ProviderPlugin)
 def providers():
+    from ape.api.networks import LOCAL_NETWORK_NAME
+    from ape_ethereum.ecosystem import NETWORKS
+
+    from .provider import FoundryForkProvider, FoundryProvider
+
     yield "ethereum", LOCAL_NETWORK_NAME, FoundryProvider
 
     for network in NETWORKS:
@@ -61,6 +58,12 @@ def providers():
     yield "blast", "sepolia-fork", FoundryForkProvider
 
 
+def __getattr__(name: str):
+    import ape_foundry.provider as module
+
+    return getattr(module, name)
+
+
 __all__ = [
     "FoundryNetworkConfig",
     "FoundryProvider",

ape_foundry/provider.py

@@ -3,7 +3,7 @@
 import shutil
 from bisect import bisect_right
 from subprocess import PIPE, call
-from typing import Literal, Optional, Union, cast
+from typing import TYPE_CHECKING, Literal, Optional, Union, cast
 
 from ape.api import (
     BlockAPI,
@@ -23,7 +23,6 @@
     VirtualMachineError,
 )
 from ape.logging import logger
-from ape.types import AddressType, BlockID, ContractCode, SnapshotID
 from ape.utils import cached_property
 from ape_ethereum.provider import Web3Provider
 from ape_ethereum.trace import TraceApproach, TransactionTrace
@@ -54,6 +53,10 @@
 except ImportError:
     Optimism = None  # type: ignore
 
+if TYPE_CHECKING:
+    from ape.types import AddressType, BlockID, ContractCode, SnapshotID
+
+
 EPHEMERAL_PORTS_START = 49152
 EPHEMERAL_PORTS_END = 60999
 DEFAULT_PORT = 8545
@@ -136,7 +139,7 @@ class FoundryProvider(SubprocessProvider, Web3Provider, TestProviderAPI):
     _did_warn_wrong_node = False
 
     @property
-    def unlocked_accounts(self) -> list[AddressType]:
+    def unlocked_accounts(self) -> list["AddressType"]:
         return list(self.account_manager.test_accounts._impersonated_accounts)
 
     @property
@@ -485,7 +488,7 @@ def build_command(self) -> list[str]:
 
         return cmd
 
-    def set_balance(self, account: AddressType, amount: Union[int, float, str, bytes]):
+    def set_balance(self, account: "AddressType", amount: Union[int, float, str, bytes]):
         is_str = isinstance(amount, str)
         is_key_word = is_str and " " in amount  # type: ignore
         _is_hex = is_str and not is_key_word and amount.startswith("0x")  # type: ignore
@@ -516,19 +519,19 @@ def mine(self, num_blocks: int = 1):
     def snapshot(self) -> str:
         return self.make_request("evm_snapshot", [])
 
-    def restore(self, snapshot_id: SnapshotID) -> bool:
+    def restore(self, snapshot_id: "SnapshotID") -> bool:
         snapshot_id = to_hex(snapshot_id) if isinstance(snapshot_id, int) else snapshot_id
         result = self.make_request("evm_revert", [snapshot_id])
         return result is True
 
-    def unlock_account(self, address: AddressType) -> bool:
+    def unlock_account(self, address: "AddressType") -> bool:
         self.make_request("anvil_impersonateAccount", [address])
         return True
 
-    def relock_account(self, address: AddressType):
+    def relock_account(self, address: "AddressType"):
         self.make_request("anvil_stopImpersonatingAccount", [address])
 
-    def get_balance(self, address: AddressType, block_id: Optional[BlockID] = None) -> int:
+    def get_balance(self, address: "AddressType", block_id: Optional["BlockID"] = None) -> int:
         if result := self.make_request("eth_getBalance", [address, block_id]):
             return int(result, 16) if isinstance(result, str) else result
 
@@ -645,7 +648,7 @@ def _extract_custom_error(self, **kwargs) -> str:
     def set_block_gas_limit(self, gas_limit: int) -> bool:
         return self.make_request("evm_setBlockGasLimit", [hex(gas_limit)]) is True
 
-    def set_code(self, address: AddressType, code: ContractCode) -> bool:
+    def set_code(self, address: "AddressType", code: "ContractCode") -> bool:
         if isinstance(code, bytes):
             code = code.hex()
 
@@ -658,7 +661,7 @@ def set_code(self, address: AddressType, code: ContractCode) -> bool:
         self.make_request("anvil_setCode", [address, code])
         return True
 
-    def set_storage(self, address: AddressType, slot: int, value: HexBytes):
+    def set_storage(self, address: "AddressType", slot: int, value: HexBytes):
         self.make_request(
             "anvil_setStorageAt",
             [
@@ -712,7 +715,7 @@ def evm_version(self) -> Optional[str]:
 
         return self.settings.evm_version
 
-    def get_block(self, block_id: BlockID) -> BlockAPI:
+    def get_block(self, block_id: "BlockID") -> BlockAPI:
         if isinstance(block_id, str) and block_id.isnumeric():
             block_id = int(block_id)
 

setup.cfg

@@ -4,7 +4,8 @@ exclude =
 	venv*
 	docs
 	build
-ignore = E704,W503,PYD002
+ignore = E704,W503,PYD002,TC003,TC006
 per-file-ignores =
     # The traces have to be formatted this way for the tests.
     tests/expected_traces.py: E501
+type-checking-pydantic-enabled = True

setup.py

@@ -15,8 +15,8 @@
         "ape-optimism",  # For Optimism integration tests
     ],
     "lint": [
-        "black>=24.8.0,<25",  # Auto-formatter and linter
-        "mypy>=1.11.1,<2",  # Static type analyzer
+        "black>=24.10.0,<25",  # Auto-formatter and linter
+        "mypy>=1.13.0,<2",  # Static type analyzer
         "types-setuptools",  # Needed for mypy type shed
         "types-requests",  # Needed for mypy type shed
         "types-PyYAML",  # Needed for mypy type shed
@@ -25,7 +25,7 @@
         "flake8-print>=5.0.0,<6",  # Detect print statements left in code
         "flake8-pydantic",  # For detecting issues with Pydantic models
         "isort>=5.13.2,<6",  # Import sorting linter
-        "mdformat>=0.7.17",  # Auto-formatter for markdown
+        "mdformat>=0.7.18",  # Auto-formatter for markdown
         "mdformat-gfm>=0.3.5",  # Needed for formatting GitHub-flavored markdown
         "mdformat-frontmatter>=0.4.1",  # Needed for frontmatters-style headers in issue templates
         "mdformat-pyproject>=0.0.1",  # Allows configuring in pyproject.toml