2013-05-02 71 views
5

我创建了一个用户应用MyUser,它是AbstractBaseUser的扩展。我的印象是,该模型MyUser将取代标准Auth.User,只要它是在settings.pyDjango 1.5:了解AbstractBaseUser和权限

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 
    ... 

回答

8

只需将PermissionsMixin添加到您的模型声明! :)

class MyUser(AbstractBaseUser, PermissionsMixin): 
    ... 

这是Django文档的relevant part

+0

如何设置用户组级别?当我用'g.user_set.add(user)'设置它时,我得到'TypeError:'用户'实例,再次得到'作为错误消息。 – neurix 2013-05-03 00:22:41

+1

用'user.groups.add(g)'做。 – 2013-05-03 06:34:37