时间戳

首页 > 解决方案 > 如何过滤 django 中的相关字段？

python - 如何过滤 django 中的相关字段？

问题描述

假设我有与模型公司具有多对多关系的模型用户；和模型 UserType，它连接到 User 和 Company。像这样：

class User(models.Model):
    name = models.TextField()

class Company(models.Model):
    name = models.TextField()
    users = models.ManyToManyField(User, related_name="companies")

class UserType(models.Model):
    name = models.TextField()
    company = models.ForeignKey(Company, related_name="user_types")
    users = models.ManyToManyField(User, related_name="user_types")

我想找到公司中的所有用户，这很简单：User.objects.filter(companies=some_company). 但是，我还想过滤user_types返回的用户对象上的字段，以便只返回连接到给定公司的 UserType 对象。用代码来解释它，这应该返回 true：

def check_user_types(users, company):
    for user in users:
        for user_type in user.user_types:
            if user_type.company != company:
                return false
    return true

我该怎么做？

标签： pythondjangodjango-models

解决方案

我想到了。对于任何面临同样问题的人，这解决了它：

from django.db.models import Prefetch

users = User.objects.filter(companies=company).prefetch_related(
    Prefetch(
        "user_types",
        queryset=UserType.objects.filter(company=company),
    )
)

推荐阅读