用戶認證在Django

Question

我上週開始學習Django，我認爲最好的方法就是構建一個App。我現在用我的測試卡住用戶認證模塊。也許有人可以幫助我： 我想爲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