24 lines
757 B
Python
24 lines
757 B
Python
|
from sqlalchemy.orm import Session
|
||
|
|
|
||
|
|
from APPData.Model import CompanyModel as Model
|
||
|
|
|
||
|
|
|
||
|
|
def create_income_by_id(db: Session, mongodb, _id, reportdate, condition, content):
|
||
|
|
item = Model.Income()
|
||
|
|
item._id = _id
|
||
|
|
item.reportdate = reportdate
|
||
|
|
obj_id = mongodb.upsert_data(dbname="WR_DataBase_v2_test", sheet="C6.2_利润表", condition=condition, data=content)
|
||
|
|
item.obj_id = obj_id
|
||
|
|
db.add(item)
|
||
|
|
db.commit()
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def find_income_by_id(db: Session, mongodb, _id):
|
||
|
|
db_item = db.query(Model.Income).filter_by(_id=_id).all()
|
||
|
|
data = list()
|
||
|
|
for item in db_item:
|
||
|
|
insert = mongodb.find_data_by_id(dbname="WR_DataBase_v2_test", sheet="C6.2_利润表", _id=item.obj_id)
|
||
|
|
data.append(insert)
|
||
|
|
return data
|