基于Django REST Framework关联嵌套模型：实现分类下全量产品API

我来帮你搞定这个获取指定分类下所有产品的API，结合你给出的模型，咱们一步步来：

首先先把你的Products模型补全（你给出的代码里外键参数不完整），同时补充常见的产品字段示例：

# Models.py
from django.db import models

class Category(models.Model):
    main_category = models.CharField(max_length=200, default='')
    
    def __str__(self):
        return self.main_category

class Subcategory(models.Model):
    category = models.ForeignKey(Category, on_delete=models.CASCADE)
    sub_category = models.CharField(max_length=200, default='')
    
    def __str__(self):
        return self.sub_category

class Products(models.Model):
    sub_category = models.ForeignKey(Subcategory, on_delete=models.CASCADE)
    # 补充产品基础字段，可根据实际需求调整
    product_name = models.CharField(max_length=200)
    price = models.DecimalField(max_digits=10, decimal_places=2)
    description = models.TextField(blank=True)
    
    def __str__(self):
        return self.product_name

接下来咱们用Django REST Framework来实现API，步骤如下：

1. 创建序列化器 #

在你的app下新建serializers.py，用来序列化产品数据，还可以嵌套展示分类信息：

# serializers.py
from rest_framework import serializers
from .models import Products

class ProductSerializer(serializers.ModelSerializer):
    # 嵌套展示子分类和主分类名称，让返回数据更直观
    sub_category_name = serializers.CharField(source='sub_category.sub_category')
    main_category_name = serializers.CharField(source='sub_category.category.main_category')
    
    class Meta:
        model = Products
        # 按需选择要返回的字段
        fields = ['id', 'product_name', 'price', 'description', 'sub_category_name', 'main_category_name']

2. 编写视图 #

这里提供两种查询方式的视图：按分类ID查询、按分类名称查询，你可以按需选用：

# views.py
from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status
from .models import Category, Products
from .serializers import ProductSerializer

# 通过分类ID获取产品
class CategoryProductsByIdView(APIView):
    def get(self, request, category_id):
        try:
            # 先找到指定分类
            target_category = Category.objects.get(id=category_id)
            # 通过关联查询，筛选出该分类下所有子分类的产品
            category_products = Products.objects.filter(sub_category__category=target_category)
            # 序列化数据并返回
            serializer = ProductSerializer(category_products, many=True)
            return Response(serializer.data, status=status.HTTP_200_OK)
        except Category.DoesNotExist:
            return Response({"error": "指定分类不存在"}, status=status.HTTP_404_NOT_FOUND)

# 通过分类名称获取产品（用查询参数传递）
class CategoryProductsByNameView(APIView):
    def get(self, request):
        category_name = request.query_params.get('name')
        if not category_name:
            return Response({"error": "请提供分类名称参数"}, status=status.HTTP_400_BAD_REQUEST)
        
        try:
            target_category = Category.objects.get(main_category=category_name)
            category_products = Products.objects.filter(sub_category__category=target_category)
            serializer = ProductSerializer(category_products, many=True)
            return Response(serializer.data, status=status.HTTP_200_OK)
        except Category.DoesNotExist:
            return Response({"error": "指定分类不存在"}, status=status.HTTP_404_NOT_FOUND)

3. 配置URL路由 #

在你的app的urls.py中添加路由映射：

# urls.py
from django.urls import path
from .views import CategoryProductsByIdView, CategoryProductsByNameView

urlpatterns = [
    # 路由示例：/api/products/category/1/
    path('products/category/<int:category_id>/', CategoryProductsByIdView.as_view(), name='category-products-by-id'),
    # 路由示例：/api/products/category/?name=数码产品
    path('products/category/', CategoryProductsByNameView.as_view(), name='category-products-by-name'),
]

测试API #

配置完成后就可以测试了：

• 按ID查询：发送GET请求到http://your-domain/api/products/category/1/（替换成你的实际域名和分类ID）
• 按名称查询：发送GET请求到http://your-domain/api/products/category/?name=数码产品（替换成你的实际分类名称）

如果需要调整返回字段或者查询逻辑，直接修改序列化器或视图里的过滤条件即可。

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