try again boi

Author: cmmdotsh

crytic_compile/cryticparser/cryticparser.py

@@ -328,6 +328,14 @@ def _init_etherscan(parser: ArgumentParser) -> None:
         default=DEFAULTS_FLAG_IN_CONFIG["etherscan_api_key"],
     )
 
+    group_etherscan.add_argument(
+        "--test-polygonscan-apikey",
+        help="Etherscan API key.",
+        action="store",
+        dest="test_polygonscan_api_key",
+        default=DEFAULTS_FLAG_IN_CONFIG["etherscan_api_key"],
+    )
+
     group_etherscan.add_argument(
         "--avax-apikey",
         help="Etherscan API key.",

crytic_compile/platform/etherscan.py

@@ -216,6 +216,7 @@ def compile(self, crytic_compile: "CryticCompile", **kwargs: str) -> None:
         etherscan_api_key = kwargs.get("etherscan_api_key", None)
         arbiscan_api_key = kwargs.get("arbiscan_api_key", None)
         polygonscan_api_key = kwargs.get("polygonscan_api_key", None)
+        test_polygonscan_api_key = kwargs.get("test_polygonscan_api_key", None)
         avax_api_key = kwargs.get("avax_api_key", None)
         ftmscan_api_key = kwargs.get("ftmscan_api_key", None)
         bscan_api_key = kwargs.get("bscan_api_key", None)
@@ -235,6 +236,9 @@ def compile(self, crytic_compile: "CryticCompile", **kwargs: str) -> None:
         if polygonscan_api_key and "polygonscan" in etherscan_url:
             etherscan_url += f"&apikey={polygonscan_api_key}"
             etherscan_bytecode_url += f"&apikey={polygonscan_api_key}"
+        if test_polygonscan_api_key and "polygonscan" in etherscan_url:
+            etherscan_url += f"&apikey={test_polygonscan_api_key}"
+            etherscan_bytecode_url += f"&apikey={test_polygonscan_api_key}"
         if avax_api_key and "snowtrace" in etherscan_url:
             etherscan_url += f"&apikey={avax_api_key}"
             etherscan_bytecode_url += f"&apikey={avax_api_key}"