Django rest框架根据当前userId/token获取数据

Question

在我的模型中，我有一个家庭表，用户可以在其中输入他们的家庭详细信息并存储他们的信息。但是如何根据当前 userId 获取/查询所有家庭成员？

例如：userId=1 添加了 2 个家庭成员，母亲和父亲。如何根据当前用户的 userId 查询得到这 2 个家庭成员？

这是我的代码：

模型

class MyUser(AbstractUser):
    userId = models.AutoField(primary_key=True)
    gender = models.CharField(max_length=6, blank=True, null=True)
    nric = models.CharField(max_length=40, blank=True, null=True)
    birthday = models.DateField(blank=True, null=True)
    birthTime = models.TimeField(blank=True, null=True)

class Family(models.Model):
    userId = models.ForeignKey(MyUser)
    relationship = models.CharField(max_length=100)
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    gender = models.CharField(max_length=6, blank=True, null=True)

序列化器

class MyUserSerializer(serializers.ModelSerializer):
    valid_time_formats = ['%H:%M', '%I:%M%p', '%I:%M %p']
    birthTime = serializers.TimeField(format='%I:%M %p', input_formats=valid_time_formats, allow_null=True, required=False)

    class Meta:
        model = MyUser
        fields = ['userId', 'username', 'email', 'first_name', 'last_name', 'gender', 'nric', 'birthday', 'birthTime']
        read_only_fields = ('userId',)
        extra_kwargs = {"password": {"write_only": True}}

    def update(self, instance, validated_data):
        for attr, value in validated_data.items():
            if attr == 'password':
                instance.set_password(value)
            else:
                setattr(instance, attr, value)
        instance.save()
        return instance


class FamilySerializer(serializers.ModelSerializer):

    class Meta:
        model = Family
        fields = ('id', 'userId', 'first_name', 'last_name', 'gender', 'relationship')

观看次数

class MyUserViewSet(viewsets.ModelViewSet):
    permission_classes = [AllowAny]
    queryset = MyUser.objects.all()
    serializer_class = MyUserSerializer
    filter_backends = (filters.SearchFilter,)
    search_fields = ('=userId', 'username', 'email', 'first_name', 'last_name')


class FamilyViewSet(viewsets.ModelViewSet):
    permission_classes = [AllowAny]
    queryset = Family.objects.all()
    serializer_class = FamilySerializer


def get_object(self, *args, **kwargs):
    return self.request.user

def perform_update(self, serializer):
    super(CurrentFamilyView, self).perform_update(serializer)
    user = serializer.instance
    if settings.SEND_ACTIVATION_EMAIL and not user.is_active:
        context = {'user': user}
        to = [MyUser(user)]
        email.ActivationEmail(self.request, context).send(to)

谁能帮我获取当前用户的家人

Answer 1

你可以试试这个：

class Family(models.Model):
    userId = models.ForeignKey(MyUser, related_name='user_family')
    relationship = models.CharField(max_length=100)
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    gender = models.CharField(max_length=6, blank=True, null=True)

serializers.py

class FamilySerializer(serializers.ModelSerializer):

    class Meta:
        model = Family
        fields = ('id', 'first_name', 'last_name', 'gender', 'relationship')


class MyUserSerializer(serializers.ModelSerializer):

    valid_time_formats = ['%H:%M', '%I:%M%p', '%I:%M %p']
    birthTime = serializers.TimeField(format='%I:%M %p', input_formats=valid_time_formats, allow_null=True, required=False)
    my_family = FamilySerializer(many=True, source='user_family')

    class Meta:
        model = MyUser
        fields = ('userId', 'username', 'email', 'first_name', 'last_name', 'gender', 'nric', 'birthday', 'birthTime', 'my_family')
        read_only_fields = ('userId',)
        extra_kwargs = {"password": {"write_only": True}}

MyUserSerializer 现在将显示userid 相关的家庭成员。如果要获取当前用户家庭成员，则需要覆盖get_queryset 方法。

您的views.py：

class MyUserViewSet(viewsets.ModelViewSet):

    serializer_class = MyUserSerializer
    get_queryset(self):
        return  MyUser.objects.get(id=self.request.user.id)

现在视图将返回当前用户的家庭成员。

更新

1. 在Family 模型中添加了related_name。
2. 在MyUserSerializer 中添加了source

Answer 2

假设我们有一个 id=1

的 用户

usr = MyUser.objects.get(id=1)
usr_famly = usr.family_set.values()

usr_famly 将拥有与user with id=1 相关的所有家庭数据