python: block types and wrapper

bindings/python/iota_sdk/types/block/basic.py

@@ -3,10 +3,10 @@
 
 from __future__ import annotations
 from dataclasses import dataclass, field
-from typing import List, Optional, Union
+from typing import List, Optional
 from iota_sdk.types.block.block import Block, BlockType
 from iota_sdk.types.common import HexStr, json
-from iota_sdk.types.payload import TaggedDataPayload, TransactionPayload
+from iota_sdk.types.payload import PayloadUnion
 
 
 @json
@@ -26,8 +26,7 @@ class BasicBlock(Block):
     weak_parents: List[HexStr]
     shallow_like_parents: List[HexStr]
     max_burned_mana: str
-    payload: Optional[Union[TaggedDataPayload,
-                      TransactionPayload]] = None
+    payload: Optional[PayloadUnion] = None
     type: int = field(
         default_factory=lambda: BlockType.Basic,
         init=False)

bindings/python/iota_sdk/types/block/wrapper.py

@@ -3,14 +3,16 @@
 
 from __future__ import annotations
 from dataclasses import dataclass
-from typing import Union
+from typing import TypeAlias, Union
 from iota_sdk.types.block.basic import BasicBlock
 from iota_sdk.types.block.validation import ValidationBlock
 from iota_sdk.types.common import HexStr, json, SlotIndex
 from iota_sdk.types.node_info import ProtocolParameters
-from iota_sdk.types.signature import Ed25519Signature
+from iota_sdk.types.signature import SignatureUnion
 from iota_sdk.utils import Utils
 
+BlockUnion: TypeAlias = Union[BasicBlock, ValidationBlock]
+
 
 @json
 @dataclass
@@ -34,8 +36,8 @@ class BlockWrapper:
     slot_commitment_id: HexStr
     latest_finalized_slot: SlotIndex
     issuer_id: HexStr
-    block: Union[BasicBlock, ValidationBlock]
-    signature: Union[Ed25519Signature]
+    block: BlockUnion
+    signature: SignatureUnion
 
     def id(self, params: ProtocolParameters) -> HexStr:
         """Returns the block ID as a hexadecimal string.

bindings/python/iota_sdk/types/signature.py

@@ -1,6 +1,7 @@
 # Copyright 2023 IOTA Stiftung
 # SPDX-License-Identifier: Apache-2.0
 
+from typing import TypeAlias, Union
 from dataclasses import dataclass, field
 from iota_sdk.types.common import HexStr, CoinType, json
 
@@ -28,6 +29,9 @@ class Ed25519Signature(Signature):
     type: int = field(default_factory=lambda: 0, init=False)
 
 
+SignatureUnion: TypeAlias = Union[Ed25519Signature]
+
+
 @json
 @dataclass
 class Bip44():