JYS/userprofile/views.py

from rest_framework import viewsets
from utils.page import MyPageNumberPagination
from rest_framework.filters import OrderingFilter
from django_filters.rest_framework import DjangoFilterBackend
from rest_framework.response import Response
from rest_framework import status
from rest_framework.parsers import MultiPartParser, FormParser
from rest_framework.views import APIView
import os
import uuid
from django.conf import settings

from .models import Users, AcademicProfile, ResearchGroup, GroupMembership
from .filter import UsersFilter, AcademicProfileFilter, ResearchGroupFilter, GroupMembershipFilter
from .serializers import UsersGetSerializer,UsersPostSerializer
from .serializers import AcademicProfileGetSerializer, AcademicProfilePostSerializer

class UserprofileViewSet(viewsets.ModelViewSet):
    """
    用户档案视图集
        retrieve:
            Response a data list (get)

        list:
            Response a data list (all)

        create:
            Create a data line (post)

        delete:
            Delete a data line (delete)

        partial_update:
            Partial_update a data (patch:partial_update)

        update:
            Update a data (put:update)
    """
    fliter_backends = (DjangoFilterBackend, OrderingFilter)
    ordering_fields = ('create_time', 'update_time')
    filterset_class = UsersFilter
    pagination_class = MyPageNumberPagination

    def get_project(self):
        try:
            # 表格通道
            if 'pk' in self.kwargs:
                id = self.kwargs.get('pk')
                return id
            else:
            # 用户通道
                user = self.request.auth
                if user.roles == 'USR':
                    return user.user_id
                else:
                    return None
        except:
            return None


    def get_queryset(self):
        project_id = self.get_project()
        if project_id:
            return Users.objects.filter(user_id=project_id)
        else:
            return Users.objects.all()

    def get_serializer_class(self):
        if self.action in ['list','retrieve']:
            return UsersGetSerializer
        elif self.action in ['create', 'update', 'partial_update']:
            return UsersPostSerializer
        else:
            return UsersGetSerializer
        
    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        serializer.save()

        headers = self.get_success_headers(serializer.data)
        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
    
    # 表格的写法
    def update(self, request, *args, **kwargs):
        if 'pk' in self.kwargs:
            qs = self.get_object()
            serializer = self.get_serializer(qs, data=request.data, partial=True)
            serializer.is_valid(raise_exception=True)
            serializer.save()
            return Response(serializer.data, status=status.HTTP_200_OK, headers=self.get_success_headers(serializer.data))
        else:
            # 没有qs，应该是管理员和用户更新自己的信息
            qs = request.auth
            serializer = self.get_serializer(qs, data=request.data, partial=True)
            serializer.is_valid(raise_exception=True)
            serializer.save()
            return Response(serializer.data, status=status.HTTP_200_OK, headers=self.get_success_headers(serializer.data))
        

    def partial_update(self, request, *args, **kwargs):
        qs = self.get_object()
        serializer = self.get_serializer(qs, data=request.data, partial=True)
        serializer.is_valid(raise_exception=True)
        serializer.save()

        return Response(serializer.data)
    
    def destroy(self, request, *args, **kwargs):
        qs = self.get_object()
        qs.is_delete = True
        qs.save()
        return Response(status=status.HTTP_204_NO_CONTENT)
    

    
class AvatarUploadView(APIView):
    parser_classes = [MultiPartParser]
    
    def post(self, request):
        avatar_file = request.FILES.get('avatar')
        if not avatar_file:
            return Response({"error": "未上传头像文件"}, status=400)
        
        # 生成唯一文件名
        ext = os.path.splitext(avatar_file.name)[1]
        filename = f"{uuid.uuid4().hex}{ext}"
        
        # 目标目录
        avatars_dir = os.path.join(settings.MEDIA_ROOT, 'avatars')
        
        # 确保目录存在
        if not os.path.exists(avatars_dir):
            os.makedirs(avatars_dir)
        
        # 保存文件
        save_path = os.path.join(avatars_dir, filename)
        with open(save_path, 'wb+') as destination:
            for chunk in avatar_file.chunks():
                destination.write(chunk)
        
        # 返回相对路径（前端会拼接完整URL）
        avatar_url = f"/avatars/{filename}"
        
        # 更新用户头像
        user = request.auth
        user.avatar = avatar_url
        user.save()
        
        # 传递 request 到序列化器上下文
        serializer = UsersGetSerializer(user, context={'request': request})
        return Response(serializer.data)

class AcademicProfileViewSet(viewsets.ModelViewSet):
    """
    学术资料视图集
        retrieve:
            Response a data list (get)

        list:
            Response a data list (all)

        create:
            Create a data line (post)

        delete:
            Delete a data line (delete)

        partial_update:
            Partial_update a data (patch:partial_update)

        update:
            Update a data (put:update)
    """
    fliter_backends = (DjangoFilterBackend, OrderingFilter)
    ordering_fields = ('create_time', 'update_time')
    filterset_class = AcademicProfileFilter
    pagination_class = MyPageNumberPagination

    def get_project(self):
        try:
            # 表格通道
            if 'pk' in self.kwargs:
                id = self.kwargs.get('pk')
                return id
            else:
            # 用户通道
                user = self.request.auth
                if user.roles == 'USR':
                    # id= AcademicProfile.objects.filter(user=user).first().id
                    return user.user_id

                else:
                    return None
        except:
            return None


    def get_queryset(self):
        project_id = self.get_project()
        if project_id:
            return AcademicProfile.objects.filter(user_id=project_id)
        else:
            return AcademicProfile.objects.all()

    def get_serializer_class(self):
        if self.action in ['list','retrieve']:
            return AcademicProfileGetSerializer 
        elif self.action in ['create', 'update', 'partial_update']:
            return AcademicProfilePostSerializer 
        else:
            return AcademicProfileGetSerializer 
        
    
    def update(self, request, *args, **kwargs):
        if 'pk' in self.kwargs:
            qs = self.get_object()
            serializer = self.get_serializer(qs, data=request.data, partial=True)
            serializer.is_valid(raise_exception=True)
            serializer.save()
            return Response(serializer.data, status=status.HTTP_200_OK, headers=self.get_success_headers(serializer.data))
        else:
            # 没有qs，应该是管理员和用户更新自己的信息
            qs = request.auth
            AcademicProfile_obj = AcademicProfile.objects.filter(user=qs).first()
            if not AcademicProfile_obj:
                AcademicProfile_obj = AcademicProfile.objects.create(user=qs)
            serializer = self.get_serializer(AcademicProfile_obj, data=request.data, partial=True)
            serializer.is_valid(raise_exception=True)
            serializer.save()
            return Response(serializer.data, status=status.HTTP_200_OK, headers=self.get_success_headers(serializer.data))
    
    def create(self, request, *args, **kwargs):
        return super().create(request, *args, **kwargs)

    def partial_update(self, request, *args, **kwargs):
        return super().partial_update(request, *args, **kwargs)
    
    def destroy(self, request, *args, **kwargs):
        return super().destroy(request, *args, **kwargs)