Django ORM 注释定义的布尔字段是团队的用户成员
Django ORM Annotate boolean field that defines is user member of a team
models.py
from django.db import models
class UserGroup(models.Model):
members = models.ManyToManyField(User, related_name='members', through='UserGroupMember')
class UserGroupMember(models.Model):
user = models.ForeignKey(User)
usergroup = models.ForeignKey(UserGroup)
class Cohort(models.Model):
user_groups = models.ManyToManyField(UserGroup)
class Team(models.Model):
cohort = models.ForeignKey(Cohort)
members = models.ManyToManyField(User, related_name='team_members', through='TeamMembers', blank=True)
class TeamMembers(models.Model):
team = models.ForeignKey(Team)
user = models.ForeignKey(User)
单个用户只能是同类群组中一个团队的一员。
我想注释新字段(布尔值),它告诉您是分配给同类群组中某个团队的用户,例如:
User.objects.filter(
members__cohort=cohort
).annotate(
is_team_member=...
)
我正在使用 Python 2.7.13 和 Django 1.9.8。谢谢
我通过加入 Cohort
模型并使用条件表达式设法解决了这个问题:
from django.db.models import Case, When, Value, IntegerField
users = User.objects.filter(
members__cohort=cohort
).annotate(
is_team_member=Case(When(team_members__cohort=cohort, then=Value(1)), default=0, output_field=IntegerField())
)
现在我可以轻松过滤,例如,属于某个团队的用户:
users.filter(is_team_member=1)
models.py
from django.db import models
class UserGroup(models.Model):
members = models.ManyToManyField(User, related_name='members', through='UserGroupMember')
class UserGroupMember(models.Model):
user = models.ForeignKey(User)
usergroup = models.ForeignKey(UserGroup)
class Cohort(models.Model):
user_groups = models.ManyToManyField(UserGroup)
class Team(models.Model):
cohort = models.ForeignKey(Cohort)
members = models.ManyToManyField(User, related_name='team_members', through='TeamMembers', blank=True)
class TeamMembers(models.Model):
team = models.ForeignKey(Team)
user = models.ForeignKey(User)
单个用户只能是同类群组中一个团队的一员。
我想注释新字段(布尔值),它告诉您是分配给同类群组中某个团队的用户,例如:
User.objects.filter(
members__cohort=cohort
).annotate(
is_team_member=...
)
我正在使用 Python 2.7.13 和 Django 1.9.8。谢谢
我通过加入 Cohort
模型并使用条件表达式设法解决了这个问题:
from django.db.models import Case, When, Value, IntegerField
users = User.objects.filter(
members__cohort=cohort
).annotate(
is_team_member=Case(When(team_members__cohort=cohort, then=Value(1)), default=0, output_field=IntegerField())
)
现在我可以轻松过滤,例如,属于某个团队的用户:
users.filter(is_team_member=1)