Django 模型如何过滤 ForeignKey 对象的数量

发布于11月03日

我有一台A型和B型的，是这样的:

class A(models.Model):
  title = models.CharField(max_length=20)
  (...)

class B(models.Model):
  date = models.DateTimeField(auto_now_add=True)
  (...)
  a = models.ForeignKey(A)

现在我有大约A个和B个对象，我想要执行一个查询来 Select 指向它们的所有A个对象，这些对象少于2B个.

A有点像池，用户(B)加入池.如果只有1个或0个加入，则根本不应该显示池.

这样的模型设计可以吗？或者我应该稍微修改一下？

推荐答案

听起来像是extra人的工作.

A.objects.extra(
    select={
        'b_count': 'SELECT COUNT(*) FROM yourapp_b WHERE yourapp_b.a_id = yourapp_a.id',
    },
    where=['b_count < 2']
)

如果B计数是您经常需要作为过滤或排序标准的内容，或者需要在列表视图上显示，您可以考虑go 规范化，方法是向A模型添加b_count字段，并在添加或删除B时使用信号更新它:

from django.db import connection, transaction
from django.db.models.signals import post_delete, post_save

def update_b_count(instance, **kwargs):
    """
    Updates the B count for the A related to the given B.
    """
    if not kwargs.get('created', True) or kwargs.get('raw', False):
        return
    cursor = connection.cursor()
    cursor.execute(
        'UPDATE yourapp_a SET b_count = ('
            'SELECT COUNT(*) FROM yourapp_b '
            'WHERE yourapp_b.a_id = yourapp_a.id'
        ') '
        'WHERE id = %s', [instance.a_id])
    transaction.commit_unless_managed()

post_save.connect(update_b_count, sender=B)
post_delete.connect(update_b_count, sender=B)

另一种解决方案是在添加或删除相关B时管理A对象上的状态标志.

B.objects.create(a=some_a)
if some_a.hidden and some_a.b_set.count() > 1:
    A.objects.filter(id=some_a.id).update(hidden=False)

...

some_a = b.a
some_b.delete()
if not some_a.hidden and some_a.b_set.count() < 2:
    A.objects.filter(id=some_a.id).update(hidden=True)