使用 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` 来处理文件上传。在实际的开发过程中，你可以根据需要使用其他的解析器来处理文件上传。