추가: 마이페이지 viewset 추가

apps/account/v1/apis/mypage_api.py

@@ -0,0 +1,132 @@
+from django.core.cache import cache
+from rest_framework import permissions, renderers, status, viewsets
+from rest_framework.decorators import action
+from rest_framework.response import Response
+
+from alaltalk import cache_key
+from apps.account.models import CustomUser, UserProfileImage
+from apps.account.services.user_service import UserService
+from apps.account.v1.serializers.user_serializer import (
+    UserReadSerializer,
+    UserUpdateSerializer,
+)
+from apps.friend.services.friend_selector import FriendSelector
+from apps.friend.services.friend_service import FriendService
+
+
+class MyPageViewSet(viewsets.ModelViewSet):
+
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_queryset(self):
+        return (
+            CustomUser.objects.filter(is_deleted=False)
+            .filter(id=self.request.user.id)
+            .prefetch_related("youtube", "news", "book", "shopping")
+        )
+
+    def get_renderers(self):
+        if self.action == "list":
+            renderer_classes = [renderers.TemplateHTMLRenderer]
+        else:
+            renderer_classes = [renderers.JSONRenderer]
+        return [renderer() for renderer in renderer_classes]
+
+    def get_serializer_class(self):
+
+        if self.action == "list":
+            serializer_class = UserReadSerializer
+        elif self.action in ["update", "partial_update"]:
+            serializer_class = UserUpdateSerializer
+        return serializer_class
+
+    def list(self, request, *args, **kwargs):
+        """유저의 마이페이지를 렌더링 합니다."""
+
+        user = request.user
+        friend_requests = FriendSelector.get_recieved_requests(user_id=user.id)
+        youtubes = cache.get_or_set(
+            cache_key.USER_YOUTUBE.format(user_id=user.id),
+            self.get_queryset().youtube.all(),
+            timeout=86400,
+        )
+        news = cache.get_or_set(
+            cache_key.USER_NEWS.format(user_id=user.id),
+            self.get_queryset().news.all(),
+            timeout=86400,
+        )
+        books = cache.get_or_set(
+            cache_key.USER_BOOK.format(user_id=user.id),
+            self.get_queryset().book.all(),
+            timeout=86400,
+        )
+        shoppings = cache.get_or_set(
+            cache_key.USER_SHOPPING.format(user_id=user.id),
+            self.get_queryset().shopping.all(),
+            timeout=86400,
+        )
+
+        context = {
+            "user": user,
+            "friend_requests": friend_requests,
+            "youtubes": youtubes,
+            "news": news,
+            "books": books,
+            "shoppings": shoppings,
+        }
+
+        return Response(
+            context, status=status.HTTP_200_OK, template_name="account/mypage.html"
+        )
+
+    def update(self, request, *args, **kwargs):
+        """회원 정보를 업데이트 합니다."""
+
+        user = request.user
+
+        request_data = {
+            "nickname": request.data.get("nickname"),
+            "password": request.data.get("password"),
+            "bio": request.data.get("bio"),
+            "profile_image": request.data.get("img"),
+        }
+
+        serializer = self.get_serializer(data=request_data)
+
+        if serializer.is_valid(raise_exception=True):
+            validated_data = serializer.validated_data
+
+            user.email = validated_data["email"]
+            user.nickname = validated_data["nickname"]
+            user.bio = validated_data["bio"]
+
+            if validated_data.get("profile_image"):
+                img = validated_data["profile_image"]
+                UserProfileImage.objects.create(user_id=user.id, img=img)
+
+            if validated_data.get("password"):
+                password = validated_data["password"]
+                user.set_password(password)
+
+            user.save()
+
+            data = {"msg": "ok", "user": UserReadSerializer(instance=user).data}
+            return Response(data=data, status=status.HTTP_200_OK)
+        return Response(
+            data=serializer.error_messages, status=status.HTTP_400_BAD_REQUEST
+        )
+
+    def destroy(self, request, *args, **kwargs):
+        """회원 탈퇴"""
+        user = UserService.delete_user_account(user_id=request.user.id)
+        data = {"msg": "deleted", "data": UserReadSerializer(user).data}
+        return Response(data=data, status=status.HTTP_200_OK)
+
+    @action(detail=False, methods=["post"])
+    def auth_check(self, request, *args, **kwargs):
+        user = request.user
+        password = request.data.get("password")
+
+        if user.check_password(password):
+            return Response(data={"msg": "ok"}, status=status.HTTP_200_OK)
+        return Response(data={"msg": "no"}, status=status.HTTP_400_BAD_REQUEST)

apps/account/v1/apis/user_api.py

@@ -44,7 +44,7 @@ def post(self, request, *args, **kwargs):
             "nickname": request.data.get("nickname"),
             "password": request.data.get("password"),
             "bio": request.data.get("bio"),
-            "img": request.data.get("img", None),
+            "profile_image": request.data.get("img", None),
         }
 
         is_present = UserSelector.check_email_duplication(email=request_data["email"])
@@ -64,7 +64,7 @@ def post(self, request, *args, **kwargs):
                 nickname=validated_data["nickname"],
                 bio=validated_data["bio"],
                 password=validated_data["password"],
-                img=validated_data.get("img", None),
+                img=validated_data.get("profile_image", None),
             )
 
             data = {"msg": "sent", "user": UserReadSerializer(instance=user).data}

apps/account/v1/serializers/user_serializer.py

@@ -17,6 +17,18 @@ class Meta:
         fields = ("email", "password", "nickname", "bio", "profile_image")
 
 
+class UserUpdateSerializer(serializers.ModelSerializer):
+
+    nickname = serializers.CharField(required=False)
+    password = serializers.CharField(required=False, write_only=True)
+    profile_image = serializers.ImageField(required=False)
+    bio = serializers.CharField(required=False)
+
+    class Meta:
+        model = CustomUser
+        fields = ("nickname", "password", "profile_image", "bio")
+
+
 class UserReadSerializer(serializers.ModelSerializer):
 
     profile_image = serializers.SerializerMethodField()
@@ -34,6 +46,7 @@ class Meta:
             "is_active",
             "is_recommend_on",
             "is_like_public",
+            "is_deleted",
         )
 
     def get_profile_image(self, obj):