from sqlalchemy.orm import Session
from module_admin.entity.do.ndstand_do import SysNdstand
from module_admin.entity.vo.ndstand_vo import NdstandModel
from utils.time_format_util import list_format_datetime


class NdstandDao:
    """
    数据标准管理模块数据库操作层 
    """

    @classmethod
    def get_ndstand_by_id(cls, db: Session, onum: int):
        """
        根据数据标准id获取在用数据标准详细信息
        :param db: orm对象
        :param onum: 数据标准id
        :return: 在用数据标准信息对象
        """
        ndstand_info = db.query(SysNdstand) \
            .filter(SysNdstand.onum == onum,
                    SysNdstand.status == 0) \
            .first()

        return ndstand_info

    @classmethod
    def get_ndstand_detail_by_id(cls, db: Session, onum: int):
        """
        根据数据标准id获取数据标准详细信息
        :param db: orm对象
        :param onum: 数据标准id
        :return: 数据标准信息对象
        """
        ndstand_info = db.query(SysNdstand) \
            .filter(SysNdstand.onum == onum) \
            .first()

        return ndstand_info

    @classmethod
    def get_all_ndstand(cls, db: Session):
        """
        获取所有的代码表信息
        :param db: orm对象
        :return: 代码表信息列表对象
        """
        ndstand_info = db.query(SysNdstand).all()

        return list_format_datetime(ndstand_info)


    @classmethod
    def get_ndstand_detail_by_info(cls, db: Session, ndstand: NdstandModel):
        """
        根据数据标准参数获取数据标准信息
        :param db: orm对象
        :param ndstand: 数据标准参数对象
        :return: 数据标准信息对象
        """
        ndstand_info = db.query(SysNdstand) \
            .filter(SysNdstand.data_std_no == ndstand.data_std_no if ndstand.data_std_no else True,
                    SysNdstand.data_std_cn_name == ndstand.data_std_cn_name if ndstand.data_std_cn_name else True,
                    SysNdstand.busi_defn == ndstand.busi_defn if ndstand.busi_defn else True) \
            .first()

        return ndstand_info

    @classmethod
    def get_ndstand_select_option_dao(cls, db: Session):
        """
        获取所有在用数据标准信息
        :param db: orm对象
        :return: 在用数据标准信息列表
        """
        ndstand_info = db.query(SysNdstand) \
            .filter(SysNdstand.status == 0) \
            .all()

        return ndstand_info

    @classmethod
    def get_ndstand_list(cls, db: Session, query_object: NdstandModel):
        """
        根据查询参数获取数据标准列表信息
        :param db: orm对象
        :param query_object: 查询参数对象
        :return: 数据标准列表信息对象
        """
        ndstand_list = db.query(SysNdstand) \
            .filter(SysNdstand.data_std_schm.like(f'%{query_object.data_std_schm}%') if query_object.data_std_schm else True,
                    SysNdstand.std_pri_clas.like(f'%{query_object.std_pri_clas}%') if query_object.std_pri_clas else True,
                    SysNdstand.std_scd_clas.like(f'%{query_object.std_scd_clas}%') if query_object.std_scd_clas else True,
                    SysNdstand.std_thre_clas.like(f'%{query_object.std_thre_clas}%') if query_object.std_thre_clas else True,
                    SysNdstand.data_std_no.like(f'%{query_object.data_std_no}%') if query_object.data_std_no else True,
                    SysNdstand.data_std_cn_name.like(f'%{query_object.data_std_cn_name}%') if query_object.data_std_cn_name else True,
					SysNdstand.busi_defn.like(f'%{query_object.busi_defn}%') if query_object.busi_defn else True,
                    SysNdstand.status == query_object.status if query_object.status else True
                    ) \
            .order_by(SysNdstand.onum) \
            .distinct().all()

        return list_format_datetime(ndstand_list)

    @classmethod
    def add_ndstand_dao(cls, db: Session, ndstand: NdstandModel):
        """
        新增数据标准数据库操作
        :param db: orm对象
        :param ndstand: 数据标准对象
        :return:
        """
        db_ndstand = SysNdstand(**ndstand.dict())
        db.add(db_ndstand)
        db.flush()

        return db_ndstand

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

    @classmethod
    def delete_ndstand_dao(cls, db: Session, ndstand: NdstandModel):
        """
        删除数据标准数据库操作
        :param db: orm对象
        :param ndstand: 数据标准对象
        :return:
        """
        db.query(SysNdstand) \
            .filter(SysNdstand.onum == ndstand.onum) \
            .delete()


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

        return query_ndstand_info