flex-options

from confluent_kafka import Consumer
from unusualwhales.schema_pb2 import FlexConsolidation, FlexOptionState

# The record key selects the message type.
BY_KEY = {
    b"consolidation": FlexConsolidation,
    b"option-state": FlexOptionState,
}

# CONFIG comes from the Connecting section of the Kafka overview.
consumer = Consumer(CONFIG | {"group.id": "my-flex-options-consumer"})
consumer.subscribe(["flex-options"])

while True:
    msg = consumer.poll(1.0)
    if msg is None or msg.error():
        continue

    message_type = BY_KEY.get(msg.key())
    if message_type is None:
        continue

    event = message_type()
    event.ParseFromString(msg.value())
    print(event)