style: apply black linting

Author: LVivona

binding/python/py/synlink/multiaddr.py

@@ -1,18 +1,39 @@
 """Re-export of the multiaddr library."""
 
+from typing import Union
 from multiaddr import Multiaddr
-from multiaddr.protocols import (P_DNS, P_IP4, P_IP6, P_P2P, P_TCP, P_UDP,
-                                 Protocol)
-
-__all__ = ["Multiaddr", "Protocol", "is_valid_address"]
+from multiaddr.protocols import (
+    P_DNS,
+    P_IP4,
+    P_IP6,
+    P_P2P,
+    P_TCP,
+    P_UDP,
+    Protocol,
+)
+from multiaddr.exceptions import (
+    StringParseError,
+    BinaryParseError,
+    ProtocolExistsError,
+    ProtocolNotFoundError,
+)
+
+__all__ = [
+    "Multiaddr",
+    "Protocol",
+    "is_valid_address",
+    "StringParseError",
+    "BinaryParseError",
+    "ProtocolExistsError",
+    "ProtocolNotFoundError",
+]
 
 _PROTOCOL_CONFIG = {
     P_IP4: {"transports": [P_TCP, P_UDP], "overlays": [P_P2P]},
     P_IP6: {"transports": [P_TCP, P_UDP], "overlays": [P_P2P]},
     P_DNS: {"transports": [P_TCP], "overlays": [P_P2P]},
 }
 
-
 def _generate_supported_protocols():
     """Generate all valid protocol combinations."""
     protocols = set()
@@ -28,11 +49,9 @@ def _generate_supported_protocols():
 
     return protocols
 
-
 _SUPPORTED_PROTOCOLS = _generate_supported_protocols()
 
-
-def is_valid_address(addr: Multiaddr) -> bool:
+def is_valid_address(addr: Union[Multiaddr, str, bytes]) -> bool:
     """
     Check if a multiaddr uses only supported protocols.
 
@@ -46,5 +65,9 @@ def is_valid_address(addr: Multiaddr) -> bool:
     "/dns/example.com/tcp/443/p2p/QmYyQSo1c1Ym7orWxLYvCrM2EmxFTANf8wXmmE7DWjhx5N"
     ```
     """
-    protocol_tuple = tuple(protocol.code for protocol in addr.protocols())
+    address = addr
+    if isinstance(addr, (str, bytes)):
+        address = Multiaddr(addr)
+
+    protocol_tuple = tuple(protocol.code for protocol in address.protocols())
     return protocol_tuple in _SUPPORTED_PROTOCOLS

binding/python/py/synlink/peer_id.py

@@ -12,7 +12,6 @@
 else:
     from typing_extensions import Self
 
-
 class PeerId(object):
     def __init__(self, data: Union[bytes, bytearray]):
         if isinstance(data, (bytearray,)):
@@ -21,50 +20,38 @@ def __init__(self, data: Union[bytes, bytearray]):
             self._id = data
         else:
             raise TypeError("only supported types; bytes, and bytearray(s).")
-
     @cached_property
     def xor_id(self) -> int:
         return int(sha256_digest(self._id).hex(), 16)
-
     @cached_property
     def base58(self) -> str:
         return base58.b58encode(self._id).decode()
-
     def to_base58(self) -> str:
         return self.base58
-
     @classmethod
     def from_base58(cls, b58_encoded_peer_id_str: str) -> Self:
         peer_id_bytes = base58.b58decode(b58_encoded_peer_id_str)
         pid = cls(peer_id_bytes)
         return pid
-
     @classmethod
     def from_bytes(cls, data: bytes):
         return cls(data)
-
     def to_bytes(self) -> bytes:
         return self._id
-
     @classmethod
     def from_pubkey(cls, key: PublicKey) -> Self:
         serialized_key = key.to_bytes()
         algo = multihash.Func.sha2_256
         mh_digest = multihash.digest(serialized_key, algo)
         return cls(mh_digest.encode())
-
     def __bytes__(self) -> bytes:
         return self.to_bytes()
-
     def __hash__(self):
         return hash(self._id)
-
     def __repr__(self):
         return f"<synlink.swarm.peer_id.PeerID {self.to_base58()}>"
-
     def __str__(self):
         return self.to_base58()
-
     def __eq__(self, other: object) -> bool:
         if isinstance(other, str):
             return self.to_base58() == other
@@ -75,7 +62,6 @@ def __eq__(self, other: object) -> bool:
         else:
             raise ValueError("Unsupported type for PeerID comparison.")
 
-
 def sha256_digest(data: Union[str, bytes]) -> bytes:
     if isinstance(data, str):
         data = data.encode("utf8")

binding/python/py/synlink/typing.py

@@ -9,7 +9,6 @@
 
 __all__ = ["TProtocol"]
 
-
 @runtime_checkable
 class Serializable(Protocol):
     """
@@ -32,11 +31,9 @@ class Serializable(Protocol):
     def to_bytes(self) -> bytes:
         """serialize the object into bytes."""
         ...
-
     @classmethod
     def from_bytes(cls, data: bytes) -> Self:
         """deserialize the object from bytes."""
         ...
 
-
 TProtocol = NewType("TProtocol", str)

binding/python/py/synlink/utils.py

@@ -3,7 +3,6 @@
 
 __all__ = ["_check_minimum_version", "_check_exact_version"]
 
-
 def _check_minimum_version(
     major: int, minor: Optional[int] = 0, micro: Optional[int] = 0
 ):
@@ -12,7 +11,6 @@ def _check_minimum_version(
     required = (major, minor, micro)
     return current[:3] >= required
 
-
 def _check_exact_version(
     major: int, minor: Optional[int] = 0, micro: Optional[int] = 0
 ):