Don't like ads? PRO users don't see any ads ;-)
Guest

Untitled

By: a guest on May 8th, 2012  |  syntax: None  |  size: 0.75 KB  |  hits: 10  |  expires: Never
download  |  raw  |  embed  |  report abuse  |  print
Text below is selected. Please press Ctrl+C to copy to your clipboard. (⌘+C on Mac)
  1. def import_users(infile, api, delete=False, dry_run=False, batch_size=1000):
  2.     """Import from file infile user data.  The input file is assumed to be in
  3.        csv format."""
  4.     reader = csv.DictReader(infile)
  5.  
  6.     batch_nr = 1
  7.     batch = []
  8.     for nr, d in enumerate(reader):
  9.         batch.append(d)
  10.         if (nr % batch_size) == 0 and nr>0:
  11.             print "Importing batch %d ... " % batch_nr,
  12.             sys.stdout.flush()
  13.             start = time.time()
  14.  
  15.             # import the users
  16.             if not dry_run:
  17.                 result = api.bulk_import(batch, delete=delete)
  18.  
  19.             print "done (%d seconds)." % (time.time() - start)
  20.             batch_nr = batch_nr + 1
  21.  
  22.             # reset batch list
  23.             batch = []