本文介绍了在使用 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()代码解释：
- 获取 request.data: 首先，通过 request.data 获取客户端上传的 FormData 数据。
- 迭代 request.data: 使用 for k, v in request.data.items(): 迭代 FormData 中的每一个键值对。
- 检查键名: 使用 if k == 'company_details[0][documents]': 判断当前键是否为目标键。
- 迭代文件列表: 如果键名匹配，则使用 for i in v: 迭代文件列表。
- 处理文件: 在内部循环中，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 需要仔细地迭代和访问数据。通过使用循环和条件判断，可以有效地提取所需的信息。对于更复杂的嵌套结构，可以考虑使用递归函数来简化处理过程。 在处理文件上传时，请务必注意安全性和文件大小限制，以防止潜在的安全风险。
