1.添加机器人角色和机器人对应表
This commit is contained in:
parent
4093c78a53
commit
0d6909760f
@ -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))
|
||||
120
ruoyi-fastapi-backend/module_admin/dao/robot_role_pairing_dao.py
Normal file
120
ruoyi-fastapi-backend/module_admin/dao/robot_role_pairing_dao.py
Normal file
@ -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])))
|
||||
|
||||
@ -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='修改者')
|
||||
|
||||
|
||||
|
||||
@ -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')
|
||||
@ -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
|
||||
Loading…
Reference in New Issue
Block a user