def main(): if not args.files: raise Exception("WTF - need a config") config = args.files.pop(0) obj = json.load(open(config, 'r')) machines = obj['machines'] configs = obj['configs'] users = obj['users'] puts("Loading users:") for conf in progress.bar(users): u = User(**conf) u.save() puts("Loading machines:") for conf in progress.bar(machines): m = Machine(**conf) m.save() puts("Loading configs:") for conf in progress.bar(configs): c = Config(**conf) c.save()
def setup(): connection = Connection('localhost', 27017) db = connection.lucy_test lucy.core.db = db for x in db.collection_names(): if x.startswith('system'): continue db.drop_collection(x) name = "default" incoming = "%s/incoming/" % (ROOT) pool = "%s/pool/" % (ROOT) if name != Config(_id=name, incoming=incoming, pool=pool).save(): raise Exception