Added support for testnets

litneet64 · litneet64 · commit 61abdba248a6 · 2021-03-08T02:34:30.000-03:00
diff --git a/etherscan/enums/fields_enum.py b/etherscan/enums/fields_enum.py
@@ -23,7 +23,7 @@ class FieldsEnum:
     OFFSET: str = "&offset="
     PAGE: str = "&page="
     POSITION: str = "&position="
-    PREFIX: str = "https://api.etherscan.io/api?"
+    PREFIX: str = "https://api-{}.etherscan.io/api?"
     SORT: str = "&sort="
     START_BLOCK: str = "&startblock="
     START_DATE: str = "&startdate="
diff --git a/etherscan/etherscan.py b/etherscan/etherscan.py
@@ -10,35 +10,35 @@
 
 
 class Etherscan:
-    def __new__(cls, api_key: str):
-        with resources.path(configs, "stable.json") as path:
+    def __new__(cls, api_key: str, net: str = "MAIN"):
+        with resources.path(configs, f"{net.upper()}-stable.json") as path:
             config_path = str(path)
-        return cls.from_config(api_key=api_key, config_path=config_path)
+        return cls.from_config(api_key=api_key, config_path=config_path, net=net)
 
     @staticmethod
     def __load_config(config_path: str) -> dict:
         with open(config_path, "r") as f:
             return json.load(f)
 
     @staticmethod
-    def __run(func, api_key: str):
+    def __run(func, api_key: str, net: str):
         def wrapper(*args, **kwargs):
             url = (
-                f"{fields.PREFIX}"
+                f"{fields.PREFIX.format(net.lower()).replace('-main','')}"
                 f"{func(*args, **kwargs)}"
                 f"{fields.API_KEY}"
                 f"{api_key}"
             )
-            r = requests.get(url)
+            r = requests.get(url, headers={'User-Agent':''})
             return parser.parse(r)
 
         return wrapper
 
     @classmethod
-    def from_config(cls, api_key: str, config_path: str):
+    def from_config(cls, api_key: str, config_path: str, net: str):
         config = cls.__load_config(config_path)
         for func, v in config.items():
             if not func.startswith("_"):  # disabled if _
                 attr = getattr(getattr(etherscan, v["module"]), func)
-                setattr(cls, func, cls.__run(attr, api_key))
+                setattr(cls, func, cls.__run(attr, api_key, net))
         return cls
