Django REST Framework 处理嵌套 FormData 方法

本文介绍了在使用 Django REST Framework 处理包含嵌套键的 FormData 时，如何正确地迭代和访问数据。特别针对 company_details[0][documents] 这种形式的键，提供了一种有效的迭代方法，帮助开发者从 FormData 中提取所需的文件信息，并进行后续处理，例如保存文件或进行其他业务逻辑操作。

处理 Django REST Framework 中的嵌套 FormData

在使用 Django REST Framework (DRF) 开发 API 时，经常需要处理客户端上传的 FormData。FormData 允许客户端以键值对的形式发送数据，其中键可以是简单的字符串，也可以包含嵌套结构，例如 company[company_owner][email]。

当 FormData 包含嵌套键时，直接访问数据可能会遇到问题。例如，对于 company_details[0][documents] 这样的键，直接使用 request.data['company_details[0][documents]'] 可能无法获取到期望的结果。

迭代 FormData 并访问嵌套键

以下是一个处理包含 company_details[0][documents] 嵌套键的 FormData 的示例：

from rest_framework.decorators import api_view, permission_classes, parser_classes
from rest_framework.parsers import FormParser, MultiPartParser
from rest_framework.permissions import AllowAny
from rest_framework.response import Response

@api_view(['POST'])
@permission_classes([AllowAny])
@parser_classes([FormParser, MultiPartParser])
def test(request):
    print(request.data)
    data = request.data  # 获取 request.data

    for k, v in data.items():
        if k == 'company_details[0][documents]':
            for i in v:
                print(i)
                # 在这里编写处理文件的逻辑
                # 例如，保存文件到服务器
                # with open(i.name, 'wb+') as destination:
                #     for chunk in i.chunks():
                #         destination.write(chunk)

    return Response()

代码解释：

1. 获取 request.data: 首先，通过 request.data 获取客户端上传的 FormData 数据。
2. 迭代 request.data: 使用 for k, v in request.data.items(): 迭代 FormData 中的每一个键值对。
3. 检查键名: 使用 if k == 'company_details[0][documents]': 判断当前键是否为目标键。
4. 迭代文件列表: 如果键名匹配，则使用 for i in v: 迭代文件列表。
5. 处理文件: 在内部循环中，i 代表一个上传的文件对象。你可以使用 i.name 获取文件名，并使用 i.chunks() 迭代文件内容，然后将文件保存到服务器。

重要提示：

• 确保 FormData 中包含 company_details[0][documents] 键。如果键不存在，上述代码将不会执行文件处理逻辑。
• 文件上传时，确保前端正确设置 FormData 的 Content-Type 为 multipart/form-data。

更通用的解决方案

上面的代码只针对特定的键 company_details[0][documents]。如果需要处理更通用的嵌套结构，可以考虑使用递归函数来解析 request.data：

def process_nested_data(data, prefix=''):
    for key, value in data.items():
        new_prefix = f"{prefix}[{key}]" if prefix else key
        if isinstance(value, dict):
            process_nested_data(value, new_prefix)
        elif isinstance(value, list):
            for i, item in enumerate(value):
                if hasattr(item, 'read'): # Check if it's a file-like object
                    print(f"Found file at {new_prefix}[{i}]: {item.name}")
                    # Handle file here
                else:
                    print(f"Value at {new_prefix}[{i}]: {item}")
        else:
            print(f"Value at {new_prefix}: {value}")

# Example usage in your view
@api_view(['POST'])
@permission_classes([AllowAny])
@parser_classes([FormParser, MultiPartParser])
def test(request):
    print(request.data)
    process_nested_data(request.data)
    return Response()

这个递归函数可以遍历任意深度的嵌套字典和列表，并且可以检测到文件对象。

总结

处理 Django REST Framework 中的嵌套 FormData 需要仔细地迭代和访问数据。通过使用循环和条件判断，可以有效地提取所需的信息。对于更复杂的嵌套结构，可以考虑使用递归函数来简化处理过程。 在处理文件上传时，请务必注意安全性和文件大小限制，以防止潜在的安全风险。