|
| 1 | +from __future__ import annotations |
| 2 | + |
| 3 | +import typing |
| 4 | + |
| 5 | +from admin_api.serializers.cms import PageAdminSerializer, SectionAdminSerializer, SitemapAdminSerializer |
| 6 | +from cms.models import Page, Section, Sitemap |
| 7 | +from core.const.tag import OpenAPITag |
| 8 | +from core.permissions import IsSuperUser |
| 9 | +from core.viewset.json_schema_viewset import JsonSchemaViewSet |
| 10 | +from django.db import transaction |
| 11 | +from drf_spectacular.utils import extend_schema, extend_schema_view |
| 12 | +from drf_standardized_errors.openapi_serializers import ( |
| 13 | + ValidationErrorEnum, |
| 14 | + ValidationErrorResponseSerializer, |
| 15 | + ValidationErrorSerializer, |
| 16 | +) |
| 17 | +from rest_framework import decorators, request, response, status, viewsets |
| 18 | + |
| 19 | +ADMIN_METHODS = ["list", "retrieve", "create", "update", "partial_update", "destroy"] |
| 20 | + |
| 21 | + |
| 22 | +class SectionData(typing.TypedDict): |
| 23 | + id: typing.NotRequired[str] |
| 24 | + page_id: typing.NotRequired[str] |
| 25 | + order: int |
| 26 | + body_ko: str |
| 27 | + body_en: str |
| 28 | + |
| 29 | + |
| 30 | +@extend_schema_view(**{m: extend_schema(tags=[OpenAPITag.ADMIN_CMS]) for m in ADMIN_METHODS}) |
| 31 | +class SitemapAdminViewSet(JsonSchemaViewSet, viewsets.ModelViewSet): |
| 32 | + http_method_names = ["get", "post", "patch", "delete"] |
| 33 | + serializer_class = SitemapAdminSerializer |
| 34 | + permission_classes = [IsSuperUser] |
| 35 | + queryset = Sitemap.objects.filter_active().select_related("created_by", "updated_by", "deleted_by") |
| 36 | + |
| 37 | + |
| 38 | +@extend_schema_view(**{m: extend_schema(tags=[OpenAPITag.ADMIN_CMS]) for m in ADMIN_METHODS}) |
| 39 | +class PageAdminViewSet(JsonSchemaViewSet, viewsets.ModelViewSet): |
| 40 | + serializer_class = PageAdminSerializer |
| 41 | + permission_classes = [IsSuperUser] |
| 42 | + queryset = Page.objects.filter_active().select_related("created_by", "updated_by", "deleted_by") |
| 43 | + |
| 44 | + @staticmethod |
| 45 | + def _response_section_validation_error(detail: str) -> response.Response: |
| 46 | + return response.Response( |
| 47 | + data=ValidationErrorResponseSerializer( |
| 48 | + instance={ |
| 49 | + "type": ValidationErrorEnum.VALIDATION_ERROR, |
| 50 | + "errors": ValidationErrorSerializer( |
| 51 | + instance=[ |
| 52 | + { |
| 53 | + "code": "section_validation_error", |
| 54 | + "detail": detail, |
| 55 | + "attr": "sections", |
| 56 | + }, |
| 57 | + ], |
| 58 | + many=True, |
| 59 | + ).data, |
| 60 | + }, |
| 61 | + ).data, |
| 62 | + status=status.HTTP_400_BAD_REQUEST, |
| 63 | + ) |
| 64 | + |
| 65 | + @extend_schema(tags=[OpenAPITag.ADMIN_CMS], responses={status.HTTP_200_OK: SectionAdminSerializer(many=True)}) |
| 66 | + @decorators.action(detail=True, methods=["get"], url_path="section") |
| 67 | + def list_sections(self, request: request.Request, *args: tuple, **kwargs: dict) -> response.Response: |
| 68 | + if not (page_id := kwargs.get("pk")): |
| 69 | + return self._response_section_validation_error("페이지 ID가 제공되지 않았습니다.") |
| 70 | + |
| 71 | + return response.Response( |
| 72 | + data=SectionAdminSerializer( |
| 73 | + instance=( |
| 74 | + Section.objects.filter_active() |
| 75 | + .filter(page_id=page_id) |
| 76 | + .select_related("created_by", "updated_by", "deleted_by") |
| 77 | + .order_by("order") |
| 78 | + ), |
| 79 | + many=True, |
| 80 | + ).data, |
| 81 | + ) |
| 82 | + |
| 83 | + @extend_schema( |
| 84 | + tags=[OpenAPITag.ADMIN_CMS], |
| 85 | + request=SectionAdminSerializer(many=True), |
| 86 | + responses={ |
| 87 | + status.HTTP_200_OK: SectionAdminSerializer(many=True), |
| 88 | + status.HTTP_400_BAD_REQUEST: ValidationErrorResponseSerializer, |
| 89 | + }, |
| 90 | + ) |
| 91 | + @decorators.action(detail=True, methods=["put"], url_path="section/bulk-update") |
| 92 | + @transaction.atomic |
| 93 | + def bulk_update_sections(self, request: request.Request, *args: tuple, **kwargs: dict) -> response.Response: |
| 94 | + if not (page_id := kwargs.get("pk")): |
| 95 | + return self._response_section_validation_error("페이지 ID가 제공되지 않았습니다.") |
| 96 | + |
| 97 | + section_qs = Section.objects.filter_active().filter(page_id=page_id).order_by("order") |
| 98 | + sections_data: list[SectionData] = request.data.get("sections", []) |
| 99 | + if not isinstance(sections_data, list): |
| 100 | + return self._response_section_validation_error("섹션 데이터는 리스트 형식이어야 합니다.") |
| 101 | + if not sections_data: |
| 102 | + return self._response_section_validation_error("섹션 데이터가 비어 있습니다.") |
| 103 | + |
| 104 | + id_in_new_sections = {sid for section_datum in sections_data if (sid := section_datum.get("id"))} |
| 105 | + section_qs.exclude(id__in=id_in_new_sections).delete() |
| 106 | + |
| 107 | + for section_datum in sections_data: |
| 108 | + section_id = section_datum.get("id") |
| 109 | + section_datum["page"] = page_id |
| 110 | + section_instance: Section | None = section_id and section_qs.filter(id=section_id).first() |
| 111 | + |
| 112 | + if section_id and not section_instance: |
| 113 | + return self._response_section_validation_error(f"<{section_id}> 섹션이 존재하지 않습니다.") |
| 114 | + |
| 115 | + serializer = SectionAdminSerializer(instance=section_instance, data=section_datum) |
| 116 | + serializer.is_valid(raise_exception=True) |
| 117 | + serializer.save() |
| 118 | + |
| 119 | + return response.Response(data=SectionAdminSerializer(instance=section_qs, many=True).data) |
0 commit comments