dump redis sorted set keys

dump_redis_sorted_set_keys.py

#!/usr/bin/env python

''' Dumps the location ids in a sorted set '''

import redis
import sys


def dump(key, redis_host, redis_port=6379):
    rconn = redis.StrictRedis(redis_host, port=redis_port)

    cursor, arr = rconn.zscan(key)
    for ulid, score in arr:
        print ulid
    
    while cursor != 0:
        cursor, arr = rconn.zscan(key, cursor=cursor)
        for ulid, score in arr:
            print ulid


if __name__ == '__main__':
    if len(sys.argv) != 2:
        print >> sys.stderr, (
            "Usage:\n"
            "  %s [sorted_set_key]\n"
            "e.g.\n"
            "  %s /my_sorted_set" % (
                sys.argv[0], sys.argv[0]))
        sys.exit(1)

    dump(sys.argv[1])