feat: refactor checksum cache for speed (#796)

* feat: refactor checksum cache for speed

* feat: quicker cache loader

* chore: `black .`

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

y/_db/utils/logs.py

@@ -348,11 +348,9 @@ def _wrap_query_with_addresses(self, generator) -> Query:
         if not (addresses := self.addresses):
             return generator
         elif isinstance(addresses, str):
-            address = _remove_0x_prefix(convert.to_address(addresses))
+            address = convert.to_address(addresses)[2:]
             return (log for log in generator if log.address.hash == address)
-        addresses = [
-            _remove_0x_prefix(convert.to_address(address)) for address in addresses
-        ]
+        addresses = tuple(convert.to_address(address)[2:] for address in addresses)
         return (log for log in generator if log.address.hash in addresses)
 
     def _wrap_query_with_topic(self, generator, topic_id: str) -> Query:

y/convert.py

@@ -16,16 +16,34 @@
 
 @lru_cache(maxsize=ENVS.CHECKSUM_CACHE_MAXSIZE)
 def checksum(address: AnyAddressType) -> ChecksumAddress:
-    return str(EthAddress(address))
+    try:
+        return to_checksum_address(address)
+    except ValueError:
+        raise ValueError(f"'{address}' is not a valid ETH address") from None
 
 
 def to_address(address: AnyAddressType) -> ChecksumAddress:
-    address = _int_to_address(address) if type(address) == int else str(address)
+    if isinstance(address, bytes):
+        address = (
+            address.hex()
+            if type(address).__name__ == "HexBytes"
+            else HexBytes(address).hex()
+        )
+    elif type(address) == int:
+        address = _int_to_address(address)
+    else:
+        address = str(address)
+
     if address in _is_checksummed:
         return address
     elif address in _is_not_checksummed:
         return checksum(address)
-    checksummed = str(EthAddress(address))
+
+    try:
+        checksummed = to_checksum_address(address)
+    except ValueError:
+        raise ValueError(f"'{address}' is not a valid ETH address") from None
+
     if address == checksummed:
         _is_checksummed.add(address)
     else:

y/prices/dex/uniswap/v2.py

@@ -375,7 +375,9 @@ def _log_factory_helper_failure(
     )
 
 
-_all_pools_semaphore = a_sync.Semaphore(10, name=f"{__name__}.UniswapRouterV2.all_pools_for")
+_all_pools_semaphore = a_sync.Semaphore(
+    10, name=f"{__name__}.UniswapRouterV2.all_pools_for"
+)
 
 
 class UniswapRouterV2(ContractBase):