399 lines
16 KiB
Python
399 lines
16 KiB
Python
from django.shortcuts import render, get_object_or_404
|
||
from rest_framework import viewsets, status
|
||
from rest_framework.decorators import action
|
||
from rest_framework.response import Response
|
||
from rest_framework.permissions import IsAuthenticated
|
||
from apps.user.authentication import CustomTokenAuthentication
|
||
import logging
|
||
|
||
from .models import Brand, Product, Campaign, BrandChatSession
|
||
from .serializers import (
|
||
BrandSerializer,
|
||
ProductSerializer,
|
||
CampaignSerializer,
|
||
BrandChatSessionSerializer,
|
||
BrandDetailSerializer
|
||
)
|
||
from .services.status_polling_service import polling_service
|
||
from .services.offer_status_service import OfferStatusService
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
def api_response(code=200, message="成功", data=None):
|
||
"""统一API响应格式"""
|
||
return Response({
|
||
'code': code,
|
||
'message': message,
|
||
'data': data
|
||
})
|
||
|
||
class BrandViewSet(viewsets.ModelViewSet):
|
||
|
||
"""品牌API视图集"""
|
||
queryset = Brand.objects.all()
|
||
serializer_class = BrandSerializer
|
||
authentication_classes = [CustomTokenAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def get_serializer_class(self):
|
||
if self.action == 'retrieve':
|
||
return BrandDetailSerializer
|
||
return BrandSerializer
|
||
|
||
def list(self, request, *args, **kwargs):
|
||
queryset = self.filter_queryset(self.get_queryset())
|
||
serializer = self.get_serializer(queryset, many=True)
|
||
return api_response(data=serializer.data)
|
||
|
||
def retrieve(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance)
|
||
return api_response(data=serializer.data)
|
||
|
||
def update(self, request, *args, **kwargs):
|
||
partial = kwargs.pop('partial', False)
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance, data=request.data, partial=partial)
|
||
if serializer.is_valid():
|
||
self.perform_update(serializer)
|
||
return api_response(data=serializer.data)
|
||
return api_response(code=400, message="更新失败", data=serializer.errors)
|
||
|
||
def destroy(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
self.perform_destroy(instance)
|
||
return api_response(message="删除成功", data=None)
|
||
|
||
@action(detail=True, methods=['get'])
|
||
def products(self, request, pk=None):
|
||
"""获取品牌下的所有产品"""
|
||
brand = self.get_object()
|
||
products = Product.objects.filter(brand=brand, is_active=True)
|
||
serializer = ProductSerializer(products, many=True)
|
||
return api_response(data=serializer.data)
|
||
|
||
@action(detail=True, methods=['get'])
|
||
def campaigns(self, request, pk=None):
|
||
"""获取品牌下的所有活动"""
|
||
brand = self.get_object()
|
||
campaigns = Campaign.objects.filter(brand=brand, is_active=True)
|
||
serializer = CampaignSerializer(campaigns, many=True)
|
||
return api_response(data=serializer.data)
|
||
|
||
@action(detail=True, methods=['get'])
|
||
def dataset_ids(self, request, pk=None):
|
||
"""获取品牌的所有知识库ID"""
|
||
brand = self.get_object()
|
||
return api_response(data={'dataset_id_list': brand.dataset_id_list})
|
||
|
||
|
||
class ProductViewSet(viewsets.ModelViewSet):
|
||
"""产品API视图集"""
|
||
queryset = Product.objects.filter(is_active=True)
|
||
serializer_class = ProductSerializer
|
||
authentication_classes = [CustomTokenAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def list(self, request, *args, **kwargs):
|
||
queryset = self.filter_queryset(self.get_queryset())
|
||
serializer = self.get_serializer(queryset, many=True)
|
||
return api_response(data=serializer.data)
|
||
|
||
def retrieve(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance)
|
||
return api_response(data=serializer.data)
|
||
|
||
def update(self, request, *args, **kwargs):
|
||
partial = kwargs.pop('partial', False)
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance, data=request.data, partial=partial)
|
||
if serializer.is_valid():
|
||
self.perform_update(serializer)
|
||
return api_response(data=serializer.data)
|
||
return api_response(code=400, message="更新失败", data=serializer.errors)
|
||
|
||
def destroy(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
self.perform_destroy(instance)
|
||
return api_response(message="删除成功", data=None)
|
||
|
||
def perform_create(self, serializer):
|
||
# 创建产品时自动更新品牌的dataset_id_list
|
||
product = serializer.save()
|
||
brand = product.brand
|
||
|
||
# 确保dataset_id添加到品牌的dataset_id_list中
|
||
if product.dataset_id and product.dataset_id not in brand.dataset_id_list:
|
||
brand.dataset_id_list.append(product.dataset_id)
|
||
brand.save(update_fields=['dataset_id_list', 'updated_at'])
|
||
|
||
def perform_update(self, serializer):
|
||
# 获取原始产品信息
|
||
old_product = self.get_object()
|
||
old_dataset_id = old_product.dataset_id
|
||
|
||
# 保存更新后的产品
|
||
product = serializer.save()
|
||
brand = product.brand
|
||
|
||
# 从品牌的dataset_id_list中移除旧的dataset_id，添加新的dataset_id
|
||
if old_dataset_id in brand.dataset_id_list:
|
||
brand.dataset_id_list.remove(old_dataset_id)
|
||
|
||
if product.dataset_id and product.dataset_id not in brand.dataset_id_list:
|
||
brand.dataset_id_list.append(product.dataset_id)
|
||
|
||
brand.save(update_fields=['dataset_id_list', 'updated_at'])
|
||
|
||
def perform_destroy(self, instance):
|
||
# 软删除产品，并从品牌的dataset_id_list中移除对应的ID
|
||
instance.is_active = False
|
||
instance.save()
|
||
|
||
brand = instance.brand
|
||
if instance.dataset_id in brand.dataset_id_list:
|
||
brand.dataset_id_list.remove(instance.dataset_id)
|
||
brand.save(update_fields=['dataset_id_list', 'updated_at'])
|
||
|
||
|
||
class CampaignViewSet(viewsets.ModelViewSet):
|
||
"""活动API视图集"""
|
||
queryset = Campaign.objects.filter(is_active=True)
|
||
serializer_class = CampaignSerializer
|
||
authentication_classes = [CustomTokenAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def list(self, request, *args, **kwargs):
|
||
queryset = self.filter_queryset(self.get_queryset())
|
||
serializer = self.get_serializer(queryset, many=True)
|
||
return api_response(data=serializer.data)
|
||
|
||
def retrieve(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance)
|
||
return api_response(data=serializer.data)
|
||
|
||
def update(self, request, *args, **kwargs):
|
||
partial = kwargs.pop('partial', False)
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance, data=request.data, partial=partial)
|
||
if serializer.is_valid():
|
||
self.perform_update(serializer)
|
||
return api_response(data=serializer.data)
|
||
return api_response(code=400, message="更新失败", data=serializer.errors)
|
||
|
||
def destroy(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
self.perform_destroy(instance)
|
||
return api_response(message="删除成功", data=None)
|
||
|
||
def perform_create(self, serializer):
|
||
# 创建活动时自动更新品牌的dataset_id_list
|
||
campaign = serializer.save()
|
||
brand = campaign.brand
|
||
|
||
# 确保dataset_id添加到品牌的dataset_id_list中
|
||
if campaign.dataset_id and campaign.dataset_id not in brand.dataset_id_list:
|
||
brand.dataset_id_list.append(campaign.dataset_id)
|
||
brand.save(update_fields=['dataset_id_list', 'updated_at'])
|
||
|
||
def perform_update(self, serializer):
|
||
# 获取原始活动信息
|
||
old_campaign = self.get_object()
|
||
old_dataset_id = old_campaign.dataset_id
|
||
|
||
# 保存更新后的活动
|
||
campaign = serializer.save()
|
||
brand = campaign.brand
|
||
|
||
# 从品牌的dataset_id_list中移除旧的dataset_id，添加新的dataset_id
|
||
if old_dataset_id in brand.dataset_id_list:
|
||
brand.dataset_id_list.remove(old_dataset_id)
|
||
|
||
if campaign.dataset_id and campaign.dataset_id not in brand.dataset_id_list:
|
||
brand.dataset_id_list.append(campaign.dataset_id)
|
||
|
||
brand.save(update_fields=['dataset_id_list', 'updated_at'])
|
||
|
||
def perform_destroy(self, instance):
|
||
# 软删除活动，并从品牌的dataset_id_list中移除对应的ID
|
||
instance.is_active = False
|
||
instance.save()
|
||
|
||
brand = instance.brand
|
||
if instance.dataset_id in brand.dataset_id_list:
|
||
brand.dataset_id_list.remove(instance.dataset_id)
|
||
brand.save(update_fields=['dataset_id_list', 'updated_at'])
|
||
|
||
@action(detail=True, methods=['post'])
|
||
def add_product(self, request, pk=None):
|
||
"""将产品添加到活动中"""
|
||
campaign = self.get_object()
|
||
product_id = request.data.get('product_id')
|
||
|
||
if not product_id:
|
||
return api_response(code=400, message="缺少产品ID", data=None)
|
||
|
||
try:
|
||
product = Product.objects.get(id=product_id, is_active=True)
|
||
campaign.link_product.add(product)
|
||
return api_response(message="产品添加成功", data=None)
|
||
except Product.DoesNotExist:
|
||
return api_response(code=404, message="产品不存在", data=None)
|
||
except Exception as e:
|
||
return api_response(code=500, message=f"添加产品失败: {str(e)}", data=None)
|
||
|
||
@action(detail=True, methods=['post'])
|
||
def remove_product(self, request, pk=None):
|
||
"""从活动中移除产品"""
|
||
campaign = self.get_object()
|
||
product_id = request.data.get('product_id')
|
||
|
||
if not product_id:
|
||
return api_response(code=400, message="缺少产品ID", data=None)
|
||
|
||
try:
|
||
product = Product.objects.get(id=product_id)
|
||
campaign.link_product.remove(product)
|
||
return api_response(message="产品移除成功", data=None)
|
||
except Product.DoesNotExist:
|
||
return api_response(code=404, message="产品不存在", data=None)
|
||
except Exception as e:
|
||
return api_response(code=500, message=f"移除产品失败: {str(e)}", data=None)
|
||
|
||
@action(detail=True, methods=['get'])
|
||
def creator_list(self, request, pk=None):
|
||
"""获取活动关联的达人列表"""
|
||
campaign = self.get_object()
|
||
from apps.daren_detail.models import CreatorCampaign, CreatorProfile
|
||
|
||
# 获取所有达人数据
|
||
creator_list = OfferStatusService.get_campaign_creator_data(campaign.id)
|
||
|
||
# 启动状态轮询（当有用户请求此接口时）
|
||
try:
|
||
# 获取产品ID
|
||
product_id = None
|
||
if campaign.link_product.exists():
|
||
product = campaign.link_product.first()
|
||
product_id = product.id
|
||
|
||
# 如果没有关联产品，则使用活动ID作为产品ID
|
||
if not product_id:
|
||
product_id = campaign.id
|
||
|
||
# 构建达人-产品对
|
||
creator_product_pairs = []
|
||
for creator_data in creator_list:
|
||
creator_id = creator_data['id']
|
||
creator_product_pairs.append((creator_id, product_id))
|
||
|
||
# 启动轮询
|
||
if creator_product_pairs:
|
||
polling_service.start_polling(
|
||
campaign_id=campaign.id,
|
||
creator_product_pairs=creator_product_pairs,
|
||
interval=30 # 每30秒轮询一次
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"启动状态轮询时出错: {str(e)}")
|
||
|
||
return api_response(data=creator_list)
|
||
|
||
@action(detail=True, methods=['post'])
|
||
def update_creator_status(self, request, pk=None):
|
||
"""手动更新达人状态"""
|
||
campaign = self.get_object()
|
||
from apps.daren_detail.models import CreatorCampaign
|
||
from .services.offer_status_service import OfferStatusService
|
||
|
||
# 获取传入的达人ID
|
||
creator_id = request.data.get('creator_id')
|
||
|
||
if not creator_id:
|
||
return api_response(code=400, message="缺少必要参数: creator_id", data=None)
|
||
|
||
try:
|
||
# 查询达人与活动的关联
|
||
creator_campaign = CreatorCampaign.objects.get(
|
||
campaign_id=campaign.id,
|
||
creator_id=creator_id
|
||
)
|
||
|
||
# 获取产品ID
|
||
product_id = None
|
||
if campaign.link_product.exists():
|
||
product = campaign.link_product.first()
|
||
product_id = product.id
|
||
|
||
# 如果没有关联产品，则使用活动ID作为产品ID
|
||
if not product_id:
|
||
product_id = campaign.id
|
||
|
||
# 获取最新状态
|
||
status = OfferStatusService.fetch_status(creator_id, product_id)
|
||
|
||
if status:
|
||
# 更新状态
|
||
creator_campaign.status = status
|
||
creator_campaign.save()
|
||
|
||
# 获取所有达人的最新数据
|
||
creator_list = OfferStatusService.get_campaign_creator_data(campaign.id)
|
||
|
||
# 发送WebSocket更新
|
||
OfferStatusService.send_status_update(campaign.id, creator_id, status)
|
||
|
||
return api_response(message="状态已更新", data=creator_list)
|
||
else:
|
||
return api_response(code=500, message="获取状态失败", data=None)
|
||
|
||
except CreatorCampaign.DoesNotExist:
|
||
return api_response(code=404, message="找不到达人与活动的关联", data=None)
|
||
except Exception as e:
|
||
logger.error(f"更新达人状态时出错: {str(e)}")
|
||
return api_response(code=500, message=f"更新状态失败: {str(e)}", data=None)
|
||
|
||
|
||
class BrandChatSessionViewSet(viewsets.ModelViewSet):
|
||
"""品牌聊天会话API视图集"""
|
||
queryset = BrandChatSession.objects.filter(is_active=True)
|
||
serializer_class = BrandChatSessionSerializer
|
||
authentication_classes = [CustomTokenAuthentication]
|
||
permission_classes = [IsAuthenticated]
|
||
|
||
def list(self, request, *args, **kwargs):
|
||
queryset = self.filter_queryset(self.get_queryset())
|
||
serializer = self.get_serializer(queryset, many=True)
|
||
return api_response(data=serializer.data)
|
||
|
||
def retrieve(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance)
|
||
return api_response(data=serializer.data)
|
||
|
||
def update(self, request, *args, **kwargs):
|
||
partial = kwargs.pop('partial', False)
|
||
instance = self.get_object()
|
||
serializer = self.get_serializer(instance, data=request.data, partial=partial)
|
||
if serializer.is_valid():
|
||
self.perform_update(serializer)
|
||
return api_response(data=serializer.data)
|
||
return api_response(code=400, message="更新失败", data=serializer.errors)
|
||
|
||
def destroy(self, request, *args, **kwargs):
|
||
instance = self.get_object()
|
||
self.perform_destroy(instance)
|
||
return api_response(message="删除成功", data=None)
|
||
|
||
def perform_create(self, serializer):
|
||
# 创建聊天会话时，可以设置使用特定品牌下的所有知识库
|
||
chat_session = serializer.save()
|
||
|
||
# 如果没有提供dataset_id_list，则使用品牌的dataset_id_list
|
||
if not chat_session.dataset_id_list:
|
||
brand = chat_session.brand
|
||
chat_session.dataset_id_list = brand.dataset_id_list
|
||
chat_session.save(update_fields=['dataset_id_list', 'updated_at'])
|