inspect whether package is installed

.github/workflows/read_storage.yml

@@ -39,7 +39,7 @@ jobs:
 
       - name: Install python dependencies
         run: |
-          pip install ".[slither-read-storage]"
+          pip install ".[read-storage]"
           solc-select install 0.8.1
           solc-select install 0.8.10
           solc-select use 0.8.1

setup.py

@@ -31,7 +31,7 @@ setup(
             "openai",
             "pdoc",
         ],
-        "slither-read-storage": ["solc-select>=v1.0.0b1", "web3>=6.0.0b"],
+        "read-storage": ["solc-select>=v1.0.0b1", "web3>=6.0.0b"],
     },
     license="AGPL-3.0",
     long_description=long_description,

slither/tools/read_storage/read_storage.py

@@ -1,27 +1,31 @@
+import importlib.util
 import logging
 import sys
 from math import floor
-from typing import Callable, Optional, Tuple, Union, List, Dict, Any
+from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
+if importlib.util.find_spec("web3").loader is None:
+    print(
+        "Please install slither with `pip install slither-analyzer[read-storage]` to use slither-read-storage."
+    )
+    sys.exit(-1)
+
+import dataclasses
 
-from web3 import Web3
-from eth_typing.evm import ChecksumAddress
 from eth_abi import decode, encode
+from eth_typing.evm import ChecksumAddress
 from eth_utils import keccak
-from .utils import (
-    get_offset_value,
-    get_storage_data,
-    coerce_type,
-)
-
+from web3 import Web3
 
-import dataclasses
-from slither.utils.myprettytable import MyPrettyTable
-from slither.core.solidity_types.type import Type
-from slither.core.solidity_types import ArrayType, ElementaryType, UserDefinedType, MappingType
 from slither.core.declarations import Contract, Structure
+from slither.core.solidity_types import (ArrayType, ElementaryType,
+                                         MappingType, UserDefinedType)
+from slither.core.solidity_types.type import Type
 from slither.core.variables.state_variable import StateVariable
 from slither.core.variables.structure_variable import StructureVariable
+from slither.utils.myprettytable import MyPrettyTable
+
+from .utils import coerce_type, get_offset_value, get_storage_data
 
 logging.basicConfig()
 logger = logging.getLogger("Slither-read-storage")

slither/tools/read_storage/utils/__init__.py

@@ -1,5 +1 @@
-from .utils import (
-    get_offset_value,
-    get_storage_data,
-    coerce_type,
-)
+from .utils import coerce_type, get_offset_value, get_storage_data

slither/tools/read_storage/utils/utils.py

@@ -1,7 +1,7 @@
 from typing import Union
 
 from eth_typing.evm import ChecksumAddress
-from eth_utils import to_int, to_text, to_checksum_address
+from eth_utils import to_checksum_address, to_int, to_text
 
 
 def get_offset_value(hex_bytes: bytes, offset: int, size: int) -> bytes: