maybe enum in AVRO

maybe_enum.py

from io import BytesIO
import fastavro

schema = {
    "fields": [
        {
            "name": "maybe_enum",
            "type": ["null", {
                "type": "enum",
                "name": "methods",
                "symbols": [
                    "POST",
                    "GET",
                    "PUT",
                    "DELETE",
                    "OPTIONS",
                    "HEAD"
                ]
            }],
        },
    ],
    "namespace": "namespace",
    "name": "name",
    "type": "record"
}


def serialize(schema, data):
    bytes_writer = BytesIO()
    fastavro.schemaless_writer(bytes_writer, schema, data)
    return bytes_writer.getvalue()


def deserialize(schema, binary):
    bytes_writer = BytesIO()
    bytes_writer.write(binary)
    bytes_writer.seek(0)

    res = fastavro.schemaless_reader(bytes_writer, schema)
    return res


# ok
binary = serialize(schema, dict(maybe_enum=None))
print(deserialize(schema, binary))


binary = serialize(schema, dict(maybe_enum="GET"))
print(deserialize(schema, binary))