diff --git a/finance/serializers.py b/finance/serializers.py index 381c227..76f3085 100644 --- a/finance/serializers.py +++ b/finance/serializers.py @@ -6,6 +6,8 @@ from finance.models import Bill, Invoice class BillSerializer(serializers.ModelSerializer): opener = serializers.SerializerMethodField() user = serializers.SerializerMethodField() + course = serializers.SerializerMethodField() + invoices = serializers.SerializerMethodField() class Meta: model = Bill @@ -15,6 +17,14 @@ class BillSerializer(serializers.ModelSerializer): def get_user(self): return self.user.email + @staticmethod + def get_course(self): + return self.course.title + + @staticmethod + def get_invoices(self): + return [InvoiceSerializer(i).data for i in self.invoice_set.all()] + @staticmethod def get_opener(self): return self.opener.email