Django Rest Framework - “用户”对象没有属性“付款”

Question

我花了很多时间试图理解我遇到的这个错误。

来自models.py

class Company(models.Model):
    name = models.CharField(max_length=100, unique=True, default="")
    email = models.CharField(max_length=100, unique=True, default="")
    password = models.CharField(max_length=100, unique=True, default="")
    bsb = models.IntegerField(default=0)
    account = models.IntegerField(default=0)
    sign_up_date = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name


class Payment(models.Model):
    company = models.ForeignKey(Company, on_delete=models.CASCADE, related_name="payments")
    name = models.CharField(max_length=100, unique=True, default="")
    bsb = models.IntegerField(default=0)
    account = models.IntegerField(default=0)
    created_date = models.DateTimeField(auto_now_add=True)
    paid_date = models.DateTimeField(default=datetime.now() + timedelta(days=36500))
    status = models.IntegerField(default=0)

来自serializers.py

from rest_framework import serializers

from .models import Payment, Company


class PaymentSerializer(serializers.ModelSerializer):
    company = serializers.ReadOnlyField(source='company.name')

    class Meta:
        model = Payment
        fields = ['company', 'name', 'bsb', 'account', 'created_date', 'paid_date', 'status']


class CompanySerializer(serializers.ModelSerializer):
    payments = serializers.PrimaryKeyRelatedField(many=True, queryset=Payment.objects.all())

    class Meta:
        model = Company
        fields = ["name", "email", "password", "payments",
                  "bsb", "account", "sign_up_date"]

如您所见，我在 Payment 类中包含了相关名称作为公司属性的“payments”。但是当我访问 http://localhost:8000/bill_payer/resources/company 时出现以下错误：

AttributeError: 'User' object has no attribute 'payments'

我已经通过 manage.py shell 验证了我的 Company 类确实有支付类。

有什么想法吗？我是新的。这是views.py，以防它很重要：

class PaymentList(APIView):
    permission_classes = [permissions.IsAuthenticatedOrReadOnly]

    def get(self, request):
        payments = Payment.objects.all()
        serializer = PaymentSerializer(payments, many=True)
        return Response(serializer.data)

    def perform_create(self, serializer):
        serializer.save(company=self.request.user)


class PaymentDetail(APIView):
    permission_classes = [permissions.IsAuthenticatedOrReadOnly]

    def get_object(self, pk):
        try:
            return Payment.objects.get(pk=pk)
        except Payment.DoesNotExist:
            raise Http404

    def get(self, request, pk):
        payment = self.get_object(pk)
        serializer = PaymentSerializer(payment)
        return Response(serializer.data)

    def put(self, request, pk):
        payment = self.get_object(pk)
        serializer = PaymentSerializer(payment, data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

    def delete(self, request, pk):
        payment = self.get_object(pk)
        payment.delete()
        return Response(status=status.HTTP_204_NO_CONTENT)


class CompanyList(generics.ListAPIView):
    queryset = User.objects.all()
    serializer_class = CompanySerializer


class CompanyDetail(generics.RetrieveAPIView):
    queryset = User.objects.all()
    serializer_class = CompanySerializer

Answer 1

您收到错误是因为您尝试将 User 对象用于 CompanySerializer，它需要 Company 模型对象。如果您按以下方式修复它，它将起作用。

class CompanyList(generics.ListAPIView):
    queryset = Company.objects.all()
    serializer_class = CompanySerializer


class CompanyDetail(generics.RetrieveAPIView):
    queryset = Company.objects.all()
    serializer_class = CompanySerializer

Answer 2

请改一下

class CompanyList(generics.ListAPIView):
    queryset = User.objects.all()
    serializer_class = CompanySerializer


class CompanyDetail(generics.RetrieveAPIView):
    queryset = User.objects.all()
    serializer_class = CompanySerializer

到

class CompanyList(generics.ListAPIView):
    queryset = Company.objects.all()
    serializer_class = CompanySerializer


class CompanyDetail(generics.RetrieveAPIView):
    queryset = Company.objects.all()
    serializer_class = CompanySerializer

你在 perform_create 方法中也做错了。

class PaymentList(APIView):
    permission_classes = [permissions.IsAuthenticatedOrReadOnly]

    def get(self, request):
        payments = Payment.objects.all()
        serializer = PaymentSerializer(payments, many=True)
        return Response(serializer.data)

    def perform_create(self, serializer):
        serializer.save(company=self.request.user)

不应该

serializer.save(company=self.request.user)

因为它不是公司对象。 self.request.user 是一个用户对象。并且您正在尝试将用户对象保存到公司对象。这是错的。 尝试使用

serializer.save()
or
serializer.save(company=serializer.data.get("company"))