使用 DRF 實現文件上傳

文件上傳是 web 應用程序中最常見的需求之一，本文將介紹如何使用 Django Rest Framework (DRF) 來實現文件上傳。通過本文，你將學習到如何使用 DRF 中的 `FileUploadParser` 來完成文件上傳。

一、創建視圖

首先，在 app 中創建一個視圖來處理文件上傳請求。我們將在視圖中指定我們的文件上傳解析器為 Django 自帶的文件上傳解析器 `FileUploadParser`。這樣會使上傳過程變得非常簡單。

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

class FileUploadView(APIView):
    parser_classes = (FileUploadParser,)

    def post(self, request, format=None):
        file_obj = request.data['file']
        # 處理上傳的文件
        return Response({'status': 'success'})

在上面的代碼中，我們將視圖命名為 `FileUploadView`。我們將該視圖設置為 `APIView` 的子類，並指定我們將要使用的文件上傳解析器 `FileUploadParser`。

在視圖中，我們重寫了 `post` 方法來處理文件上傳請求。在該方法中，我們從請求數據中獲取文件對象並進行處理。

二、配置路由

現在我們需要將我們新創建的視圖與一個 URL 綁定。我們可以在項目的 `urls.py` 文件中完成這一過程。

from django.urls import path

from .views import FileUploadView

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

在上面的代碼中，我們將 `FileUploadView` 註冊為 URL 路由的處理程序。我們指定了 `upload/` 作為文件上傳終端路由。

三、測試

現在，我們準備好進行測試了。你可以使用任何適合您的 HTTP 客戶端。在這裡，我們使用 Python 自帶的 HTTP 客戶端 `Requests` 庫來測試 API 接口。

import requests

url = 'http://example.com/api/upload/'
filename = 'example.txt'
with open(filename, 'rb') as f:
    r = requests.post(url, files={'file': f})
    print(r.json())

在上面的代碼中，我們設定了上傳文件所在的路徑和上傳文件名稱。我們使用 `requests` 庫發送 POST 請求並指定要上傳的文件。服務器將該文件保存到一個臨時文件中，並向客戶端返回一個 JSON 格式的響應。

至此，我們已經完成了使用 DRF 完成文件上傳。在本文中，我們在 Django 中使用 DRF 中的 `FileUploadParser` 來處理文件上傳。在實際的開發過程中，你可以根據需要使用其他的解析器來處理文件上傳。