sec-filings

from confluent_kafka import Consumer
from unusualwhales.schema_pb2 import Form10K, Form13DG, Form13F_HR, Form345, Form497, GeneralFiling, InsiderTradeAgg, ThirteenFAlert

# The record key selects the message type.
BY_KEY = {
    b"13_f_alert": ThirteenFAlert,
    b"company-reports": Form10K,
    b"form13dg": Form13DG,
    b"form13f": Form13F_HR,
    b"form345": Form345,
    b"form497": Form497,
    b"general": GeneralFiling,
    b"insider_trade_agg": InsiderTradeAgg,
}

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

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)