2013-05-02 17 views
5

Ich habe eine Benutzer-App MyUser erstellt, die eine Erweiterung von AbstractBaseUser ist. Ich habe den Eindruck, dass das Modell MyUser wird der Standard Auth.User, so lange ersetzen, wie es in settings.pyDjango 1.5: Verständnis von AbstractBaseUser und Berechtigungen

AUTH_PROFILE_MODULE erwähnt = ‚profile.MyUser‘

Mein Problem ist jetzt, dass ich nicht die festlegen Berechtigungen für Benutzer, die im MyUser-Modell registriert sind. Wenn ich versuche, die Gruppenmitgliedschaften und Berechtigungen festzulegen, erhalte ich den Fehler User' instance expected, got <MyUser: username>.

Wie kann ich Benutzer meines Benutzermodells den Berechtigungen und den richtigen Gruppen hinzufügen?

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

Antwort

8

Fügen Sie einfach PermissionsMixin zu Ihrem Modell Erklärung! :)

class MyUser(AbstractBaseUser, PermissionsMixin): 
    ... 

Hier ist die relevant part von Django docs.

+0

Wie kann ich die Gruppenstufe für den Benutzer festgelegt? Wenn ich es mit 'g.user_set.add (user)' setze, bekomme ich 'TypeError: 'User' Instanz erwartet, habe ' als Fehlermeldung bekommen. – neurix

+1

Tun Sie es mit 'user.groups.add (g)'. –

Verwandte Themen