在Django中如何关联层级车型数据至Battery模型及实现查询

模型关联方案 #

直接给Battery模型添加指向树形车型模型的ManyToMany字段完全可行。因为不管车型层级有多深，每个品牌、细分市场、具体车型都是独立的CarModel实例，多对多关联可以精准绑定到任意层级的节点——既可以关联单个具体车型（比如奔驰ModelA），也可以后续通过树形查询批量关联某品牌/细分市场下的所有车型，无需修改字段结构。

伪代码示例（基于django-mptt） #

先定义树形车型模型：

from django.db import models
from mptt.models import MPTTModel, TreeForeignKey

class CarModel(MPTTModel):
    name = models.CharField(max_length=100)
    parent = TreeForeignKey('self', on_delete=models.CASCADE, null=True, blank=True, related_name='children')

    class MPTTMeta:
        order_insertion_by = ['name']

再定义关联电池的模型：

class Battery(models.Model):
    name = models.CharField(max_length=100)
    compatible_models = models.ManyToManyField(CarModel, related_name='compatible_batteries')
    # 其他电池属性示例
    capacity = models.FloatField(verbose_name='容量(kWh)')
    voltage = models.FloatField(verbose_name='电压(V)')

查询写法示例 #

查询兼容奔驰ModelA的所有电池 #

先定位到目标车型节点，再反向查询关联的电池：

# 精准定位奔驰ModelA节点
mercedes_model_a = CarModel.objects.filter(
    name='ModelA',
    parent__name='Consumer',
    parent__parent__name='Mercedes'
).first()

# 获取所有兼容该车型的电池
target_batteries = Battery.objects.filter(compatible_models=mercedes_model_a)

如果需要更灵活的树形查询（比如查询兼容奔驰所有Consumer细分市场车型的电池），可以利用django-mptt的树形方法：

# 定位奔驰Consumer节点
mercedes_consumer = CarModel.objects.filter(
    name='Consumer',
    parent__name='Mercedes'
).first()

# 查询该节点下所有子车型对应的电池
target_batteries = Battery.objects.filter(
    compatible_models__in=mercedes_consumer.get_descendants(include_self=False)
)

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