from sqlalchemy.orm import Session

from APPData.Model import CompanyModel as Model


def create_business_by_id(db: Session, mongodb, _id, content):
    item = Model.Business()
    item._id = _id
    obj_id = mongodb.insert_data(dbname="WR_DataBase", sheet="C3.1_工商信息", data=content)
    item.obj_id = obj_id
    db.add(item)
    db.commit()
    return True


def update_business_by_id(db: Session, mongodb, _id, content):
    item = db.query(Model.Business).filter_by(_id=_id).first()
    mongodb.update_data_by_id(dbname="WR_DataBase", sheet="C3.1_工商信息", _id=item.obj_id, data=content)
    return True


def find_business_by_id(db: Session, mongodb, _id):
    item = db.query(Model.Business).filter_by(_id=_id).first()
    data = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C3.1_工商信息", _id=item.obj_id)
    return data