from module_admin.entity.vo.mmdt_vo import *
from module_admin.entity.vo.user_vo import *
from module_admin.dao.mmdt_dao import *
from utils.common_util import *
from utils.pwd_util import *


class MmdtService:
    """
    模型市场管理模块服务层
    """

    @classmethod
    def get_mmdt_select_option_services(cls, result_db: Session):
        """
        获取模型市场列表信息service
        :param result_db: orm对象
        :return: 模型市场列表信息对象
        """
        mmdt_list_result = MmdtDao.get_mmdt_select_option_dao(result_db)

        return mmdt_list_result

    @classmethod
    def get_mmdt_list_services(cls, result_db: Session, query_object: MmdtModel):
        """
        获取模型市场列表信息service
        :param result_db: orm对象
        :param query_object: 查询参数对象
        :return: 模型市场列表信息对象
        """
        mmdt_list_result = MmdtDao.get_mmdt_list(result_db, query_object)

        return mmdt_list_result

    @classmethod
    def add_mmdt_services(cls, result_db: Session, page_object: MmdtModel):
        """
        新增代码管理信息service
        :param result_db: orm对象
        :param page_object: 新增模型市场对象
        :return: 新增模型市场校验结果
        """
        mmdt = MmdtDao.get_mmdt_detail_by_info(result_db, MmdtModel(**dict(model_name_en=page_object.model_name_en)))
        if mmdt:
            result = dict(is_success=False, message='模型市场名称已存在')
        else:
            try:
                MmdtDao.add_mmdt_dao(result_db, page_object)
                result_db.commit()
                result = dict(is_success=True, message='新增成功')
            except Exception as e:
                result_db.rollback()
                result = dict(is_success=False, message=str(e))

        return CrudMmdtResponse(**result)

    @classmethod
    def edit_mmdt_services(cls, result_db: Session, page_object: MmdtModel):
        """
        编辑代码管理信息service
        :param result_db: orm对象
        :param page_object: 编辑模型市场对象
        :return: 编辑模型市场校验结果
        """
        edit_mmdt = page_object.dict(exclude_unset=True)
        mmdt_info = cls.detail_mmdt_services(result_db, edit_mmdt.get('onum'))
        if mmdt_info:
            if mmdt_info.model_name_en != page_object.model_name_en:
                mmdt = MmdtDao.get_mmdt_detail_by_info(result_db, MmdtModel(**dict(model_name_en=page_object.model_name_en)))
                if mmdt:
                    result = dict(is_success=False, message='模型市场名称已存在')
                    return CrudMmdtResponse(**result)
            try:
                MmdtDao.edit_mmdt_dao(result_db, edit_mmdt)
                result_db.commit()
                result = dict(is_success=True, message='更新成功')
            except Exception as e:
                result_db.rollback()
                result = dict(is_success=False, message=str(e))
        else:
            result = dict(is_success=False, message='模型市场不存在')

        return CrudMmdtResponse(**result)

    @classmethod
    def delete_mmdt_services(cls, result_db: Session, page_object: DeleteMmdtModel):
        """
        删除代码管理信息service
        :param result_db: orm对象
        :param page_object: 删除模型市场对象
        :return: 删除模型市场校验结果
        """
        if page_object.onums.split(','):
            onum_list = page_object.onums.split(',')
            try:
                for onum in onum_list:
                    onum_dict = dict(onum=onum)
                    MmdtDao.delete_mmdt_dao(result_db, MmdtModel(**onum_dict))
                result_db.commit()
                result = dict(is_success=True, message='删除成功')
            except Exception as e:
                result_db.rollback()
                result = dict(is_success=False, message=str(e))
        else:
            result = dict(is_success=False, message='传入模型市场id为空')
        return CrudMmdtResponse(**result)

    @classmethod
    def detail_mmdt_services(cls, result_db: Session, onum: int):
        """
        获取模型市场详细信息service
        :param result_db: orm对象
        :param onum: 模型市场id
        :return: 模型市场id对应的信息
        """
        mmdt = MmdtDao.get_mmdt_detail_by_id(result_db, onum=onum)

        return mmdt