def setUp(self): Host.destroy_all()
Datacenter.destroy_all() datacenter_cache = {} for datacenter_name, datacenter_desc in data["Datacenters"].items(): dc = Datacenter(name=datacenter_name, human_readable=datacenter_desc["golem_name"]) dc.save() datacenter_cache[datacenter_name] = dc print "Setting parents" for datacenter_name, datacenter_desc in data["Datacenters"].items(): if datacenter_desc["parent"] is not None: dc = datacenter_cache[datacenter_name] parent = datacenter_cache[datacenter_desc["parent"]] dc.set_parent(parent) print "Processing hosts" Host.destroy_all() i = 0 for fqdn, host_desc in data["Hosts"].items(): if host_desc["datacenter"] is not None: dc_id = datacenter_cache[host_desc["datacenter"]]._id else: dc_id = None h = Host(fqdn=fqdn, datacenter_id=dc_id, group_id=group_cache[host_desc["group"]]._id, short_name=host_desc["short_name"]) h.save() i += 1 if i % 1000 == 0: print i
def tearDownClass(cls): User.destroy_all() Project.destroy_all() Group.destroy_all() Host.destroy_all()