237 lines
7.6 KiB
Python
237 lines
7.6 KiB
Python
|
from rest_framework import viewsets, status, permissions
|
||
|
|
from rest_framework.decorators import action
|
||
|
|
from rest_framework.response import Response
|
||
|
|
from django.utils import timezone
|
||
|
|
from .models import Meeting, Participant, Message, MeetingMinutes
|
||
|
|
from .serializers import (
|
||
|
|
MeetingSerializer, ParticipantSerializer,
|
||
|
|
MessageSerializer, InboxSerializer
|
||
|
|
)
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
from django.shortcuts import get_object_or_404
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
User = get_user_model()
|
||
|
|
|
||
|
|
|
||
|
|
class MeetingViewSet(viewsets.ModelViewSet):
|
||
|
|
"""会议室视图集"""
|
||
|
|
queryset = Meeting.objects.all()
|
||
|
|
serializer_class = MeetingSerializer
|
||
|
|
permission_classes = [] # 临时开放所有权限
|
||
|
|
|
||
|
|
def get_queryset(self):
|
||
|
|
# 简单返回所有会议
|
||
|
|
return Meeting.objects.all().order_by('-created_at')
|
||
|
|
|
||
|
|
def create(self, request, *args, **kwargs):
|
||
|
|
"""创建会议"""
|
||
|
|
serializer = self.get_serializer(data=request.data)
|
||
|
|
serializer.is_valid(raise_exception=True)
|
||
|
|
|
||
|
|
# 临时：使用第一个用户作为 host
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
User = get_user_model()
|
||
|
|
host = User.objects.first()
|
||
|
|
|
||
|
|
meeting = serializer.save(host=host)
|
||
|
|
|
||
|
|
# 创建主持人参会记录
|
||
|
|
Participant.objects.create(
|
||
|
|
meeting=meeting,
|
||
|
|
user=host,
|
||
|
|
agent_type='human',
|
||
|
|
nickname=host.username,
|
||
|
|
is_host=True
|
||
|
|
)
|
||
|
|
|
||
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
||
|
|
|
||
|
|
@action(detail=True, methods=['post'])
|
||
|
|
def start(self, request, pk=None):
|
||
|
|
"""开始会议"""
|
||
|
|
meeting = self.get_object()
|
||
|
|
if meeting.host != request.user:
|
||
|
|
return Response(
|
||
|
|
{'error': '只有主持人可以开始会议'},
|
||
|
|
status=status.HTTP_403_FORBIDDEN
|
||
|
|
)
|
||
|
|
|
||
|
|
meeting.status = 'active'
|
||
|
|
meeting.started_at = timezone.now()
|
||
|
|
meeting.save()
|
||
|
|
|
||
|
|
return Response({'status': '会议已开始'})
|
||
|
|
|
||
|
|
@action(detail=True, methods=['post'])
|
||
|
|
def end(self, request, pk=None):
|
||
|
|
"""结束会议"""
|
||
|
|
meeting = self.get_object()
|
||
|
|
if meeting.host != request.user:
|
||
|
|
return Response(
|
||
|
|
{'error': '只有主持人可以结束会议'},
|
||
|
|
status=status.HTTP_403_FORBIDDEN
|
||
|
|
)
|
||
|
|
|
||
|
|
meeting.status = 'ended'
|
||
|
|
meeting.ended_at = timezone.now()
|
||
|
|
meeting.save()
|
||
|
|
|
||
|
|
return Response({'status': '会议已结束'})
|
||
|
|
|
||
|
|
@action(detail=True, methods=['post'])
|
||
|
|
def join(self, request, pk=None):
|
||
|
|
"""加入会议"""
|
||
|
|
meeting = self.get_object()
|
||
|
|
|
||
|
|
if meeting.status == 'ended':
|
||
|
|
return Response(
|
||
|
|
{'error': '会议已结束'},
|
||
|
|
status=status.HTTP_400_BAD_REQUEST
|
||
|
|
)
|
||
|
|
|
||
|
|
invite_code = request.data.get('invite_code')
|
||
|
|
if invite_code != meeting.invite_code:
|
||
|
|
return Response(
|
||
|
|
{'error': '邀请码错误'},
|
||
|
|
status=status.HTTP_400_BAD_REQUEST
|
||
|
|
)
|
||
|
|
|
||
|
|
# 检查是否已加入
|
||
|
|
existing = Participant.objects.filter(
|
||
|
|
meeting=meeting,
|
||
|
|
user=request.user,
|
||
|
|
left_at__isnull=True
|
||
|
|
).first()
|
||
|
|
|
||
|
|
if existing:
|
||
|
|
return Response(ParticipantSerializer(existing).data)
|
||
|
|
|
||
|
|
# 创建参会记录
|
||
|
|
participant = Participant.objects.create(
|
||
|
|
meeting=meeting,
|
||
|
|
user=request.user,
|
||
|
|
agent_type='human',
|
||
|
|
nickname=request.user.username
|
||
|
|
)
|
||
|
|
|
||
|
|
return Response(ParticipantSerializer(participant).data, status=status.HTTP_201_CREATED)
|
||
|
|
|
||
|
|
@action(detail=True, methods=['get'])
|
||
|
|
def participants(self, request, pk=None):
|
||
|
|
"""获取参会者列表"""
|
||
|
|
meeting = self.get_object()
|
||
|
|
participants = meeting.participants.filter(left_at__isnull=True)
|
||
|
|
serializer = ParticipantSerializer(participants, many=True)
|
||
|
|
return Response(serializer.data)
|
||
|
|
|
||
|
|
@action(detail=True, methods=['get'])
|
||
|
|
def messages(self, request, pk=None):
|
||
|
|
"""获取消息（人类轮询）"""
|
||
|
|
meeting = self.get_object()
|
||
|
|
last_id = request.query_params.get('last_id', 0)
|
||
|
|
|
||
|
|
messages = meeting.messages.filter(id__gt=last_id).select_related('sender')
|
||
|
|
serializer = MessageSerializer(messages, many=True)
|
||
|
|
|
||
|
|
return Response({'messages': serializer.data})
|
||
|
|
|
||
|
|
@action(detail=True, methods=['post'])
|
||
|
|
def send_message(self, request, pk=None):
|
||
|
|
"""发送消息"""
|
||
|
|
meeting = self.get_object()
|
||
|
|
|
||
|
|
# 获取或创建参会者（临时：使用第一个参会者或创建）
|
||
|
|
participant = Participant.objects.filter(
|
||
|
|
meeting=meeting,
|
||
|
|
left_at__isnull=True
|
||
|
|
).first()
|
||
|
|
|
||
|
|
if not participant:
|
||
|
|
# 创建默认参会者
|
||
|
|
host = meeting.host
|
||
|
|
participant = Participant.objects.create(
|
||
|
|
meeting=meeting,
|
||
|
|
user=host,
|
||
|
|
agent_type='human',
|
||
|
|
nickname=host.username
|
||
|
|
)
|
||
|
|
|
||
|
|
content = request.data.get('content')
|
||
|
|
if not content:
|
||
|
|
return Response(
|
||
|
|
{'error': '消息内容不能为空'},
|
||
|
|
status=status.HTTP_400_BAD_REQUEST
|
||
|
|
)
|
||
|
|
|
||
|
|
message = Message.objects.create(
|
||
|
|
meeting=meeting,
|
||
|
|
sender=participant,
|
||
|
|
content=content,
|
||
|
|
is_broadcast=request.data.get('is_broadcast', True),
|
||
|
|
requires_response=request.data.get('requires_response', False)
|
||
|
|
)
|
||
|
|
|
||
|
|
return Response(MessageSerializer(message).data, status=status.HTTP_201_CREATED)
|
||
|
|
|
||
|
|
@action(detail=True, methods=['get'])
|
||
|
|
def inbox(self, request, pk=None):
|
||
|
|
"""Agent 查阅信箱"""
|
||
|
|
meeting = self.get_object()
|
||
|
|
agent_id = request.query_params.get('agent_id')
|
||
|
|
|
||
|
|
if not agent_id:
|
||
|
|
return Response(
|
||
|
|
{'error': '缺少 agent_id 参数'},
|
||
|
|
status=status.HTTP_400_BAD_REQUEST
|
||
|
|
)
|
||
|
|
|
||
|
|
# 找到这个 Agent 的参会记录
|
||
|
|
participant = get_object_or_404(
|
||
|
|
Participant,
|
||
|
|
meeting=meeting,
|
||
|
|
agent_id=agent_id,
|
||
|
|
left_at__isnull=True
|
||
|
|
)
|
||
|
|
|
||
|
|
# 获取发给这个 Agent 的消息（未读）
|
||
|
|
messages = Message.objects.filter(
|
||
|
|
meeting=meeting
|
||
|
|
).exclude(
|
||
|
|
read_by=participant
|
||
|
|
)
|
||
|
|
|
||
|
|
# 如果是群发消息，所有人都能看到
|
||
|
|
# 如果是指定消息，需要检查 recipients
|
||
|
|
# 简化版：所有未读消息都返回
|
||
|
|
|
||
|
|
serializer = MessageSerializer(messages, many=True)
|
||
|
|
|
||
|
|
return Response({
|
||
|
|
'unread_count': messages.count(),
|
||
|
|
'messages': serializer.data
|
||
|
|
})
|
||
|
|
|
||
|
|
|
||
|
|
class ParticipantViewSet(viewsets.ModelViewSet):
|
||
|
|
"""参会者视图集"""
|
||
|
|
queryset = Participant.objects.all()
|
||
|
|
serializer_class = ParticipantSerializer
|
||
|
|
permission_classes = [permissions.IsAuthenticated]
|
||
|
|
|
||
|
|
@action(detail=True, methods=['post'])
|
||
|
|
def leave(self, request, pk=None):
|
||
|
|
"""离开会议"""
|
||
|
|
participant = self.get_object()
|
||
|
|
|
||
|
|
if participant.user != request.user:
|
||
|
|
return Response(
|
||
|
|
{'error': '无权操作'},
|
||
|
|
status=status.HTTP_403_FORBIDDEN
|
||
|
|
)
|
||
|
|
|
||
|
|
participant.left_at = timezone.now()
|
||
|
|
participant.save()
|
||
|
|
|
||
|
|
return Response({'status': '已离开会议'})
|