construct-typing

import dataclasses import typing as t from construct import Array, Byte, Const, Int8ub, this from construct_typed import DataclassMixin, DataclassStruct, TEnum, csfield # Define a typed Enum class Orientation(TEnum, Int8ub): HORIZONTAL: t.ClassVar[int] = 0 VERTICAL: t.ClassVar[int] = 1 # Define a typed Struct using dataclasses @dataclasses.dataclass class Image(DataclassMixin): signature: bytes = csfield(Const(b"BMP")) orientation: Orientation = csfield(TEnum(Int8ub, Orientation)) width: int = csfield(Int8ub) height: int = csfield(Int8ub) pixels: t.List[int] = csfield(Array(this.width * this.height, Byte)) # Example usage: Parse binary data img_bytes = b"BMP\x00\x03\x02\x07\x08\t\x0b\x0c\r" parsed_image = Image.parse(img_bytes) print(f"Parsed Image: {parsed_image}") # Expected: Parsed Image: Image(signature=b'BMP', orientation=<Orientation.HORIZONTAL: 0>, width=3, height=2, pixels=[7, 8, 9, 11, 12, 13]) # Example usage: Build binary data from a typed object built_bytes = Image.build( Image( orientation=Orientation.HORIZONTAL, width=3, height=2, pixels=[7, 8, 9, 11, 12, 13] ) ) print(f"Built Bytes: {built_bytes}") # Expected: Built Bytes: b'BMP\x00\x03\x02\x07\x08\t\x0b\x0c\r'