0

我正在使用 django-mongodb-engine 创建一个站点。

我有一个看起来像这样的自定义用户模型：

class User(AbstractUser):
    trigrama = models.CharField(max_length=3, unique=True)
    external_email = models.EmailField(unique=True)
    phone_number = models.CharField(max_length=20, unique=True)

    AbstractUser._meta.get_field_by_name('email')[0]._unique=True

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = [
        'first_name', 'last_name', 'trigrama',
        'external_email', 'phone_number', 'username'
    ]

它工作得很好，但后来我尝试向用户模型添加一个嵌入式文档：

    permissions = EmbeddedModelField('Permission')

Permission是一个看起来像这样的类：

class Permission(models.Model):
    computers = ListField(EmbeddedModelField('Computer'))
    projects = ListField(EmbeddedModelField('Project'))
    scripts = ListField(EmbeddedModelField('Script'))

现在，当我尝试使用在 shell 中创建用户时，User.objects.create()出现以下错误：

Traceback (most recent call last):
  File "<console>", line 1, in <module>
  File "C:\Python27\lib\site-packages\django\db\models\manager.py", line 149, in create
    return self.get_query_set().create(**kwargs)
  File "C:\Python27\lib\site-packages\django\db\models\query.py", line 416, in create
    obj.save(force_insert=True, using=self.db)
  File "C:\Python27\lib\site-packages\django\db\models\base.py", line 548, in save
    force_update=force_update, update_fields=update_fields)
  File "C:\Python27\lib\site-packages\django\db\models\base.py", line 668, in save_base
    result = manager._insert([self], fields=fields, return_id=update_pk, using=using, raw=raw)
  File "C:\Python27\lib\site-packages\django\db\models\manager.py", line 215, in _insert
    return insert_query(self.model, objs, fields, **kwargs)
  File "C:\Python27\lib\site-packages\django\db\models\query.py", line 1675, in insert_query
    return query.get_compiler(using=using).execute_sql(return_id)
  File "C:\Python27\lib\site-packages\djangotoolbox\db\basecompiler.py", line 583, in execute_sql
    "field) to None!" % field.name)
IntegrityError: You can't set permissions (a non-nullable field) to None!

我尝试将默认设置为Permission()使用：

permissions = EmbeddedModelField('Permission', default=Permission())

但后来我收到以下错误：

Traceback (most recent call last):
  File "<console>", line 1, in <module>
  File "C:\Python27\lib\site-packages\django\db\models\manager.py", line 149, in create
    return self.get_query_set().create(**kwargs)
  File "C:\Python27\lib\site-packages\django\db\models\query.py", line 416, in create
    obj.save(force_insert=True, using=self.db)
  File "C:\Python27\lib\site-packages\django\db\models\base.py", line 548, in save
    force_update=force_update, update_fields=update_fields)
  File "C:\Python27\lib\site-packages\django\db\models\base.py", line 668, in save_base
    result = manager._insert([self], fields=fields, return_id=update_pk, using=using, raw=raw)
  File "C:\Python27\lib\site-packages\django\db\models\manager.py", line 215, in _insert
    return insert_query(self.model, objs, fields, **kwargs)
  File "C:\Python27\lib\site-packages\django\db\models\query.py", line 1675, in insert_query
    return query.get_compiler(using=using).execute_sql(return_id)
  File "C:\Python27\lib\site-packages\djangotoolbox\db\basecompiler.py", line 579, in execute_sql
    connection=self.connection
  File "C:\Python27\lib\site-packages\djangotoolbox\fields.py", line 364, in get_db_prep_save
    (embedded_model, type(embedded_instance)))
TypeError: Expected instance of type <class 'users.models.Permission'>, not <type 'unicode'>.

那么我应该如何将嵌入式文档添加到自定义用户模型中呢？

PS如果我Permission()在shell中输入它实例化就好了，返回aPermission object而不是unicode。

4

1 回答 1

3

如果您阅读有关为字段设置默认值的方式的文档，您会看到它期望接收一个值...或一个可调用对象。你不能只是在那时和那里创建实例，它不会像那样工作（强调我的）：

默认值不能是可变对象（模型实例、列表、集合等），因为对该对象的同一实例的引用将用作所有新模型实例中的默认值。

当您自己调用它时（即Permission()），您得到的是一个实例，该实例又返回一个 unicode（它正在调用该__unicode__函数）。

换句话说，只需删除尾随()：

permissions = EmbeddedModelField('Permission', default=Permission)

现在，说实话，这可能是有问题的，因为它实际上并没有保存新的权限对象，整个事情可能会变得混乱（这是权限和所有，你要格外小心）。所以最好创建一个函数来包装操作，并使用它来代替：

def get_initial_perms():
    p = Permission()
    p.save()
    return p

# later:
permissions = EmbeddedModelField('Permission', default=get_initial_perms)

类似的解决方案是save()在调用实际save部分之前覆盖该方法并授予对象权限。

于 2014-04-02T19:17:01.863 回答