from sqlalchemy.orm import Session
from module_admin.entity.do.tsmcb_do import SysTsmcb
from module_admin.entity.vo.tsmcb_vo import TsmcbModel
from utils.time_format_util import list_format_datetime


class TsmcbDao:
    """
   词性文本替换模块数据库操作层
    """

    @classmethod
    def get_tsmcb_by_id(cls, db: Session, onum: int):
        """
        根据数据字典id获取在用数据字典详细信息
        :param db: orm对象
        :param onum:代码id
        :return: 在用词性文本替换信息对象
        """
        tsmcb_info = db.query(SysTsmcb) \
            .filter(SysTsmcb.onum == onum,
                    SysTsmcb.status == 0) \
            .first()

        return tsmcb_info

    @classmethod
    def get_tsmcb_detail_by_id(cls, db: Session, onum: int):
        """
        根据数据字典id获取数据字典详细信息
        :param db: orm对象
        :param onum:代码id
        :return:代码信息对象
        """
        tsmcb_info = db.query(SysTsmcb) \
            .filter(SysTsmcb.onum == onum) \
            .first()

        return tsmcb_info

    @classmethod
    def get_tsmcb_detail_by_info(cls, db: Session, tsmcb: TsmcbModel):
        """
        根据数据字典参数获取词性文本替换信息
        :param db: orm对象
        :param tsmcb:代码参数对象
        :return:代码信息对象
        """
        tsmcb_info = db.query(SysTsmcb) \
            .filter(SysTsmcb.pos == tsmcb.pos if tsmcb.pos else True,
                    SysTsmcb.type == tsmcb.type if tsmcb.type else True,
                    SysTsmcb.std_rpl_str == tsmcb.std_rpl_str if tsmcb.std_rpl_str else True,
                    SysTsmcb.status == tsmcb.status if tsmcb.status else True) \
            .first()

        return tsmcb_info

    @classmethod
    def get_tsmcb_select_option_dao(cls, db: Session):
        """
        获取所有在用词性文本替换信息
        :param db: orm对象
        :return: 在用词性文本替换信息列表
        """
        tsmcb_info = db.query(SysTsmcb) \
            .filter(SysTsmcb.status == 0) \
            .all()

        return tsmcb_info

    @classmethod
    def get_tsmcb_list(cls, db: Session, query_object: TsmcbModel):
        """
        根据查询参数获取数据字典列表信息
        :param db: orm对象
        :param query_object: 查询参数对象
        :return:代码列表信息对象 
        """
        tsmcb_list = db.query(SysTsmcb) \
            .filter(SysTsmcb.pos.like(f'%{query_object.pos}%') if query_object.pos else True,
                    SysTsmcb.type.like(f'%{query_object.type}%') if query_object.type else True,
                    SysTsmcb.std_rpl_str.like(f'%{query_object.std_rpl_str}%') if query_object.std_rpl_str else True,
                    SysTsmcb.status == query_object.status if query_object.status else True
                    ) \
            .order_by(SysTsmcb.onum) \
            .distinct().all()

        return list_format_datetime(tsmcb_list)

    @classmethod
    def add_tsmcb_dao(cls, db: Session, tsmcb: TsmcbModel):
        """
        新增数据字典数据库操作
        :param db: orm对象
        :param tsmcb:代码对象
        :return:
        """
        db_tsmcb = SysTsmcb(**tsmcb.dict())
        db.add(db_tsmcb)
        db.flush()

        return db_tsmcb

    @classmethod
    def edit_tsmcb_dao(cls, db: Session, tsmcb: dict):
        """
        编辑数据字典数据库操作
        :param db: orm对象 pos onum
        :param tsmcb: 需要更新的数据字典字典
        :return:
        """
        db.query(SysTsmcb) \
            .filter(SysTsmcb.pos == tsmcb.get('pos') ) \
            .update(tsmcb)
        db.query(SysTsmcb) \
            .filter(SysTsmcb.onum == tsmcb.get('onum') ) \
            .update(tsmcb)
        #(SysTsmcb.onum == tsmcb.get('onum'))

    @classmethod
    def delete_tsmcb_dao(cls, db: Session, tsmcb: TsmcbModel):
        """
        删除数据字典数据库操作
        :param db: orm对象
        :param tsmcb:代码对象
        :return:
        """
        db.query(SysTsmcb) \
            .filter(SysTsmcb.onum == tsmcb.onum) \
            .delete()

    @classmethod
    def get_tsmcb_by_info(cls, db: Session, tsmcb: TsmcbModel):
        """
        根据用户参数获取tsmcb信息
        :param db: orm对象
        :param user: tsmcb参数
        :return: 当前tsmcb参数的tsmcb信息对象
        """
        query_tsmcb_info = db.query(SysTsmcb) \
            .filter(SysTsmcb.status == 0,
                    SysTsmcb.onum == tsmcb.onum) \
            .order_by(SysTsmcb.onum).distinct().first()

        return query_tsmcb_info
    
    @classmethod
    def get_tsmcb_by_info_imp(cls, db: Session, tsmcb: TsmcbModel):
        """
        根据用户参数获取tsmcb信息
        :param db: orm对象
        :param user: tsmcb参数
        :return: 当前tsmcb参数的tsmcb信息对象
        """
        query_tsmcb_info_imp = db.query(SysTsmcb) \
            .filter(SysTsmcb.status == 0,
                    SysTsmcb.pos == tsmcb.pos) \
            .order_by(SysTsmcb.onum).distinct().first()

        return query_tsmcb_info_imp