Python script to get Kafka Brokers from Zookeeper

zookeeperKafka.py

from kazoo.client import KazooClient
import json

####
# A quick function to get Kafka brokers from Zookeeper
###
# Probably you need only the first one because the broker will advertise the other brokers
# This is need only for producers due there you can only use bootstrap servers
# Arguments
#       - zk: Zookeeper conn string
#       - path: The path what used for Kafka in Zookeeper
#


def getKafkaBroker(zk, path):
  zk = KazooClient(hosts=zk,read_only=True)
  zk.start()
  for node in zk.get_children(path+'/brokers/ids'):
    data, stats = zk.get(path+'/brokers/ids/'+node)
    props = json.loads(data)
    yield props['host']+':'+str(props['port'])
  zk.stop()


print next(getKafkaBroker('localhost:2171', 'kafka'))