def run():
for model, num_to_create in to_create.items():
model_name = model._meta.model_name
bar = Bar('Creating {}'.format(model_name), max=num_to_create)
model_count = model.objects.count()
create_f = globals()['populate_{}'.format(model_name)]
for i in range(num_to_create):
ident = '{}{}'.format(model_name, i)
if i < model_count:
unit = model.objects.all()[i]
else:
unit = create_f(model, i)
globals()[ident] = unit
bar.next()
bar.finish()
# This bit is special: Associate all rpms with the first repo,
# for maximum relational query fun
num_units = platform.ContentUnit.objects.count()
repo = globals()['repository0']
bar = Bar('Adding all units to {} repo'.format(repo.slug))
bar.max = num_units
for unit in platform.ContentUnit.objects.all():
repo.add_units(unit)
bar.next()
bar.finish()
