如何重写Django REST框架创建接口响应仅返回主键

直接利用Django模型实例的pk属性即可，不管你的模型主键字段叫什么（id、uid、user_code等），pk都会自动指向该模型的主键值。修改后的create方法如下：

def create(self, request, *args, **kwargs):
    serializer = self.get_serializer(data=request.data)
    serializer.is_valid(raise_exception=True)
    self.perform_create(serializer)
    headers = self.get_success_headers(serializer.data)
    # 直接从实例获取主键，无需关心字段名
    instance_pk = serializer.instance.pk
    # 返回包装成字典的主键（推荐，符合JSON响应规范）
    return Response({"pk": instance_pk}, status=status.HTTP_201_CREATED, headers=headers)

关键说明 #

• serializer.instance：调用perform_create后，序列化器会持有刚创建的模型实例，通过这个属性可以直接访问实例对象。
• .pk属性：Django所有模型都内置这个属性，它是模型主键字段的通用别名，不管你在模型里定义的主键字段是id还是自定义名称，都能通过pk拿到主键值。
• 如果需要返回纯主键值而非字典，可改成：

  return Response(instance_pk, status=status.HTTP_201_CREATED, headers=headers)
  

  但更推荐返回字典结构，这样前端可以通过固定的键名（比如pk）获取值，避免类型解析问题。
• 保留get_success_headers：这部分会生成符合REST规范的响应头（比如Location），不要省略。

内容的提问来源于stack exchange，提问作者Step