User Auth in Django

Question

Я начал изучать Django на прошлой неделе, и я подумал, что лучший способ - просто создать приложение. Я застрял теперь с модулем User Auth с моими испытаниями. Может быть, кто-то может мне помочь: Я хочу моделировать сотрудников в db, а также помещать их в модель User Auth, поэтому я создаю модель и после нескольких проверок я хочу добавить ее в User.

class Employee(models.Model): 

session = models.ForeignKey(
    'sessions.Session', 
    verbose_name='Session', 
    blank=True, null=True, 
) 
email = models.CharField(max_length=255) 
password1 = forms.CharField(max_length=30, widget=forms.PasswordInput()) #render_value=False 
password2 = forms.CharField(max_length=30, widget=forms.PasswordInput()) 
first_name = models.CharField(max_length=255) 
last_name = models.CharField(max_length=255) 
region = models.ForeignKey(Region, on_delete=models.CASCADE) 
workgroup = models.ForeignKey(Workgroups, on_delete=models.CASCADE) 
user_since = models.DateTimeField('Signed up since', default=timezone.now()) 
rights = models.IntegerField(default=0) 

def __str__(self): 
    return self.email 

def clean_username(self): # check if username dos not exist before 
    try: 
     User.objects.get(username=self.cleaned_data['email']) #get user from user model 
    except User.DoesNotExist : 
     return self.cleaned_data['email'] 

    raise ValidationError("this user exist already") 

def clean(self): # check if password 1 and password2 match each other 
    if 'password1' in self.cleaned_data and 'password2' in self.cleaned_data:#check if both pass first validation 
     if self.cleaned_data['password1'] != self.cleaned_data['password2']: # check if they match each other 
      raise ValidationError("passwords dont match each other") 

    return self.cleaned_data 

def dosave(self): # create new user 
    new_user=User.objects.create_user(self.cleaned_data['username'], 
           self.cleaned_data['email'], 
           self.cleaned_data['password1']) 
    new_user.save() 
    return new_user 

Итак, потом я написал несколько тестов, один из них следующие:

myuser = Employee.objects.create(first_name='Annie', 
            last_name='Levers', email='annielevers@test.com', 
            region=mymun, workgroup=mygroup, user_since=timezone.now()) 
    self.assertEquals(
     str(myuser), 
     'annielevers@test.com', 
    ) 

    # Check user.auth db 
    userpk = myuser.id 
    user = get_object_or_404(User, pk=userpk) 
    user.set_password('abcxyz123') 
    user.save() 

Когда я запускаю этот тест, питон не может найти запись для userpk в пользователя, но дает мне 404 объект.

Любые идеи? Большое спасибо!

Answer 1

Если вы хотите создать пользовательскую модель пользователя, вы должны использовать AbstractBaseUser.

https://docs.djangoproject.com/es/1.9/topics/auth/customizing/#specifying-a-custom-user-model