
Untitled
By: a guest on
May 8th, 2012 | syntax:
None | size: 0.75 KB | hits: 10 | expires: Never
def import_users(infile, api, delete=False, dry_run=False, batch_size=1000):
"""Import from file infile user data. The input file is assumed to be in
csv format."""
reader = csv.DictReader(infile)
batch_nr = 1
batch = []
for nr, d in enumerate(reader):
batch.append(d)
if (nr % batch_size) == 0 and nr>0:
print "Importing batch %d ... " % batch_nr,
sys.stdout.flush()
start = time.time()
# import the users
if not dry_run:
result = api.bulk_import(batch, delete=delete)
print "done (%d seconds)." % (time.time() - start)
batch_nr = batch_nr + 1
# reset batch list
batch = []