我在 Django 中工作,但在对我的数据模型实现特定约束时遇到了一些问题。我有一个由一组成员组成的模型。
class Member:
name = models.CharField(max_length=100)
class Team:
members = models.ManyToManyField(Member)
我想强制执行一个约束,使得对于任何给定的团队,成员集都是唯一的。也就是说,允许以下团队:
team_a: {A, B, C}
team_b: {A, B, D}
但是 team_c: {C, B, A} 是不允许的,因为成员集与 team_a 相同。
如果我尝试添加 team_c(或具有相同成员集的任何其他团队),我如何强制执行此约束以抛出错误?
我考虑过为每个成员设置一个外键,然后使用 unique_together,但这行不通,因为我希望模型不知道成员内的顺序(因为有数据中没有固有的顺序)。我认为这可能是 the m2m_changed signal 的工作,但我无法让它正常工作。
请您参考如下方法:
嗯,没有直接的方法,但在我的头脑中这可能有效。
只有当您要添加的成员存在于其他团队中时,一组成员与另一个团队成员发生冲突的唯一可能性,因此首先我们需要获取那些已添加用户的团队:
# fetch teams where user is already added
teams = Team.objects.filter(members__id=user.id)
will_clash = False
if teams:
# prepare list of tuples of sorted members id for each team
members_set = [tuple(team.members.values_list('id', flat=True).order_by('id')) for team in teams]
# we know already in which team we want to add lets say it is `team_c`
team_c_members_set = tuple(sorted(list(team_c.members.values_list('id', flat=True)) + [user.id]))
if team_c_members_set in members_set:
# adding current `user` is going to create a clash
will_clash = True
if not will_clash:
# all good lets add the user in team
team_c.members.add(user)
试运行:
假设您有以下三个团队:
Team A:
1: member
2: member
3: member
Team B:
1: member
2: member
4: member
Team C:
3: member
2: member
现在我们要将成员 id: 1 添加到团队 C 中。应该避免这种情况,因为这会创建与 A 团队中相同的一组用户,让我们运行上面的代码看看这是否是 True:
# fetch teams where user is already added
teams = [A, B] # user already there
# prepare list of tuples of sorted members id for each team
members_set = [(1, 2, 3), (1, 2, 4)]
# we know already in which team we want to add lets say it is `team_c` so it will become
team_c_members_set = (1, 2, 3) # tuple(sorted([3, 2] + [1]))
team_c_members_set in members_set # <-- True
一次性添加完整团队的场景
假设您要立即添加完整的团队 C,因此您已经知道您将在该团队中拥有哪些成员,在这种情况下过程几乎相似,但 team_c_members_set首先创建用于过滤团队:
users_to_add = [user_a, user_b, user_c]
team_c_members_set = tuple(sorted([u.id for u in users_to_add])
teams = Team.objects.filter(members__id__in=team_c_members_set).distinct()
will_clash = False
if teams:
# prepare list of tuples of sorted members id for each team
members_set = [tuple(team.members.values_list('id', flat=True).order_by('id')) for team in teams]
if team_c_members_set in members_set:
will_clash = True
if not will_clash:
# all good lets add the users in team
team_c.members.add(*users_to_add)
