在django中实现“团队”的模型

时间:2015-04-25 16:56:40

标签: python django django-models

我想在django 1.8中实现团队功能。 (团队在体育团队中)

每个用户一次最多可以加入一个团队,因此团队可以容纳许多用户。现在我不确定如何定义我的models.py

我从这个核心开始,但现在我不确定如何建立Team< - > User

的连接
from django.db import models

class Team(models.Model):
    name = models.CharField(max_length=64, unique=True)
    description = models.TextField(max_length=1024)
    logo = models.ImageField()

from django.contrib.auth.models import User

class Player(models.Model):
    user = models.OneToOneField(User)
    team = ForeignKey('Team')

我现在是否创建了第二个类user_team,或者我只是将团队作为外键添加到用户? (如果那就是我需要这样做的方式吗?)

谢谢,

Wegi

//编辑:我在底部添加了一些代码。这个玩家模型是否足以定义关系?

2 个答案:

答案 0 :(得分:8)

对于这个用例,我仍然会建议使用 ManyToMany 字段,使用中间模型和模型管理器。

快速样本结构如下所示:

from django.db import models
from django.contrib.auth.models import User


class Team(models.Model):
    name = models.CharField(max_length=64, unique=True)
    description = models.TextField(max_length=1024)
    logo = models.ImageField()
    players = models.ManyToManyField(User, through='Player')


class PlayerManager(models.Manager):
    use_for_related_fields = True

    def add_player(self, user, team):
        # ... your code here ...

    def remove_player(self, user, team):
        # ... your code here ...

    def trasnfer_player(self, user, team):
        # ... your code here ...


class Player(model.Model):
    user = models.ForeignKey(User)
    team = models.ForeignKey(Team)
    other_fields = #...

    objects = PlayerManager()

用法:

Player.objects.add_player(user, team, *other_fields)

然后,您将能够获得与用户相关的团队,例如:

team_with_user = Team.objects.filter(players__name="hello")
user_in_team = User.objects.filter(team__name="world")

注意:我还没有对代码进行测试,如果我上面有任何错误,请纠正我。

我更喜欢这种方式的原因是将数据库逻辑抽象为应用程序。因此,将来如果需要允许User加入多个团队,您只需更改应用程序逻辑以允许它通过管理器。

答案 1 :(得分:2)

正如@aumo所建议,我通过添加如下用户个人资料模型解决了这个问题:

from django.contrib.auth.models import User

class Player(models.Model):
    user = models.OneToOneField(User)
    team = models.ForeignKey('Team')

我选择此解决方案而不是将团队添加为Teams类中的ManyToMany字段,因为我不确定在开发期间是否需要将更多字段添加到播放器。

感谢大家的帮助。