获得Django的用户会员资格

Question

我需要获得请求用户是否在我的网站中有会员资格，但是我有一个错误可以帮助我吗？ 用户模型模型是

    class UserMembership(models.Model):
    user = models.ForeignKey('accounts.User', on_delete=models.CASCADE, related_name='user_membership')
    membership = models.ForeignKey(Membership, on_delete=models.DO_NOTHING, null=True, related_name='membership')

    def __str__(self):
        return self.user.email

序列化器IS：

    class UserMemberShipSerializer(serializers.ModelSerializer):
    class Meta:
        model = UserMembership
        fields = ['id', 'user', 'membership']

并且视图是，

    class MembershipView(viewsets.ModelViewSet):
    model = UserMembership
    serializer_class = UserMemberShipSerializer

    def get_user_membership(self):
        user_membership_qs = UserMembership.objects.get(user=self.request.user)
        print(user_membership_qs)
        if user_membership_qs.exists():
            return user_membership_qs.first()

    def get_queryset(self):
        current_membership = self.get_user_membership(self.request)
        return current_membership

并且错误在此行中，

            current_membership = self.get_user_membership(self.request)

错误是 'current_membership = self.get_user_membership（self.request） TypeError：get_user_membership（）采用1个位置参数，但给出了2个'

Answer 1

从get_user_membership函数的输入中删除“ self.request”：

current_membership = self.get_user_membership()

或将“请求”添加为arg to函数：

    def get_user_membership(self,request):
        user_membership_qs = UserMembership.objects.get(user=request.user)
        print(user_membership_qs)
        if user_membership_qs.exists():
            return user_membership_qs.first()