Dataclass improvement from testing

pythereum/dclasses.py

@@ -59,25 +59,33 @@ def hex_list_decoder(hex_string_list: list[str] | None) -> list[HexStr] | None:
         return None
 
 
-def hex_list_encoder(hex_obj_list: list[HexStr]) -> list[str] | None:
+def hex_list_encoder(hex_obj_list: list[HexStr] | None) -> list[str] | None:
     if hex_obj_list is not None:
         return [hex_encoder(hex_obj) for hex_obj in hex_obj_list]
     else:
         return None
 
 
-def transaction_decoder(transaction_hex: dict | str) -> "TransactionFull | HexStr":
+def transaction_decoder(
+    transaction_hex: dict | str | None,
+) -> "TransactionFull | HexStr | None":
     if isinstance(transaction_hex, dict):
         return TransactionFull.from_dict(transaction_hex, infer_missing=True)
-    else:
+    elif transaction_hex is not None:
         return hex_decoder(transaction_hex)
+    else:
+        return None
 
 
-def transaction_encoder(transaction_obj: "HexStr | TransactionFull") -> str | dict:
+def transaction_encoder(
+    transaction_obj: "HexStr | TransactionFull | None",
+) -> str | dict | None:
     if isinstance(transaction_obj, TransactionFull):
         return transaction_obj.to_dict()
-    else:
+    elif transaction_obj is not None:
         return hex_encoder(transaction_obj)
+    else:
+        return None
 
 
 def transaction_list_decoder(
@@ -160,6 +168,7 @@ class Block:
     """
     Full information about a block, either including full transactions or transaction hashes
     """
+
     # Integer of the difficulty for the block
     difficulty: int | None = field(
         metadata=config(decoder=hex_int_decoder, encoder=hex_int_encoder)
@@ -292,6 +301,7 @@ class Receipt:
     """
     A receipt generated by a transaction's execution
     """
+
     # 32 Byte hash of transaction
     transaction_hash: HexStr | None = field(
         metadata=config(decoder=hex_decoder, encoder=hex_encoder)
@@ -374,6 +384,7 @@ class Log:
     """
     A log generated by smart contract event triggers
     """
+
     address: HexStr | None = field(
         metadata=config(decoder=hex_decoder, encoder=hex_encoder)
     )
@@ -398,7 +409,7 @@ class Log:
     transaction_index: int | None = field(
         metadata=config(decoder=hex_int_decoder, encoder=hex_int_encoder)
     )
-    removed: bool
+    removed: bool | None
 
 
 @dataclass_json(letter_case=LetterCase.CAMEL)
@@ -407,6 +418,7 @@ class TransactionFull:
     """
     The full information on a transaction to be executed, including metadata with reference to its inclusion on chain
     """
+
     block_hash: HexStr | None = field(
         metadata=config(decoder=hex_decoder, encoder=hex_encoder)
     )
@@ -572,6 +584,7 @@ class Bundle(dict):
     """
     A bundle of transactions to submit to a block builder
     """
+
     def __init__(
         self,
         txs: list[str] | list[HexStr],