如何在 Django 中将现有 model 与用户 model 链接

Question

我认为这是一个非常简单的问题，但我是 Django 的新手，正在寻找最佳/正确的做事方式。 如果我问愚蠢的问题，请原谅我。

我的 Django 应用程序中有一个名为 Person 的 model。 它仅包含一个人的基本信息，尚未涉及个人数据：

class Person(models.Model):

    birthday = models.DateField(null=True)

    gender = models.CharField(max_length=10)

    height = models.CharField(max_length=3)

    weight = models.CharField(max_length=3)

现在我还想为这个人添加一个注册和登录的功能。这样做的正确方法是什么？ 许多教程似乎建议使用forms.py文件，例如：

from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth.models import User

class RegisterForm(UserCreationForm):
    birthdate = forms.DateField()
    discord_id = forms.CharField(max_length=100, help_text='Discord ID')
    zoom_id = forms.CharField(max_length=100, help_text='Zoom ID')

    class Meta:
        model = User
        fields = ["username", "password1", "password2", "birthdate", "email", "discord_id", "zoom_id"]

我将如何使用/链接我现有的 Person model 与这个内置的用户模型？ 或者我是否需要 Models.py 中的单独Models.py ？ 我是否只是在forms.py中创建像上面一样的用户模型并将我的自定义字段添加到其中？

Answer 1

您可以参考 django 文档扩展用户 model：

https://docs.djangoproject.com/en/3.2/topics/auth/customizing/#extending-the-existing-user-model

有两种方法。 您可以与 model 与 django 用户 model 建立 OneToOne 关系以添加其他字段。 或者，您可以从 django 扩展抽象用户 model 以创建自定义用户。

from django.contrib.auth.models import AbstractUser

    class Person(AbstractUser):
        birthday = models.DateField(null=True)

        gender = models.CharField(max_length=10)

        height = models.CharField(max_length=3)

        weight = models.CharField(max_length=3)

您还需要在设置中标记自定义用户：

AUTH_USER_MODEL = 'myapp.Person'

那么 model 可以按以下形式使用：

from django.contrib.auth.forms import UserCreationForm
from .models import Person

class RegisterForm(UserCreationForm):
    birthdate = forms.DateField()
    discord_id = forms.CharField(max_length=100, help_text='Discord ID')
    zoom_id = forms.CharField(max_length=100, help_text='Zoom ID')

    class Meta:
        model = Person
        fields = ["username", "password1", "password2", "birthdate", "email", "discord_id", "zoom_id"]

或者，您可以执行 OneToOne 关系：

from django.contrib.auth.models import User

class Person(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)

    birthday = models.DateField(null=True)

    gender = models.CharField(max_length=10)

    height = models.CharField(max_length=3)

    weight = models.CharField(max_length=3)

Answer 2

有多种方法可以扩展用户 model 以根据您的要求添加更多自定义字段。它们是代理管理器、OneToOneField、AbstractUser。 我的选择是使用一对一的关系字段，因为它将用户绑定到只有一个配置文件。

from django.contrib.auth.models import User

class foo(models.Model):
   user = OneToOneField(User, on_delete=models.CASCADE)

此外，您需要有 django 信号才能将扩展 foo model 与用户 model 绑定。 这样无论对 foo 进行什么更改，用户都可以收听它。

from django.db.models.signals import post_save
from django.dispatch import reciever

@reciever(sender=User, post_save)
def create_foo(sender, instance, created, **kwargs):
   if created:
      foo.objects.create(user=instance)

@reciever(sender=User, post_save)
def save_foo(sender, instance, **kwargs):
   instance.foo.save()