python kerberos solr

#!/usr/local/bin/python

from contextlib import contextmanager
from datetime import datetime
from itertools import islice
from random import choice, randint
from string import lowercase
import pycurl
import pprint
import json
import cStringIO

MAX_RANDOM_NUMBER = 10000000
LENGTH_RANDOM_STRING = 20

@contextmanager
def time_function(action_name):
        """ A context manager that prints out how long a function took to run to stdout. """

        start = datetime.now()
        yield
        end = datetime.now()

        delta = end - start
        print '%s executed in %ss.' % (action_name, delta.total_seconds())
def get_random_number():
        """ Return a random number. """

        return randint(1, MAX_RANDOM_NUMBER)

def get_random_string(size=LENGTH_RANDOM_STRING):
        """ Return a random string. """

        return ''.join(choice(lowercase) for x in range(size))

def make_solr_call():

        curl = pycurl.Curl()
        curl.setopt(pycurl.HTTPAUTH, pycurl.HTTPAUTH_GSSNEGOTIATE)
        curl.setopt(pycurl.USERPWD, ':')
        curl.setopt(pycurl.URL, 'https://server-name:port/solr/collection-name/select?q=*%3A*&wt=json')

        response = cStringIO.StringIO()
        curl.setopt(curl.WRITEFUNCTION, response.write)

        curl.perform()
        curl.close()

        raw_data =  response.getvalue()
        return raw_data

def to_pretty_json(raw_data):
        return json.dumps(json.loads(raw_data), indent=4, sort_keys=True)


def main():
        tests = [
                (2000000, 10000),
                (2000000, 100000),
        ]
        print "result :" +  to_pretty_json(make_solr_call())

if __name__ == "__main__":
        main()