使用自定义用户模型在Django的作者字段中自动添加用户名字

时间:2019-09-08 07:18:27

标签: python django

我使用以下代码在Django中创建了自定义用户模型

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(_('email address'), unique=True)
    username = models.CharField(_("Username"), max_length=50, unique=True)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    date_joined = models.DateTimeField(default=timezone.now)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['username', ]

    objects = CustomUserManager()

    def __str__(self):
        return self.email

我还为用户个人资料创建了另一个模型,如下所示。

class UserProfile(models.Model):
    user = models.OneToOneField(User, verbose_name=_("User"), on_delete=models.CASCADE)
    first_name = models.CharField(_("First name"), max_length=50)
    middle_name = models.CharField(_("Middle name"), max_length=50, blank=True, null=True)
    last_name = models.CharField(_("Last name"), max_length=50)
    dob = models.DateField(_("D.O.B"), auto_now=False, auto_now_add=False, blank=False, null=False)
    profile_image = models.ImageField(_("Profile picture"), upload_to='user/profile/', blank=True)
    webiste_link = models.URLField(_("Websites"), max_length=200, null=True, blank=True)

最后,我为“类别”创建了另一个模型,如下所示。

class Category(models.Model):
    author = models.ForeignKey(settings.AUTH_USER_MODEL, verbose_name=_(
        "Author"), on_delete=models.CASCADE, related_name='blog_category')
    title = models.CharField(_("Title"), max_length=50)

我想自动保存登录的用户全名,而不是用户的电子邮件地址。顺便说一下,我使用下面的代码来自动保存登录用户。可以,但是只保存用户的电子邮件地址。但是我想要用户全名,而不是电子邮件或用户名。

    obj.author = request.user
    super().save_model(request, obj, form, change)

2 个答案:

答案 0 :(得分:1)

您可以在UserProfile模型中定义一个方法来构建此全名,因为您可能需要在另一个函数中使用它:

class UserProfile(models.Model):
    user = models.OneToOneField(User, verbose_name=_("User"), on_delete=models.CASCADE)
    [...]

    def get_fullname(self):
        return ' '.join([name for name in [
                      self.first_name, 
                      self.middle_name, 
                      self.last_name
                      ] if name])

因此,您可以采用这种方式将用户外键保存在Category中:

obj.author = request.user

然后,要在类别实例上调用全名,可以使用:

# cat = Category.objects.first() 
cat.author.userprofile.get_fullname()

答案 1 :(得分:0)

您可以在 UserProfile 模型中添加名为全名的文件

Class UserProfile(models.Model):
     user = models.OneToOneField(User, verbose_name=_("User"), on_delete=models.CASCADE)
     first_name = models.CharField(_("First name"), max_length=50)
     full_name = mdoels.CharField(_("Full name"),max_length=100)
     middle_name = models.CharField(_("Middle name"), max_length=50, blank=True, null=True)
     last_name = models.CharField(_("Last name"), max_length=50)
     dob = models.DateField(_("D.O.B"), auto_now=False, auto_now_add=False, blank=False, null=False)
     profile_image = models.ImageField(_("Profile picture"), upload_to='user/profile/', blank=True)
     webiste_link = models.URLField(_("Websites"), max_length=200, null=True, blank=True)

     def save(self,*args,**kwargs):
          if not self.full_name:
              self.full_name = self.first_name + " " + self.middle_name + " " + self.last_name
          super().save(*args, **kwargs)

然后使用作者,您可以叫用户全名