56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
# dev tests
|
|
# to be deleted later
|
|
|
|
import betterproto
|
|
from dataclasses import dataclass
|
|
from typing import Dict, List, Optional
|
|
|
|
@dataclass(repr=False)
|
|
class Baz(betterproto.Message):
|
|
a: float = betterproto.float_field(1, group = "x")
|
|
b: int = betterproto.int64_field(2, group = "x")
|
|
c: float = betterproto.float_field(3, group = "y")
|
|
d: int = betterproto.int64_field(4, group = "y")
|
|
e: Optional[int] = betterproto.int32_field(5, group = "_e", optional = True)
|
|
|
|
@dataclass(repr=False)
|
|
class Foo(betterproto.Message):
|
|
x: int = betterproto.int32_field(1)
|
|
y: float = betterproto.double_field(2)
|
|
z: List[Baz] = betterproto.message_field(3)
|
|
|
|
class Enm(betterproto.Enum):
|
|
A = 0
|
|
B = 1
|
|
C = 2
|
|
|
|
@dataclass(repr=False)
|
|
class Bar(betterproto.Message):
|
|
foo1: Foo = betterproto.message_field(1)
|
|
foo2: Foo = betterproto.message_field(2)
|
|
packed: List[int] = betterproto.int64_field(3)
|
|
enm: Enm = betterproto.enum_field(4)
|
|
map: Dict[int, bool] = betterproto.map_field(5, betterproto.TYPE_INT64, betterproto.TYPE_BOOL)
|
|
maybe: Optional[bool] = betterproto.message_field(6, wraps=betterproto.TYPE_BOOL)
|
|
bts: bytes = betterproto.bytes_field(7)
|
|
|
|
# Serialization has not been changed yet. So nothing unusual here
|
|
buffer = bytes(
|
|
Bar(
|
|
foo1=Foo(1, 2.34),
|
|
foo2=Foo(3, 4.56, [Baz(a = 1.234), Baz(b = 5, e=1), Baz(b = 2, d = 3)]),
|
|
packed=[5, 3, 1],
|
|
enm=Enm.B,
|
|
map={
|
|
1: True,
|
|
42: False
|
|
},
|
|
maybe=True,
|
|
bts=b'Hi There!'
|
|
)
|
|
)
|
|
|
|
# Native deserialization happening here
|
|
bar = Bar().parse(buffer)
|
|
print(bar)
|