Question

我创建了一个用户应用MyUser，它是AbstractBaseUser的扩展。我的印象是模型MyUser将取代标准Auth.User，只要settings.py

中提到它

AUTH_PROFILE_MODULE ='profile.MyUser'

我现在的麻烦是我无法为在MyUser模型中注册的用户设置权限。当我尝试设置组成员身份和权限时，收到错误User' instance expected, got <MyUser: username>。

如何将用户模型的用户添加到权限和更正组？

class MyUserManager(BaseUserManager):
    def create_user(self, username, email, phone, password=None, company=False):
        if not email:
            raise ValueError('Users must have an email address')
        if not username: username = email.split('@')[0]
        user = self.model(
            email=MyUserManager.normalize_email(email),
            username=username,phone=phone,)
        user.set_password(password)
        user.save(using=self._db)

        # add to user group and set permissions
        if company: 
            g = Group.objects.get(name='company')
            p = Permission.objects.get(codename='add_company')
        else: 
            g = Group.objects.get(name='user')
            p = Permission.objects.get(codename='add_user')
        g.user_set.add(user)
        user.user_permissions.add(p)
        return user

class MyUser(AbstractBaseUser):
    username = models.CharField(max_length=254, unique=True, blank=True, null=True)
    email = models.EmailField(max_length=254, unique=True, db_index=True)
    phone = models.CharField(_('Phone Number'), max_length=25, blank=True, null=True,)

    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)
    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['phone']

    objects = MyUserManager()

    def get_full_name(self):
        return self.email
    ...

Answer 1

只需将PermissionsMixin添加到您的模型声明中即可！：）

class MyUser(AbstractBaseUser, PermissionsMixin):
    ...

以下是Django文档的relevant part。

Answer 2

您可能已经解决了您的问题但是为了完整性，您是指在Django 1.5中创建自定义用户模型时参考设置AUTH_USER_MODEL与AUTH_PROFILE_MODULE。

参考：https://docs.djangoproject.com/en/1.5/topics/auth/customizing/#substituting-a-custom-user-model

Django 1.5：了解AbstractBaseUser和权限

2 个答案: