Django Query - 基于查询数据的预取过滤器?

时间:2018-01-21 09:53:10

标签: python django

我试图根据其父查询对象过滤预取,这可能吗?希望下面的例子可以解释。

来自SiteData的实时链接类型和来自Circuits的电路类型使用相同的子模型。

所以例如,如果实时链接类型是'光纤',我想只预取具有'光纤'电路类型的电路。每个站点可以有很多电路,但我现在只想要现场电视

这是查询:

conn_stats = SiteData.objects.all()
    .exclude(site_type__site_type='Factory') \
    .exclude(site_type__site_type='Data Centre') \
    .Prefetch(
        'circuits_set',
        queryset=Circuits.objects.filter(SiteData.objects.live_link_type.circuit_type)
        )
)

这些是模型:

class CircuitTypes(models.Model):
    circuit_type = models.CharField(max_length=50)
    monitor_priority = models.IntegerField(verbose_name="Monitoring Priority", blank=True, null=True)
    class Meta:
        verbose_name = "Circuit Types"
        verbose_name_plural = "Circuit Types"
    def __str__(self):
        return self.circuit_type

class SiteData(models.Model):
    location = models.CharField(max_length=50)
    site_type = models.ForeignKey(SiteTypes, verbose_name="Site Type", \
                on_delete=models.PROTECT)
    is_live = models.BooleanField(default=False, verbose_name="Is this a live site?")
    live_link_type = models.ForeignKey(CircuitTypes, verbose_name="Link Type", \
                on_delete=models.PROTECT, default=1)
    live_link_preference = models.CharField(max_length=200, blank=True, null=True)
    live_link_query_timestamp = models.DateTimeField(auto_now_add=True, blank=True, null=True)

class Circuits(models.Model):
    site_data = models.ForeignKey(SiteData, verbose_name="Site", on_delete=models.PROTECT)
    order_no = models.CharField(max_length=200, verbose_name="Order No")
    expected_install_date = models.DateField()
    install_date = models.DateField(blank=True, null=True)
    circuit_type = models.ForeignKey(CircuitTypes, verbose_name="Circuit Type", on_delete=models.PROTECT)
    ...

1 个答案:

答案 0 :(得分:1)

据我所知,你不能用一个查询集做你要求的事。

如果CircuitTypes的数量很小,您可以考虑为每种类型的电路构建一个查询集:

circuit_types = CircuitType.object.all()
for circuit_type in circuit_types:
    prefetch = Prefetch('circuits_set',
                        queryset=Circuits.objects.filter(
                            circuit_type=circuit_type)
    conn_stats = SiteData.objects.filter(live_link_type=circuit_type) \
        .exclude(site_type__site_type='Factory') \
        .exclude(site_type__site_type='Data Centre') \
        .prefetch_related(prefetch)

如果有许多电路类型并且这是禁止的,那么您可以向active添加一个布尔Circuits字段,并在更改每个SiteData.live_link_type时更新其值。这样,您就可以直接查询活跃的Circuits

prefetch = Prefetch('circuits_set',
                    queryset=Circuits.objects.filter(active=True)
conn_stats = SiteData.objects.all() \
    .exclude(site_type__site_type='Factory') \
    .exclude(site_type__site_type='Data Centre') \
    .prefetch_related(prefetch)