Added shelve cache.

zmbbb · zmbbb · commit daf809c15bd2 · 2023-01-08T14:51:17.000+01:00
diff --git a/etherscan/etherscan.py b/etherscan/etherscan.py
@@ -8,8 +8,13 @@
 from etherscan.enums.fields_enum import FieldsEnum as fields
 from etherscan.utils.parsing import ResponseParser as parser
 
+from etherscan.utils.shelve import Shelve as shelve
+
 
 class Etherscan:
+    # Enable/disable caching of all requests
+    CACHING = True
+    
     def __new__(cls, api_key: str, net: str = "MAIN"):
         with resources.path(configs, f"{net.upper()}-stable.json") as path:
             config_path = str(path)
@@ -29,7 +34,14 @@ def wrapper(*args, **kwargs):
                 f"{fields.API_KEY}"
                 f"{api_key}"
             )
+
+            r = shelve.shelve_load(url)
+            if r and Etherscan.CACHING:
+                return parser.parse(r)
+
             r = requests.get(url, headers={"User-Agent": ""})
+            shelve.shelve_store(url, r)
+
             return parser.parse(r)
 
         return wrapper
diff --git a/etherscan/utils/shelve.py b/etherscan/utils/shelve.py
@@ -0,0 +1,24 @@
+import shelve
+
+
+class Shelve:
+    SHELVE_FILE = r'data/etherscan_shelve'
+
+    @staticmethod
+    def shelve_store(key, data):
+        # store result in cache
+        d = shelve.open(Shelve.SHELVE_FILE)
+        d[key] = data
+        d.close()
+
+    @staticmethod
+    def shelve_load(key):
+        d = shelve.open(Shelve.SHELVE_FILE)
+        try:
+            data = d[key]
+            return data
+        except KeyError as err:
+            return None
+        finally:
+            d.close()
+
