avro_gen_python_class_generation_serialization_deserialization.py

1import io
2import avro.schema
3from avro.io import DatumReader, DatumWriter, BinaryEncoder, BinaryDecoder
4
5# Note: In a typical workflow, you would first run the CLI tool to generate classes:
6# avro-generate-python --schema-file user.avsc --output-dir ./generated
7# Below is the implementation logic for using the generated classes.
8
9from generated.com.example import User
10
11# 1. Create an instance of the generated class
12user = User(name="John Doe", favorite_number=42, favorite_color="blue")
13
14# 2. Serialize the object to bytes
15writer = DatumWriter(user.SCHEMA)
16bytes_io = io.BytesIO()
17encoder = BinaryEncoder(bytes_io)
18writer.write(user.to_dict(), encoder)
19raw_bytes = bytes_io.getvalue()
20
21# 3. Deserialize the bytes back into an object
22bytes_io = io.BytesIO(raw_bytes)
23decoder = BinaryDecoder(bytes_io)
24reader = DatumReader(user.SCHEMA)
25data = reader.read(decoder)
26
27# 4. Instantiate the class from the decoded dictionary
28new_user = User.from_dict(data)
29
30print(f"Name: {new_user.name}")
31print(f"Favorite Number: {new_user.favorite_number}")