Kafka zookeeper cleanup script

kafka_zookeeper.py

import sys
from kazoo.client import KazooClient

"""
Simple python script to clean up zookeeper consumer offsets after a errant kafka consumer. Only tested on Kafka 0.7.

For details on Kazoo: http://kazoo.readthedocs.org/en/latest/index.html

MIT License.
"""


def main(zkconn_str, group_id):
    kc = KazooClient(zkconn_str)

    kc.start()
    assert(kc.connected)

    node = '/consumers/%s' % group_id
    print('Deleting consumer zk node (%s)' % node)
    ans = raw_input('Confirm by retyping consumer-group: ')
    if ans != group_id:
        sys.exit('Consumer group mismatch')
    kc.delete(node, recursive=True)
    print('Done. Exiting.')


if __name__ == '__main__':
    if len(sys.argv) < 3:
        sys.exit('Usage: [ZK-CONNECT-STRING] [KAFKA-CONSUMER-GROUP-ID]')

    main(sys.argv[1], sys.argv[2])