diff --git a/api/v1/serializers/payment.py b/api/v1/serializers/payment.py index 0ac2d12a..8d50f217 100644 --- a/api/v1/serializers/payment.py +++ b/api/v1/serializers/payment.py @@ -3,7 +3,7 @@ from rest_framework import serializers from apps.payment.models import ( AuthorBalance, Payment, CoursePayment, SchoolPayment, -) + GiftCertificatePayment) from .user import UserSerializer from .course import CourseSerializer @@ -114,6 +114,8 @@ class PaymentSerializer(serializers.ModelSerializer): return CoursePaymentSerializer(instance, context=self.context).to_representation(instance) elif isinstance(instance, SchoolPayment): return SchoolPaymentSerializer(instance, context=self.context).to_representation(instance) + elif isinstance(instance, GiftCertificatePayment): + return GiftCertificatePaymentSerializer(instance, context=self.context).to_representation(instance) class CoursePaymentCreateSerializer(serializers.ModelSerializer): @@ -181,3 +183,20 @@ class SchoolPaymentSerializer(serializers.ModelSerializer): 'created_at', 'update_at', ) + + +class GiftCertificatePaymentSerializer(serializers.ModelSerializer): + user = UserSerializer() + + class Meta: + model = GiftCertificatePayment + fields = BASE_PAYMENT_FIELDS + ( + 'gift_certificate', + ) + read_only_fields = ( + 'id', + 'user', + 'course', + 'created_at', + 'update_at', + )