Support default chain id of 1 in etherscan uris

Author: njgheorghita

ethpm_cli/_utils/etherscan.py

@@ -1,4 +1,4 @@
-from typing import Any
+from typing import Any, Tuple
 from urllib import parse
 
 from eth_utils import is_checksum_address
@@ -20,10 +20,10 @@ def is_etherscan_uri(value: Any) -> bool:
     if parsed.scheme != "etherscan" or not parsed.netloc:
         return False
 
-    if ":" not in parsed.netloc:
+    if parsed.path:
         return False
 
-    address, chain_id = parsed.netloc.split(":")
+    address, chain_id = parse_etherscan_uri(value)
     if not is_checksum_address(address):
         return False
 
@@ -33,5 +33,14 @@ def is_etherscan_uri(value: Any) -> bool:
     return True
 
 
+def parse_etherscan_uri(uri: str) -> Tuple[str, str]:
+    parsed = parse.urlparse(uri)
+    if ":" in parsed.netloc:
+        address, _, chain_id = parsed.netloc.partition(":")
+    else:
+        address, chain_id = (parsed.netloc, "1")
+    return address, chain_id
+
+
 def get_etherscan_network(chain_id: str) -> str:
     return ETHERSCAN_SUPPORTED_CHAIN_IDS[chain_id]

ethpm_cli/commands/etherscan.py

@@ -1,7 +1,6 @@
 import json
 import os
 from typing import Any, Dict, Iterable, Tuple
-from urllib import parse
 
 from eth_typing import URI
 from eth_utils import to_dict, to_hex, to_int
@@ -10,7 +9,11 @@
 from ethpm.uri import create_latest_block_uri
 import requests
 
-from ethpm_cli._utils.etherscan import get_etherscan_network, is_etherscan_uri
+from ethpm_cli._utils.etherscan import (
+    get_etherscan_network,
+    is_etherscan_uri,
+    parse_etherscan_uri,
+)
 from ethpm_cli.config import get_ipfs_backend, setup_w3
 from ethpm_cli.constants import ETHERSCAN_KEY_ENV_VAR
 from ethpm_cli.exceptions import ContractNotVerified
@@ -41,7 +44,7 @@ def fetch_uri_contents(
 def build_etherscan_manifest(
     uri: URI, package_name: str, version: str
 ) -> Iterable[Tuple[str, Any]]:
-    address, chain_id = parse.urlparse(uri).netloc.split(":")
+    address, chain_id = parse_etherscan_uri(uri)
     network = get_etherscan_network(chain_id)
     body = make_etherscan_request(address, network)
     contract_type = body["ContractName"]

tests/core/_utils/test_etherscan.py

@@ -6,6 +6,7 @@
 @pytest.mark.parametrize(
     "uri,expected",
     (
+        ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729", True),
         ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729:1", True),
         ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729:3", True),
         ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729:4", True),
@@ -15,10 +16,10 @@
         ("etherscan://invalid:1", False),
         # non-checksummed
         ("etherscan://0x6b5da3ca4286baa7fbaf64eeee1834c7d430b729:1", False),
-        # bad path
+        # paths are not allowed
         ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729/1", False),
+        ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729/[email protected]", False),
         # no chain_id
-        ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729", False),
         ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729:", False),
         ("etherscan://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729:10", False),
         ("://0x6b5DA3cA4286Baa7fBaf64EEEE1834C7d430B729:1", False),