86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
"""
|
||
智能体模块视图 - App端
|
||
"""
|
||
from rest_framework import viewsets, status
|
||
from rest_framework.permissions import IsAuthenticated
|
||
|
||
from utils.response import success, error
|
||
from utils.exceptions import ErrorCode
|
||
from apps.admins.authentication import AppJWTAuthentication
|
||
from .models import Spirit
|
||
from .serializers import (
|
||
SpiritSerializer,
|
||
SpiritListSerializer,
|
||
CreateSpiritSerializer,
|
||
)
|
||
|
||
|
||
class SpiritViewSet(viewsets.ModelViewSet):
|
||
"""智能体视图集（App端）"""
|
||
|
||
authentication_classes = [AppJWTAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def get_queryset(self):
|
||
return Spirit.objects.filter(user=self.request.user)
|
||
|
||
def get_serializer_class(self):
|
||
if self.action == 'list':
|
||
return SpiritListSerializer
|
||
if self.action == 'create':
|
||
return CreateSpiritSerializer
|
||
return SpiritSerializer
|
||
|
||
def list(self, request, *args, **kwargs):
|
||
"""
|
||
获取我的智能体列表
|
||
GET /api/v1/spirits
|
||
"""
|
||
queryset = self.get_queryset()
|
||
serializer = SpiritListSerializer(queryset, many=True)
|
||
return success(data=serializer.data)
|
||
|
||
def create(self, request, *args, **kwargs):
|
||
"""
|
||
创建智能体
|
||
POST /api/v1/spirits
|
||
"""
|
||
serializer = CreateSpiritSerializer(data=request.data)
|
||
if serializer.is_valid():
|
||
spirit = serializer.save(user=request.user)
|
||
return success(data=SpiritSerializer(spirit).data, message='创建成功')
|
||
return error(message=str(serializer.errors))
|
||
|
||
def retrieve(self, request, *args, **kwargs):
|
||
"""
|
||
获取智能体详情
|
||
GET /api/v1/spirits/{id}
|
||
"""
|
||
try:
|
||
instance = self.get_object()
|
||
serializer = SpiritSerializer(instance)
|
||
return success(data=serializer.data)
|
||
except Spirit.DoesNotExist:
|
||
return error(code=ErrorCode.SPIRIT_NOT_FOUND, message='智能体不存在')
|
||
|
||
def update(self, request, *args, **kwargs):
|
||
"""
|
||
更新智能体
|
||
PUT /api/v1/spirits/{id}
|
||
"""
|
||
instance = self.get_object()
|
||
serializer = SpiritSerializer(instance, data=request.data, partial=True)
|
||
if serializer.is_valid():
|
||
serializer.save()
|
||
return success(data=serializer.data, message='更新成功')
|
||
return error(message=str(serializer.errors))
|
||
|
||
def destroy(self, request, *args, **kwargs):
|
||
"""
|
||
删除智能体
|
||
DELETE /api/v1/spirits/{id}
|
||
"""
|
||
instance = self.get_object()
|
||
instance.delete()
|
||
return success(message='删除成功')
|