Fix deserialize type signature for receipt and txn

Author: carver

eth/abc.py

@@ -11,6 +11,7 @@
     FrozenSet,
     Iterable,
     Iterator,
+    List,
     MutableMapping,
     NamedTuple,
     Optional,
@@ -51,6 +52,9 @@
 
 T = TypeVar('T')
 
+# A decoded RLP object of unknown interpretation, with a maximum "depth" of 1.
+DecodedZeroOrOneLayerRLP = Union[bytes, List[bytes]]
+
 
 class MiningHeaderAPI(ABC):
     """
@@ -241,7 +245,7 @@ class ReceiptBuilderAPI(ReceiptDecoderAPI):
 
     @classmethod
     @abstractmethod
-    def deserialize(cls, encoded: bytes) -> 'ReceiptAPI':
+    def deserialize(cls, encoded: DecodedZeroOrOneLayerRLP) -> 'ReceiptAPI':
         """
         Extract a receipt from an encoded RLP object.
 
@@ -251,7 +255,7 @@ def deserialize(cls, encoded: bytes) -> 'ReceiptAPI':
 
     @classmethod
     @abstractmethod
-    def serialize(cls, obj: 'ReceiptAPI') -> bytes:
+    def serialize(cls, obj: 'ReceiptAPI') -> DecodedZeroOrOneLayerRLP:
         """
         Encode a receipt to a series of bytes used by RLP.
 
@@ -448,7 +452,7 @@ class TransactionBuilderAPI(TransactionDecoderAPI):
 
     @classmethod
     @abstractmethod
-    def deserialize(cls, encoded: bytes) -> 'SignedTransactionAPI':
+    def deserialize(cls, encoded: DecodedZeroOrOneLayerRLP) -> 'SignedTransactionAPI':
         """
         Extract a transaction from an encoded RLP object.
 
@@ -458,7 +462,7 @@ def deserialize(cls, encoded: bytes) -> 'SignedTransactionAPI':
 
     @classmethod
     @abstractmethod
-    def serialize(cls, obj: 'SignedTransactionAPI') -> bytes:
+    def serialize(cls, obj: 'SignedTransactionAPI') -> DecodedZeroOrOneLayerRLP:
         """
         Encode a transaction to a series of bytes used by RLP.
 

eth/vm/forks/berlin/receipts.py

@@ -17,6 +17,7 @@
 )
 
 from eth.abc import (
+    DecodedZeroOrOneLayerRLP,
     LogAPI,
     ReceiptAPI,
     ReceiptBuilderAPI,
@@ -76,13 +77,13 @@ def get_payload_codec(cls, type_id: int) -> Type[ReceiptDecoderAPI]:
             raise ValidationError(f"Cannot build typed receipt with {hex(type_id)} >= 0x80")
 
     @classmethod
-    def deserialize(cls, encoded_unchecked: bytes) -> ReceiptAPI:
+    def deserialize(cls, encoded_unchecked: DecodedZeroOrOneLayerRLP) -> ReceiptAPI:
         # binary checks a few basics, like the length of the bytes
         encoded = cls.rlp_type.deserialize(encoded_unchecked)
         return cls.decode(encoded)
 
     @classmethod
-    def serialize(cls, obj: 'TypedReceipt') -> bytes:
+    def serialize(cls, obj: 'TypedReceipt') -> DecodedZeroOrOneLayerRLP:
         encoded = obj.encode()
         return cls.rlp_type.serialize(encoded)
 
@@ -136,12 +137,12 @@ def decode(cls, encoded: bytes) -> ReceiptAPI:
             return rlp.decode(encoded, sedes=cls.legacy_sedes)
 
     @classmethod
-    def deserialize(cls, encoded: bytes) -> ReceiptAPI:
+    def deserialize(cls, encoded: DecodedZeroOrOneLayerRLP) -> ReceiptAPI:
         if isinstance(encoded, bytes):
             return TypedReceipt.deserialize(encoded)
         else:
             return cls.legacy_sedes.deserialize(encoded)
 
     @classmethod
-    def serialize(cls, obj: ReceiptAPI) -> bytes:
+    def serialize(cls, obj: ReceiptAPI) -> DecodedZeroOrOneLayerRLP:
         return cls.legacy_sedes.serialize(obj)

eth/vm/forks/berlin/transactions.py

@@ -28,6 +28,7 @@
 )
 
 from eth.abc import (
+    DecodedZeroOrOneLayerRLP,
     ReceiptAPI,
     SignedTransactionAPI,
     TransactionBuilderAPI,
@@ -254,12 +255,12 @@ def decode(cls, encoded: bytes) -> SignedTransactionAPI:
         return cls(type_id, inner_transaction)
 
     @classmethod
-    def serialize(cls, obj: 'TypedTransaction') -> bytes:
+    def serialize(cls, obj: 'TypedTransaction') -> DecodedZeroOrOneLayerRLP:
         encoded = obj.encode()
         return cls.rlp_type.serialize(encoded)
 
     @classmethod
-    def deserialize(cls, encoded_unchecked: bytes) -> SignedTransactionAPI:
+    def deserialize(cls, encoded_unchecked: DecodedZeroOrOneLayerRLP) -> SignedTransactionAPI:
         # binary checks a few basics, like the length of the bytes
         encoded = cls.rlp_type.deserialize(encoded_unchecked)
         return cls.decode(encoded)
@@ -374,14 +375,14 @@ def decode(cls, encoded: bytes) -> SignedTransactionAPI:
             return rlp.decode(encoded, sedes=cls.legacy_signed)
 
     @classmethod
-    def deserialize(cls, encoded: bytes) -> SignedTransactionAPI:
+    def deserialize(cls, encoded: DecodedZeroOrOneLayerRLP) -> SignedTransactionAPI:
         if isinstance(encoded, bytes):
             return TypedTransaction.deserialize(encoded)
         else:
             return cls.legacy_signed.deserialize(encoded)
 
     @classmethod
-    def serialize(cls, obj: SignedTransactionAPI) -> bytes:
+    def serialize(cls, obj: SignedTransactionAPI) -> DecodedZeroOrOneLayerRLP:
         if isinstance(obj, TypedTransaction):
             return TypedTransaction.serialize(obj)
         else: