from sqlalchemy.orm import Session

from App.Model import RatingConfigModel as Model
from Utils.UniqueCoder.TimeSerialNumUtils import create_time_serial_num


def create_rank_table(db: Session, mongodb, _id, content):
    item = Model.ScoreCardRankTable()
    item.id = create_time_serial_num()
    item.scorecard_id = _id
    obj_id = mongodb.insert_data(dbname="打分卡配置", sheet="评级表", data=content)
    item.obj_id = obj_id
    db.add(item)
    db.commit()
    return True


def delete_rank_table(db: Session, mongodb, _id):
    item = db.query(Model.ScoreCardRankTable).filter_by(scorecard_id=_id).first()
    mongodb.delete_data_by_id(dbname="打分卡配置", sheet="评级表", _id=item.obj_id)
    db.query(Model.ScoreCardRankTable).filter_by(id=item.id).delete()
    db.commit()
    return True