from sqlalchemy.orm import Session

from APPData.Model import CompanyModel as Model


def create_lawsuit_by_id(db: Session, mongodb, _id, content):
    item = Model.Lawsuit()
    item._id = _id
    obj_id = mongodb.insert_data(dbname="WR_DataBase", sheet="C9.6_法律诉讼", data=content)
    item.obj_id = obj_id
    db.add(item)
    db.commit()
    return True


def update_lawsuit_by_id(mongodb, _id, content):
    mongodb.update_data(dbname="WR_DataBase", sheet="C9.6_法律诉讼", search={"id": _id}, field=content)
    return True


def find_lawsuit_by_id(db: Session, mongodb, _id):
    db_item = db.query(Model.Lawsuit).filter_by(_id=_id).all()
    data = list()
    for item in db_item:
        insert = mongodb.find_data_by_id(dbname="WR_DataBase", sheet="C9.6_法律诉讼", _id=item.obj_id)
        data.append(insert)
    return data