kafka_server.py

import producer_server


def run_kafka_server():
    input_file = "police-department-calls-for-service.json"

    producer = producer_server.ProducerServer(
        input_file=input_file,
        topic="sf.police.department.calls.data",
        bootstrap_servers="localhost:9092",
        client_id="sf.police.department.calls.producer"
    )

    return producer


def feed():
    producer = run_kafka_server()
    producer.generate_data()


if __name__ == "__main__":
    feed()