2013-05-02 10 views
5

Ho creato un'app utente MyUser che è un'estensione di AbstractBaseUser. Ho avuto l'impressione che il modello MyUser sostituirà lo standard Auth.User, fintanto che è menzionato nel settings.pyDjango 1.5: Comprensione di AbstractBaseUser e autorizzazioni

AUTH_PROFILE_MODULE = 'profile.MyUser'

Il mio problema ora è che non riesco a impostare il autorizzazioni per gli utenti registrati nel modello MyUser. Quando provo a impostare le appartenenze e le autorizzazioni del gruppo, ottengo l'errore User' instance expected, got <MyUser: username>.

Come posso aggiungere utenti del mio modello utente alle autorizzazioni e ai gruppi corretti?

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

risposta

8

Basta aggiungere PermissionsMixin alla dichiarazione del modello! :)

class MyUser(AbstractBaseUser, PermissionsMixin): 
    ... 

Ecco la relevant part di documenti Django.

+0

Come faccio a impostare il livello di gruppo per l'utente? Quando l'ho impostato con 'g.user_set.add (user)', ottengo 'TypeError: L'istanza 'User' prevista, ha ottenuto ' come messaggio di errore di nuovo. – neurix

+1

Fatelo con 'user.groups.add (g)'. –