django rest 框架 - 在不更改现有代码的情况下扩展 auth.user

Question

我正在使用 django rest 框架并构建一个 RESTful api 以从 android 应用程序访问。

我想扩展用户模型并添加电话号码字段 - 但我的模型带有用户模型的外键以及基于令牌的身份验证。

例如：

class Ride(models.Model):
    driver = models.ForeignKey('auth.User')
    destination=models.ForeignKey(Destination)

显然我有默认用户表，正如我所说 - 令牌身份验证 - （在 register 上生成令牌，然后客户端 login - 发送 username:password 并获得 token 登录）。

我的问题是：

如何以代码/模型/视图的最小更改并以最佳方式将此字段添加给用户？我阅读了很多关于这个主题的答案，只是想使用最好的选择。

------------ 编辑 --------------

添加 UserProfile 后，我无法创建用户，代码如下：

serializers.py：

class UserSerializer(serializers.ModelSerializer):

    class Meta:
        model = User
        fields = ('id', 'username', 'first_name', 'last_name', 'password', )
        extra_kwargs = {'password': {'write_only': True}}

    def create(self, validated_data):
        user = User(
            first_name=validated_data['first_name'],
            username=validated_data['username'],
            last_name=validated_data['last_name']
        )

        user.set_password(validated_data['password'])
        user.save()
        user.UserProfile.phone_number = self.initial_data["phone_number"]
        return user

Models.py

class UserProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    phone_number = models.CharField(max_length=15)

@receiver(post_save, sender=User)
def handle_user_save(sender, instance=None, created=False, **kwargs):
    if created:
        UserProfile.objects.create(user=instance)
        Token.objects.create(user=instance)

user.save() 调用 handle_user_save 方法 - 然后我创建了 UserProfile。

问题是 - 在我创建它之后 - 我无法从用户那里更新它的字段。问题出在这一行：

user.UserProfile.phone_number = self.initial_data["phone_number"]

它抛出异常 - “用户没有字段 UserProfile”。

我该怎么做？

谢谢

Answer 1

提到的 OneToOne 字段很可能是指 Django 建议的Extending the Existing User Model 方式。

建议您创建一个新模型，例如 UserProfile，并在其中添加您需要的任何新字段。 UserProfile 将使用 OneToOne 字段链接到 User 模型。

from django.contrib.auth.models import User

class UserProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    new_field = models.CharField(max_length=100)

来自其他模型的所有外键引用仍应指向 Django 的用户模型。当您需要添加的新字段时，您只需从用户那里提取配置文件：

user = Users.objects.get(username='example')
print user.userprofile.new_field