[python] Django REST framework에서의 투표 기능 처리 방법

29 Nov 2023

python

소개

Django REST framework는 Django 프레임워크를 기반으로한 강력한 API 개발 도구입니다. 이것은 우리가 강력하고 유연한 API를 빠르게 구축하고, 투표 기능과 같은 복잡한 기능을 구현하는 데 도움이 됩니다.

이 글에서는 Django REST framework를 사용하여 투표 기능을 처리하는 방법에 대해 알아보겠습니다.

모델 설정

우선, 투표 기능을 구현하기 위해 모델을 설정해야 합니다. 예를 들어, “투표” 모델은 사용자가 투표한 항목과 투표 수를 저장합니다.

from django.db import models

class Vote(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    item = models.ForeignKey(Item, on_delete=models.CASCADE)
    count = models.PositiveIntegerField(default=0)

위의 코드에서 Vote 모델은 User 모델과 Item 모델에 대한 외래 키로 구성되어 있습니다. count 필드는 해당 항목에 대한 투표 수를 저장합니다.

직렬화(serializer) 설정

투표 데이터를 직렬화하여 API에서 반환하기 위해 직렬화 클래스를 설정해야 합니다. 예를 들어, 아래와 같이 VoteSerializer 클래스를 작성할 수 있습니다.

from rest_framework import serializers

class VoteSerializer(serializers.ModelSerializer):
    class Meta:
        model = Vote
        fields = ('user', 'item', 'count')

위의 코드에서 VoteSerializer 클래스는 Vote 모델과 연결되어 있으며 user, item, count 필드를 직렬화합니다.

API 뷰 설정

투표 기능을 처리하는 API 뷰를 설정해야 합니다. 예를 들어, 투표 생성, 투표 업데이트 및 투표 조회를 위한 API 뷰를 작성할 수 있습니다.

from rest_framework import generics

class VoteCreateAPIView(generics.CreateAPIView):
    serializer_class = VoteSerializer

class VoteUpdateAPIView(generics.UpdateAPIView):
    serializer_class = VoteSerializer
    queryset = Vote.objects.all()

class VoteDetailAPIView(generics.RetrieveAPIView):
    serializer_class = VoteSerializer
    queryset = Vote.objects.all()

위의 코드에서 VoteCreateAPIView는 새로운 투표를 생성하기 위한 API 뷰이고, VoteUpdateAPIView는 투표를 업데이트하기 위한 API 뷰입니다. 마지막으로, VoteDetailAPIView는 특정 투표의 세부 정보를 가져오기 위한 API 뷰입니다.

URL 설정

마지막으로, API 뷰에 대한 URL을 설정해야 합니다. 예를 들어, urls.py 파일에 아래와 같이 URL 패턴을 추가할 수 있습니다.

from django.urls import path
from .views import VoteCreateAPIView, VoteUpdateAPIView, VoteDetailAPIView

urlpatterns = [
    path('vote/create/', VoteCreateAPIView.as_view(), name='vote-create'),
    path('vote/update/<int:pk>/', VoteUpdateAPIView.as_view(), name='vote-update'),
    path('vote/detail/<int:pk>/', VoteDetailAPIView.as_view(), name='vote-detail'),
]

위의 코드에서 맨 앞에 /vote/를 붙여 투표 관련 URL을 설정하고, VoteCreateAPIView, VoteUpdateAPIView, VoteDetailAPIView를 각각의 URL에 맵핑합니다.

결론

이제 Django REST framework를 사용하여 투표 기능을 처리하는 방법에 대해 알아보았습니다. 이를 통해 우리는 손쉽게 투표 기능을 구현할 수 있으며, API를 통해 투표 데이터를 관리할 수 있습니다.

더 자세한 내용은 Django REST framework 공식 문서를 참조하시기 바랍니다.