dataplatform/APPData/Crud/BalanceCrud.py

from sqlalchemy.orm import Session

from APPData.Model import CompanyModel as Model


def create_balance_by_id(db: Session, mongodb, _id, reportdate, condition, content):
    item = Model.Balance()
    item._id = _id
    item.reportdate = reportdate
    obj_id = mongodb.upsert_data(dbname="WR_DataBase", sheet="C6.1_资产负债表", condition=condition, data=content)
    item.obj_id = obj_id
    db.add(item)
    db.commit()
    return True


def find_balance_by_id(db: Session, mongodb, _id):
    db_item = db.query(Model.Balance).filter_by(_id=_id).all()
    data = list()
    for item in db_item:
        insert = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C6.1_资产负债表", _id=item.obj_id)
        data.append(insert)
    return data