Use ENS exception wrappers for Python exceptions

Author: fselmo

ens/async_ens.py

@@ -46,6 +46,7 @@
 )
 from ens.exceptions import (
     AddressMismatch,
+    ENSValueError,
     ResolverNotFound,
     UnauthorizedError,
     UnownedName,
@@ -64,9 +65,6 @@
     normalize_name,
     raw_name_to_hash,
 )
-from web3.exceptions import (
-    Web3ValueError,
-)
 
 if TYPE_CHECKING:
     from web3.contract.async_contract import (  # noqa: F401
@@ -207,7 +205,7 @@ async def setup_address(
         elif is_binary_address(address):
             address = to_checksum_address(cast(str, address))
         elif not is_checksum_address(address):
-            raise Web3ValueError("You must supply the address in checksum format")
+            raise ENSValueError("You must supply the address in checksum format")
         if await self.address(name) == address:
             return None
         if address is None:
@@ -285,7 +283,7 @@ async def setup_name(
             if is_binary_address(address):
                 address = to_checksum_address(address)
             if not is_checksum_address(address):
-                raise Web3ValueError("You must supply the address in checksum format")
+                raise ENSValueError("You must supply the address in checksum format")
             await self._assert_control(address, name)
             if not resolved:
                 await self.setup_address(name, address, transact=transact)

ens/ens.py

@@ -30,10 +30,6 @@
     HexBytes,
 )
 
-from web3.exceptions import (
-    Web3ValueError,
-)
-
 from . import (
     abis,
 )
@@ -50,6 +46,7 @@
 )
 from .exceptions import (
     AddressMismatch,
+    ENSValueError,
     ResolverNotFound,
     UnauthorizedError,
     UnownedName,
@@ -212,7 +209,7 @@ def setup_address(
         elif is_binary_address(address):
             address = to_checksum_address(cast(str, address))
         elif not is_checksum_address(address):
-            raise Web3ValueError("You must supply the address in checksum format")
+            raise ENSValueError("You must supply the address in checksum format")
         if self.address(name) == address:
             return None
         if address is None:
@@ -288,7 +285,7 @@ def setup_name(
             if is_binary_address(address):
                 address = to_checksum_address(address)
             if not is_checksum_address(address):
-                raise Web3ValueError("You must supply the address in checksum format")
+                raise ENSValueError("You must supply the address in checksum format")
             self._assert_control(address, name)
             if not resolved:
                 self.setup_address(name, address, transact=transact)

ens/exceptions.py

@@ -12,6 +12,24 @@ class ENSException(Exception):
     pass
 
 
+class ENSValueError(ENSException, ValueError):
+    """
+    An ENS exception wrapper for `ValueError`, for better control over
+    exception handling.
+    """
+
+    pass
+
+
+class ENSTypeError(ENSException, TypeError):
+    """
+    An ENS exception wrapper for `TypeError`, for better control over
+    exception handling.
+    """
+
+    pass
+
+
 class AddressMismatch(ENSException):
     """
     In order to set up reverse resolution correctly, the ENS name should first

ens/utils.py

@@ -35,9 +35,9 @@
     HexBytes,
 )
 
-from web3.exceptions import (
-    Web3TypeError,
-    Web3ValueError,
+from ens.exceptions import (
+    ENSTypeError,
+    ENSValueError,
 )
 
 from .constants import (
@@ -204,7 +204,7 @@ def sha3_text(val: Union[str, bytes]) -> HexBytes:
 def label_to_hash(label: str) -> HexBytes:
     label = normalize_name(label)
     if "." in label:
-        raise Web3ValueError(f"Cannot generate hash for label {label!r} with a '.'")
+        raise ENSValueError(f"Cannot generate hash for label {label!r} with a '.'")
     return Web3().keccak(text=label)
 
 
@@ -269,7 +269,7 @@ def assert_signer_in_modifier_kwargs(modifier_kwargs: Any) -> ChecksumAddress:
 
     _modifier_type, modifier_dict = dict(modifier_kwargs).popitem()
     if "from" not in modifier_dict:
-        raise Web3TypeError(ERR_MSG)
+        raise ENSTypeError(ERR_MSG)
 
     return modifier_dict["from"]