Beispiel #1
0
def test_wh_save_and_load():
    name = "test_warehouse_%s" % time.time()
    config_url = REMOTE_CONFIG_URL
    wh = Warehouse(config=config_url)
    wh_id = wh.save(name, config_url, meta=dict(test=1))
    wh = Warehouse.load(wh_id)
    assert wh.name == name
    Warehouse.delete(wh_id)
Beispiel #2
0
def init_warehouse_data(db: Session) -> None:
    whs_json_file = os.getenv("INITIAL_WAREHOUSES_FILE", None)
    if not whs_json_file:
        print("#### No INITIAL_WAREHOUSES_FILE set")
        return

    print(f"#### Adding Initial Warehouses from {whs_json_file}")

    whs = json.loads(read_filepath_or_buffer(whs_json_file))
    engine = db.get_bind()

    for wh in whs:
        print("Adding Warehouse...")
        print(wh)
        engine.execute(
            Warehouses.insert(),
            id=wh["id"],
            name=wh["name"],
            params=json.dumps(wh["params"]),
            meta=None,
        )

        # Load to make sure any required data syncs occur before app start
        Warehouse.load(wh["id"])
Beispiel #3
0
def get_warehouses() -> Dict[str, Any]:
    """NOTE: this assumes Zillion Web DB is same as Zillion DB"""
    global warehouses
    if warehouses:
        # TODO: cache control?
        return warehouses

    print("Building warehouses...")
    db = SessionLocal()
    try:
        result = db.query(Warehouses).all()
        for row in result:
            warehouses[row.id] = Warehouse.load(row.id)
        pp(warehouses)
        return warehouses
    finally:
        db.close()