dataplatform/APPData/Crud/BalanceCrud.py

24 lines
757 B
Python

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