fix lint warnings

2 years ago · 2b9f192b39
parent b5e6ebbc36
commit 2b9f192b39
2 changed files with 4 additions and 6 deletions
--- a/slither/tools/read_storage/main.py
+++ b/slither/tools/read_storage/main.py
@ -128,11 +128,10 @@ def main() -> None:

    srs = SlitherReadStorage(contracts, args.max_depth)

-    if args.block:
-        try:
-            srs.block = int(args.block)
-        except ValueError:
-            srs.block = str(args.block)
+    try:
+        srs.block = int(args.block)
+    except ValueError:
+        srs.block = str(args.block or "latest")

    if args.rpc_url:
        # Remove target prefix e.g. rinkeby:0x0 -> 0x0.
--- a/tests/test_read_storage.py
+++ b/tests/test_read_storage.py
@ -112,7 +112,6 @@ def test_read_storage(web3, ganache) -> None:

    srs = SlitherReadStorage(contracts, 100)
    srs.rpc = ganache.provider
-    srs.block = "latest"
    srs.storage_address = address
    srs.get_all_storage_variables()
    srs.get_storage_layout()