在Django中为多对多关系添加记录的正确方法

时间:2010-12-31 23:47:52

标签: django django-models django-managers django-orm

首先,我计划在谷歌应用引擎上运行我的项目,所以我使用djangoappengine,据我所知,它不支持django的ManyToManyField类型。因此,我设置了这样的模型:

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

class Group(models.Model):
    name = models.CharField(max_length=200)

class UserGroup(models.Model):
    user = models.ForeignKey(User)
    group = models.ForeignKey(Group)

在页面上,我有一个表单字段,人们可以在其中输入组名。我希望此表单字段的结果为用户 - 组合组合创建一个UserGroup对象,如果该组尚不存在,则创建一个新的Group对象。起初我开始使用add_group方法将此逻辑放在UserGroup类中,但很快意识到将它放在UserGroup类中并没有意义。这样做的正确方法是什么?我看到了一些关于模型经理的事情。这是为了什么?

1 个答案:

答案 0 :(得分:3)

来自Django docs

  

Manager是为Django模型提供数据库查询操作的接口

因此,经理不是创建模型新实例的方法。

我只有一个Group模型的构造函数,它从关键字参数中提取User(如果给出)并创建一个新的UserGroup

因此,如果您将新的Group实例化为Group(name='group name', user=some_user),构造函数可以删除user关键字参数,并使用UserGroup创建相应的user

class Group(models.Model):
    name = models.CharField(max_length=200)

    def __init__(self, *args, **kwargs):
        # Remove user from the kwargs.
        # It is important that user is removed from the kwargs before calling
        # the super.__init__(). This is because the `user` kwarg is not expected.
        user = kwargs.pop('user', None)

        # call the normal init method
        super(Group, self).__init__(*args, **kwargs)

        # Create the UserGroup instance for the specified user
        if user is not None:
            # Maybe some other logic here ...
            UserGroup(user=user, group=self).save()

这样,如果您在实例化user时提供Group,则会创建UserGroup,否则将无法执行任何操作。

当然你可以在UserGroup模型的构造函数中做同样的事情,它并不是那么重要,它只是取决于你的代码隐喻有意义。

修改

修正了评论中指出的问题:

...
def __init__(self, *args, **kwargs):
    ...
    self._user_group = UserGroup(user=user, group=self)

def save(self, *args, **kwargs):
    super(Group, self).save(*args, **kwargs)
    self._user_group.save()