Question

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

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

例如：

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

显然我有默认用户表，正如我所说 - 令牌身份验证 - （在注册上生成令牌，然后客户端登录 - 发送用户名：密码< / strong>并获取令牌登录信息。

我的问题是：

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

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

添加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字段最有可能是指Extending the Existing User Model的Django建议方式。

建议您创建一个新模型，例如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

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

1 个答案: