10


我的目标是在 Django 1.5 中创建自定义用户模型

# myapp.models.py 
from django.contrib.auth.models import AbstractBaseUser

class MyUser(AbstractBaseUser):
    email = models.EmailField(
        verbose_name='email address',
        max_length=255,
        unique=True,
        db_index=True,
    )
    first_name = models.CharField(max_length=30, blank=True)
    last_name = models.CharField(max_length=30, blank=True)
    company = models.ForeignKey('Company')
    ...

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['company']

由于公司字段（models.ForeignKey（'Company'）（python manage.py createsuperuser），我无法创建超级用户。我的问题：
如何在没有公司的情况下为我的应用程序创建超级用户。我试图制作自定义 MyUserManager 没有任何成功：

class MyUserManager(BaseUserManager):
    ...

    def create_superuser(self, email, company=None, password):
        """
        Creates and saves a superuser with the given email and password.
        """
        user = self.create_user(
            email,
            password=password,
        )
        user.save(using=self._db)
        return user

还是我必须为这个用户创建一个假公司？谢谢

4

2 回答 2

7

在这种情况下，您有三种方法

1) 与公司建立关系 不需要company = models.ForeignKey('Company', null=True)

2）添加默认公司并将其作为默认值提供给外键字段company = models.ForeignKey('Company', default=1) #其中1是创建公司的ID

3) 保持模型代码不变。为名为“Superusercompany”的超级用户添加假公司，在 create_superuser 方法中设置它。

UPD：根据您的评论方式＃3 将是不破坏您的业务逻辑的最佳解决方案。

于 2013-04-13T14:59:35.693 回答
5

感谢您的反馈，这是我提出的解决方案：我在其中创建了默认公司的自定义 MyUserManager

    def create_superuser(self, email, password, company=None):
        """
        Creates and saves a superuser with the given email and password.
        """

        if not company:
            company = Company(
                name="...",
                address="...",
                code="...",
                city="..."
            )
            company.save()

        user = self.create_user(
            email,
            password=password,
            company=company
        )
        user.is_admin = True
        user.save(using=self._db)
        return user
于 2013-04-14T16:20:44.160 回答