[python] Django REST framework에서의 파일 업로드와 다운로드 처리

29 Nov 2023

python

Django REST framework은 RESTful API를 빠르고 쉽게 구축할 수 있는 강력한 도구입니다. 이 프레임워크를 사용하여 파일 업로드와 다운로드를 처리하는 방법에 대해 알아보겠습니다.

1. 파일 업로드

Django REST framework에서 파일 업로드를 처리하기 위해서는 다음과 같은 단계를 따라야 합니다.

1-1. serializer 작성

파일 업로드를 처리하기 위해서는 serializer를 작성해야 합니다. Serializer는 API 요청을 받아서 데이터를 유효성 검사하고 필요한 형식으로 변환하는 역할을 합니다. 파일을 업로드할 필드는 FileField를 사용하면 됩니다.

from rest_framework import serializers

class FileUploadSerializer(serializers.Serializer):
    file = serializers.FileField()

1-2. view 작성

파일 업로드를 처리하는 view를 작성해야 합니다. 이 view는 POST 요청을 받아 request.FILES의 값을 사용하여 업로드한 파일을 처리합니다.

from rest_framework.views import APIView
from rest_framework.response import Response

class FileUploadView(APIView):
    def post(self, request):
        serializer = FileUploadSerializer(data=request.data)
        if serializer.is_valid():
            # 파일을 저장하거나 다른 작업을 수행합니다.
            file = serializer.validated_data['file']
            # ...
            return Response({'status': 'success'})
        else:
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

1-3. URL 매핑

마지막으로 업로드를 처리하는 view를 URL에 매핑해야 합니다. urls.py 파일에 다음과 같은 코드를 추가합니다.

from django.urls import path
from .views import FileUploadView

urlpatterns = [
    path('upload/', FileUploadView.as_view(), name='file-upload')
]

이제 파일을 업로드할 수 있는 API 엔드포인트가 준비되었습니다.

2. 파일 다운로드

Django REST framework에서 파일 다운로드를 처리하기 위해서는 다음과 같은 단계를 따라야 합니다.

2-1. view 작성

파일 다운로드를 처리하는 view를 작성해야 합니다. 이 view는 GET 요청을 받아 파일을 응답으로 보내주는 역할을 합니다.

from django.http import FileResponse

class FileDownloadView(APIView):
    def get(self, request):
        # 파일을 가져오거나 다른 작업을 수행합니다.
        file_path = '/path/to/file'
        # ...
        file = open(file_path, 'rb')  # 바이너리 파일을 열어야 합니다.
        return FileResponse(file)

2-2. URL 매핑

마지막으로 다운로드를 처리하는 view를 URL에 매핑해야 합니다. urls.py 파일에 다음과 같은 코드를 추가합니다.

from django.urls import path
from .views import FileDownloadView

urlpatterns = [
    path('download/', FileDownloadView.as_view(), name='file-download')
]

이제 파일을 다운로드할 수 있는 API 엔드포인트가 준비되었습니다.

마무리

Django REST framework를 사용하여 파일 업로드와 다운로드를 처리하는 방법을 알아보았습니다. 이를 통해 다양한 파일 관련 기능을 API로 구현할 수 있습니다.

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