from sqlalchemy.orm import Session
from module_admin.entity.do.fddict_do import SysFddict
from module_admin.entity.vo.fddict_vo import FddictModel
from utils.time_format_util import list_format_datetime


class FddictDao:
    """
   代码管理模块数据库操作层
    """

    @classmethod
    def get_fddict_by_id(cls, db: Session, fddict_id: int):
        """
        根据数据字典id获取在用数据字典详细信息
        :param db: orm对象
        :param fddict_id:代码id
        :return: 在用代码管理信息对象
        """
        fddict_info = db.query(SysFddict) \
            .filter(SysFddict.fddict_id == fddict_id,
                    SysFddict.status == 0) \
            .first()

        return fddict_info

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

        return fddict_info

    @classmethod
    def get_fddict_detail_by_info(cls, db: Session, fddict: FddictModel):
        """
        根据数据字典参数获取代码管理信息
        :param db: orm对象
        :param fddict:代码参数对象
        :return:代码信息对象
        """
        fddict_info = db.query(SysFddict) \
            .filter(SysFddict.fddict_name == fddict.fddict_name if fddict.fddict_name else True,
                    SysFddict.fddict_code == fddict.fddict_code if fddict.fddict_code else True,
                    SysFddict.remark == fddict.remark if fddict.remark else True,
                    SysFddict.fddict_sort == fddict.fddict_sort if fddict.fddict_sort else True) \
            .first()

        return fddict_info

    @classmethod
    def get_fddict_select_option_dao(cls, db: Session):
        """
        获取所有在用代码管理信息
        :param db: orm对象
        :return: 在用代码管理信息列表
        """
        fddict_info = db.query(SysFddict) \
            .filter(SysFddict.status == 0) \
            .all()

        return fddict_info

    @classmethod
    def get_fddict_list(cls, db: Session, query_object: FddictModel):
        """
        根据查询参数获取数据字典列表信息
        :param db: orm对象
        :param query_object: 查询参数对象
        :return:代码列表信息对象 
        """
        fddict_list = db.query(SysFddict) \
            .filter(SysFddict.fddict_col_no.like(f'%{query_object.fddict_col_no}%') if query_object.fddict_col_no else True,
                    SysFddict.fddict_code.like(f'%{query_object.fddict_code}%') if query_object.fddict_code else True,
                    SysFddict.fddict_name.like(f'%{query_object.fddict_name}%') if query_object.fddict_name else True,
                    SysFddict.remark.like(f'%{query_object.remark}%') if query_object.remark else True,
                    SysFddict.status == query_object.status if query_object.status else True
                    ) \
            .order_by(SysFddict.fddict_id,SysFddict.fddict_sort) \
            .distinct().all()

        return list_format_datetime(fddict_list)

    @classmethod
    def add_fddict_dao(cls, db: Session, fddict: FddictModel):
        """
        新增数据字典数据库操作
        :param db: orm对象
        :param fddict:代码对象
        :return:
        """
        db_fddict = SysFddict(**fddict.dict())
        db.add(db_fddict)
        db.flush()

        return db_fddict

    @classmethod
    def edit_fddict_dao(cls, db: Session, fddict: dict):
        """
        编辑数据字典数据库操作
        :param db: orm对象
        :param fddict: 需要更新的数据字典字典
        :return:
        """
        db.query(SysFddict) \
            .filter(SysFddict.fddict_id == fddict.get('fddict_id')) \
            .update(fddict)

    @classmethod
    def delete_fddict_dao(cls, db: Session, fddict: FddictModel):
        """
        删除数据字典数据库操作
        :param db: orm对象
        :param fddict:代码对象
        :return:
        """
        db.query(SysFddict) \
            .filter(SysFddict.fddict_id == fddict.fddict_id) \
            .delete()

    @classmethod
    def get_fddict_by_info(cls, db: Session, fddict: FddictModel):
        """
        根据用户参数获取fddict信息
        :param db: orm对象
        :param user: fddict参数
        :return: 当前fddict参数的fddict信息对象
        """
        query_fddict_info = db.query(SysFddict) \
            .filter(SysFddict.status == 0,
                    SysFddict.fddict_id == fddict.fddict_id) \
            .order_by(SysFddict.fddict_id).distinct().first()

        return query_fddict_info