From 0d6909760f0d7e264ece0de959bcbc6c9a1be30a Mon Sep 17 00:00:00 2001 From: haotian <2421912570@qq.com> Date: Wed, 30 Jul 2025 15:22:46 +0800 Subject: [PATCH] =?UTF-8?q?1.=E6=B7=BB=E5=8A=A0=E6=9C=BA=E5=99=A8=E4=BA=BA?= =?UTF-8?q?=E8=A7=92=E8=89=B2=E5=92=8C=E6=9C=BA=E5=99=A8=E4=BA=BA=E5=AF=B9?= =?UTF-8?q?=E5=BA=94=E8=A1=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../robot_role_pairing_controller.py | 103 ++++++++++++++ .../dao/robot_role_pairing_dao.py | 120 ++++++++++++++++ .../entity/do/robot_role_pairing_do.py | 20 +++ .../entity/vo/robot_role_pairing_vo.py | 78 +++++++++++ .../service/robot_role_pairing_service.py | 131 ++++++++++++++++++ 5 files changed, 452 insertions(+) create mode 100644 ruoyi-fastapi-backend/module_admin/controller/robot_role_pairing_controller.py create mode 100644 ruoyi-fastapi-backend/module_admin/dao/robot_role_pairing_dao.py create mode 100644 ruoyi-fastapi-backend/module_admin/entity/do/robot_role_pairing_do.py create mode 100644 ruoyi-fastapi-backend/module_admin/entity/vo/robot_role_pairing_vo.py create mode 100644 ruoyi-fastapi-backend/module_admin/service/robot_role_pairing_service.py diff --git a/ruoyi-fastapi-backend/module_admin/controller/robot_role_pairing_controller.py b/ruoyi-fastapi-backend/module_admin/controller/robot_role_pairing_controller.py new file mode 100644 index 0000000..f25bb60 --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/controller/robot_role_pairing_controller.py @@ -0,0 +1,103 @@ +from fastapi import APIRouter, Depends, Form, Request +from pydantic_validation_decorator import ValidateFields +from sqlalchemy.ext.asyncio import AsyncSession +from config.enums import BusinessType +from config.get_db import get_db +from module_admin.annotation.log_annotation import Log +from module_admin.aspect.interface_auth import CheckUserInterfaceAuth +from module_admin.entity.vo.user_vo import CurrentUserModel +from module_admin.service.login_service import LoginService +from module_admin.service.robot_role_pairing_service import PairingService +from module_admin.entity.vo.robot_role_pairing_vo import DeletePairingModel, PairingModel, PairingPageQueryModel +from utils.common_util import bytes2file_response +from utils.log_util import logger +from utils.page_util import PageResponseModel +from utils.response_util import ResponseUtil +from datetime import datetime + + +pairingController = APIRouter(prefix='/system/pairing', dependencies=[Depends(LoginService.get_current_user)]) + + +@pairingController.get( + '/list', response_model=PageResponseModel, dependencies=[Depends(CheckUserInterfaceAuth('system:pairing:list'))] +) +async def get_system_pairing_list( + request: Request, +pairing_page_query: PairingPageQueryModel = Depends(PairingPageQueryModel.as_query), + query_db: AsyncSession = Depends(get_db), +): + # 获取分页数据 + pairing_page_query_result = await PairingService.get_pairing_list_services(query_db, pairing_page_query, is_page=True) + logger.info('获取成功') + + return ResponseUtil.success(model_content=pairing_page_query_result) + + +@pairingController.post('', dependencies=[Depends(CheckUserInterfaceAuth('system:pairing:add'))]) +@ValidateFields(validate_model='add_pairing') +@Log(title='角色-机器人-映射', business_type=BusinessType.INSERT) +async def add_system_pairing( + request: Request, + add_pairing: PairingModel, + query_db: AsyncSession = Depends(get_db), + current_user: CurrentUserModel = Depends(LoginService.get_current_user), +): + add_pairing.update_time = datetime.now() + add_pairing.update_by = current_user.user.user_name + add_pairing_result = await PairingService.add_pairing_services(query_db, add_pairing) + logger.info(add_pairing_result.message) + + return ResponseUtil.success(msg=add_pairing_result.message) + + +@pairingController.put('', dependencies=[Depends(CheckUserInterfaceAuth('system:pairing:edit'))]) +@ValidateFields(validate_model='edit_pairing') +@Log(title='角色-机器人-映射', business_type=BusinessType.UPDATE) +async def edit_system_pairing( + request: Request, + edit_pairing: PairingModel, + query_db: AsyncSession = Depends(get_db), + current_user: CurrentUserModel = Depends(LoginService.get_current_user), +): + edit_pairing.update_by = current_user.user.user_name + edit_pairing.update_time = datetime.now() + edit_pairing_result = await PairingService.edit_pairing_services(query_db, edit_pairing) + logger.info(edit_pairing_result.message) + + return ResponseUtil.success(msg=edit_pairing_result.message) + + +@pairingController.delete('/{pairing_ids}', dependencies=[Depends(CheckUserInterfaceAuth('system:pairing:remove'))]) +@Log(title='角色-机器人-映射', business_type=BusinessType.DELETE) +async def delete_system_pairing(request: Request, pairing_ids: str, query_db: AsyncSession = Depends(get_db)): + delete_pairing = DeletePairingModel(pairingIds=pairing_ids) + delete_pairing_result = await PairingService.delete_pairing_services(query_db, delete_pairing) + logger.info(delete_pairing_result.message) + + return ResponseUtil.success(msg=delete_pairing_result.message) + + +@pairingController.get( + '/{pairing_id}', response_model=PairingModel, dependencies=[Depends(CheckUserInterfaceAuth('system:pairing:query'))] +) +async def query_detail_system_pairing(request: Request, pairing_id: int, query_db: AsyncSession = Depends(get_db)): + pairing_detail_result = await PairingService.pairing_detail_services(query_db, pairing_id) + logger.info(f'获取pairing_id为{pairing_id}的信息成功') + + return ResponseUtil.success(data=pairing_detail_result) + + +@pairingController.post('/export', dependencies=[Depends(CheckUserInterfaceAuth('system:pairing:export'))]) +@Log(title='角色-机器人-映射', business_type=BusinessType.EXPORT) +async def export_system_pairing_list( + request: Request, + pairing_page_query: PairingPageQueryModel = Form(), + query_db: AsyncSession = Depends(get_db), +): + # 获取全量数据 + pairing_query_result = await PairingService.get_pairing_list_services(query_db, pairing_page_query, is_page=False) + pairing_export_result = await PairingService.export_pairing_list_services(pairing_query_result) + logger.info('导出成功') + + return ResponseUtil.streaming(data=bytes2file_response(pairing_export_result)) diff --git a/ruoyi-fastapi-backend/module_admin/dao/robot_role_pairing_dao.py b/ruoyi-fastapi-backend/module_admin/dao/robot_role_pairing_dao.py new file mode 100644 index 0000000..d0b1abb --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/dao/robot_role_pairing_dao.py @@ -0,0 +1,120 @@ +from sqlalchemy import delete, select, update +from sqlalchemy.ext.asyncio import AsyncSession +from module_admin.entity.do.pairing_do import RobotRolePairing +from module_admin.entity.vo.pairing_vo import PairingModel, PairingPageQueryModel +from utils.page_util import PageUtil + + +class PairingDao: + """ + 角色-机器人-映射模块数据库操作层 + """ + + @classmethod + async def get_pairing_detail_by_id(cls, db: AsyncSession, pairing_id: int): + """ + 根据主键ID获取角色-机器人-映射详细信息 + + :param db: orm对象 + :param pairing_id: 主键ID + :return: 角色-机器人-映射信息对象 + """ + pairing_info = ( + ( + await db.execute( + select(RobotRolePairing) + .where( + RobotRolePairing.pairing_id == pairing_id + ) + ) + ) + .scalars() + .first() + ) + + return pairing_info + + @classmethod + async def get_pairing_detail_by_info(cls, db: AsyncSession, pairing: PairingModel): + """ + 根据角色-机器人-映射参数获取角色-机器人-映射信息 + + :param db: orm对象 + :param pairing: 角色-机器人-映射参数对象 + :return: 角色-机器人-映射信息对象 + """ + pairing_info = ( + ( + await db.execute( + select(RobotRolePairing).where( + ) + ) + ) + .scalars() + .first() + ) + + return pairing_info + + @classmethod + async def get_pairing_list(cls, db: AsyncSession, query_object: PairingPageQueryModel, is_page: bool = False): + """ + 根据查询参数获取角色-机器人-映射列表信息 + + :param db: orm对象 + :param query_object: 查询参数对象 + :param is_page: 是否开启分页 + :return: 角色-机器人-映射列表信息对象 + """ + query = ( + select(RobotRolePairing) + .where( + RobotRolePairing.pairing_id == query_object.pairing_id if query_object.pairing_id else True, + RobotRolePairing.robot_role_id == query_object.robot_role_id if query_object.robot_role_id else True, + RobotRolePairing.robot_id == query_object.robot_id if query_object.robot_id else True, + RobotRolePairing.status == query_object.status if query_object.status else True, + ) + .order_by(RobotRolePairing.pairing_id) + .distinct() + ) + pairing_list = await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page) + + return pairing_list + + @classmethod + async def add_pairing_dao(cls, db: AsyncSession, pairing: PairingModel): + """ + 新增角色-机器人-映射数据库操作 + + :param db: orm对象 + :param pairing: 角色-机器人-映射对象 + :return: + """ + db_pairing = RobotRolePairing(**pairing.model_dump(exclude={})) + db.add(db_pairing) + await db.flush() + + return db_pairing + + @classmethod + async def edit_pairing_dao(cls, db: AsyncSession, pairing: dict): + """ + 编辑角色-机器人-映射数据库操作 + + :param db: orm对象 + :param pairing: 需要更新的角色-机器人-映射字典 + :return: + """ + await db.execute(update(RobotRolePairing), [pairing]) + + @classmethod + async def delete_pairing_dao(cls, db: AsyncSession, pairing: PairingModel): + """ + 删除角色-机器人-映射数据库操作 + + :param db: orm对象 + :param pairing: 角色-机器人-映射对象 + :return: + """ + await db.execute(delete(RobotRolePairing).where(RobotRolePairing.pairing_id.in_([pairing.pairing_id]))) + diff --git a/ruoyi-fastapi-backend/module_admin/entity/do/robot_role_pairing_do.py b/ruoyi-fastapi-backend/module_admin/entity/do/robot_role_pairing_do.py new file mode 100644 index 0000000..47d738f --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/entity/do/robot_role_pairing_do.py @@ -0,0 +1,20 @@ +from sqlalchemy import DateTime, Column, BigInteger, String +from config.database import Base + + +class RobotRolePairing(Base): + """ + 角色-机器人-映射表 + """ + + __tablename__ = 'robot_role_pairing' + + pairing_id = Column(BigInteger, primary_key=True, autoincrement=True, nullable=False, comment='主键ID') + robot_role_id = Column(BigInteger, nullable=False, comment='机器人角色ID') + robot_id = Column(BigInteger, nullable=False, comment='机器人ID') + status = Column(String(2), nullable=False, comment='使用情况(0停用,1使用)') + update_time = Column(DateTime, nullable=False, comment='更新时间') + update_by = Column(String(64), nullable=False, comment='修改者') + + + diff --git a/ruoyi-fastapi-backend/module_admin/entity/vo/robot_role_pairing_vo.py b/ruoyi-fastapi-backend/module_admin/entity/vo/robot_role_pairing_vo.py new file mode 100644 index 0000000..8774d91 --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/entity/vo/robot_role_pairing_vo.py @@ -0,0 +1,78 @@ +from datetime import datetime +from pydantic import BaseModel, ConfigDict, Field +from pydantic.alias_generators import to_camel +from pydantic_validation_decorator import NotBlank +from typing import Optional +from module_admin.annotation.pydantic_annotation import as_query + + + + +class PairingModel(BaseModel): + """ + 角色-机器人-映射表对应pydantic模型 + """ + model_config = ConfigDict(alias_generator=to_camel, from_attributes=True) + + pairing_id: Optional[int] = Field(default=None, description='主键ID') + robot_role_id: Optional[int] = Field(default=None, description='机器人角色ID') + robot_id: Optional[int] = Field(default=None, description='机器人ID') + status: Optional[str] = Field(default=None, description='使用情况(0停用,1使用)') + update_time: Optional[datetime] = Field(default=None, description='更新时间') + update_by: Optional[str] = Field(default=None, description='修改者') + + @NotBlank(field_name='robot_role_id', message='机器人角色ID不能为空') + def get_robot_role_id(self): + return self.robot_role_id + + @NotBlank(field_name='robot_id', message='机器人ID不能为空') + def get_robot_id(self): + return self.robot_id + + @NotBlank(field_name='status', message='使用情况(0停用,1使用)不能为空') + def get_status(self): + return self.status + + @NotBlank(field_name='update_time', message='更新时间不能为空') + def get_update_time(self): + return self.update_time + + @NotBlank(field_name='update_by', message='修改者不能为空') + def get_update_by(self): + return self.update_by + + def validate_fields(self): + self.get_robot_role_id() + self.get_robot_id() + self.get_status() + self.get_update_time() + self.get_update_by() + + + + +class PairingQueryModel(PairingModel): + """ + 角色-机器人-映射不分页查询模型 + """ + pass + + +@as_query +class PairingPageQueryModel(PairingQueryModel): + """ + 角色-机器人-映射分页查询模型 + """ + + page_num: int = Field(default=1, description='当前页码') + page_size: int = Field(default=10, description='每页记录数') + + +class DeletePairingModel(BaseModel): + """ + 删除角色-机器人-映射模型 + """ + + model_config = ConfigDict(alias_generator=to_camel) + + pairing_ids: str = Field(description='需要删除的主键ID') diff --git a/ruoyi-fastapi-backend/module_admin/service/robot_role_pairing_service.py b/ruoyi-fastapi-backend/module_admin/service/robot_role_pairing_service.py new file mode 100644 index 0000000..3b8c2e1 --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/service/robot_role_pairing_service.py @@ -0,0 +1,131 @@ +from sqlalchemy.ext.asyncio import AsyncSession +from typing import List +from config.constant import CommonConstant +from exceptions.exception import ServiceException +from module_admin.entity.vo.common_vo import CrudResponseModel +from module_admin.dao.pairing_dao import PairingDao +from module_admin.entity.vo.pairing_vo import DeletePairingModel, PairingModel, PairingPageQueryModel +from utils.common_util import CamelCaseUtil +from utils.excel_util import ExcelUtil + + +class PairingService: + """ + 角色-机器人-映射模块服务层 + """ + + @classmethod + async def get_pairing_list_services( + cls, query_db: AsyncSession, query_object: PairingPageQueryModel, is_page: bool = False + ): + """ + 获取角色-机器人-映射列表信息service + + :param query_db: orm对象 + :param query_object: 查询参数对象 + :param is_page: 是否开启分页 + :return: 角色-机器人-映射列表信息对象 + """ + pairing_list_result = await PairingDao.get_pairing_list(query_db, query_object, is_page) + + return pairing_list_result + + + @classmethod + async def add_pairing_services(cls, query_db: AsyncSession, page_object: PairingModel): + """ + 新增角色-机器人-映射信息service + + :param query_db: orm对象 + :param page_object: 新增角色-机器人-映射对象 + :return: 新增角色-机器人-映射校验结果 + """ + try: + await PairingDao.add_pairing_dao(query_db, page_object) + await query_db.commit() + return CrudResponseModel(is_success=True, message='新增成功') + except Exception as e: + await query_db.rollback() + raise e + + @classmethod + async def edit_pairing_services(cls, query_db: AsyncSession, page_object: PairingModel): + """ + 编辑角色-机器人-映射信息service + + :param query_db: orm对象 + :param page_object: 编辑角色-机器人-映射对象 + :return: 编辑角色-机器人-映射校验结果 + """ + edit_pairing = page_object.model_dump(exclude_unset=True, exclude={}) + pairing_info = await cls.pairing_detail_services(query_db, page_object.pairing_id) + if pairing_info.pairing_id: + try: + await PairingDao.edit_pairing_dao(query_db, edit_pairing) + await query_db.commit() + return CrudResponseModel(is_success=True, message='更新成功') + except Exception as e: + await query_db.rollback() + raise e + else: + raise ServiceException(message='角色-机器人-映射不存在') + + @classmethod + async def delete_pairing_services(cls, query_db: AsyncSession, page_object: DeletePairingModel): + """ + 删除角色-机器人-映射信息service + + :param query_db: orm对象 + :param page_object: 删除角色-机器人-映射对象 + :return: 删除角色-机器人-映射校验结果 + """ + if page_object.pairing_ids: + pairing_id_list = page_object.pairing_ids.split(',') + try: + for pairing_id in pairing_id_list: + await PairingDao.delete_pairing_dao(query_db, PairingModel(pairingId=pairing_id)) + await query_db.commit() + return CrudResponseModel(is_success=True, message='删除成功') + except Exception as e: + await query_db.rollback() + raise e + else: + raise ServiceException(message='传入主键ID为空') + + @classmethod + async def pairing_detail_services(cls, query_db: AsyncSession, pairing_id: int): + """ + 获取角色-机器人-映射详细信息service + + :param query_db: orm对象 + :param pairing_id: 主键ID + :return: 主键ID对应的信息 + """ + pairing = await PairingDao.get_pairing_detail_by_id(query_db, pairing_id=pairing_id) + if pairing: + result = PairingModel(**CamelCaseUtil.transform_result(pairing)) + else: + result = PairingModel(**dict()) + + return result + + @staticmethod + async def export_pairing_list_services(pairing_list: List): + """ + 导出角色-机器人-映射信息service + + :param pairing_list: 角色-机器人-映射信息列表 + :return: 角色-机器人-映射信息对应excel的二进制数据 + """ + # 创建一个映射字典，将英文键映射到中文键 + mapping_dict = { + 'pairingId': '主键ID', + 'robotRoleId': '机器人角色ID', + 'robotId': '机器人ID', + 'status': '使用情况(0停用,1使用)', + 'updateTime': '更新时间', + 'updateBy': '修改者', + } + binary_data = ExcelUtil.export_list2excel(pairing_list, mapping_dict) + + return binary_data