Source code for payments.api.v1.fields

from rest_framework import serializers



[docs]class PaymentTypeField(serializers.ChoiceField):
    NO_PAYMENT = "no_payment"

    def __init__(self, choices, **kwargs):
        choices = choices + (self.NO_PAYMENT,)
        super().__init__(choices, **kwargs)


[docs]    def get_attribute(self, instance):
        if not instance.payment:
            return self.NO_PAYMENT
        return super().get_attribute(instance)