uncloud-mravi/uncloud_pay/serializers.py

from django.contrib.auth import get_user_model
from rest_framework import serializers
from uncloud_auth.serializers import UserSerializer
from django.utils.translation import gettext_lazy as _
from stripe.error import CardError

from .models import *
import uncloud_pay.stripe as uncloud_stripe
from uncloud import AMOUNT_DECIMALS, AMOUNT_MAX_DIGITS

###
# 2020-12 Checked code

class StripeCreditCardSerializer(serializers.HyperlinkedModelSerializer):
    class Meta:
        model = StripeCreditCard
        exclude = [ "card_id", "owner" ]
        read_only_fields = [ "last4", "brand", "expiry_date" ]

class PaymentSerializer(serializers.ModelSerializer):
    owner = serializers.HiddenField(default=serializers.CurrentUserDefault())

    class Meta:
        model = Payment
        fields = '__all__'
        read_only_fields = [ "external_reference", "source", "timestamp" ]

    def validate(self, data):
        data["source"] = "stripe"
        return data
        
    def create(self, validated_data):
        try:
            if validated_data['type'] == 'deposit':
                return Payment.deposit(validated_data['owner'], validated_data['amount'], validated_data['source'], currency=validated_data['currency'], notes=validated_data['notes'])
            else:
                return Payment.objects.create(**validated_data)
        except CardError as err:
            raise serializers.ValidationError(err.user_message)

class BalanceSerializer(serializers.Serializer):
    balance = serializers.DecimalField(max_digits=AMOUNT_MAX_DIGITS, decimal_places=AMOUNT_DECIMALS)

class BillingAddressSerializer(serializers.ModelSerializer):
    class Meta:
        model = BillingAddress
        exclude = [ "owner" ]


class VATRateSerializer(serializers.ModelSerializer):

    class Meta:
        model = VATRate
        fields = '__all__'


################################################################################
# Unchecked code


###
# Orders & Products.

class OrderSerializer(serializers.ModelSerializer):
    owner = serializers.PrimaryKeyRelatedField(queryset=get_user_model().objects.all())

    def __init__(self, *args, **kwargs):
        # Don't pass the 'fields' arg up to the superclass
        admin = kwargs.pop('admin', None)

        # Instantiate the superclass normally
        super(OrderSerializer, self).__init__(*args, **kwargs)

        # Only allows owner in admin mode.
        if not admin:
            self.fields.pop('owner')

    def create(self, validated_data):
        billing_address = BillingAddress.get_preferred_address_for(validated_data["owner"])
        instance = Order(billing_address=billing_address, **validated_data)
        instance.save()

        return instance

    def validate_owner(self, value):
        if BillingAddress.get_preferred_address_for(value) == None:
            raise serializers.ValidationError("Owner does not have a valid billing address.")

        return value

    class Meta:
        model = Order
        read_only_fields = ['replaced_by', 'depends_on']
        fields = ['owner', 'description', 'creation_date', 'starting_date', 'ending_date',
                'recurring_period', 'recurring_price', 'one_time_price', 
                'config', 'pricing_plan', 'should_be_billed'] + read_only_fields


###
# Bills

# TODO: remove magic numbers for decimal fields
class BillRecordSerializer(serializers.Serializer):
    order = serializers.HyperlinkedRelatedField(
            view_name='order-detail',
            read_only=True)
    description = serializers.CharField()
    one_time_price = serializers.DecimalField(AMOUNT_MAX_DIGITS, AMOUNT_DECIMALS)
    recurring_price = serializers.DecimalField(AMOUNT_MAX_DIGITS, AMOUNT_DECIMALS)
#    recurring_period = serializers.ChoiceField()
    recurring_count = serializers.DecimalField(AMOUNT_MAX_DIGITS, AMOUNT_DECIMALS)
    vat_rate = serializers.DecimalField(AMOUNT_MAX_DIGITS, AMOUNT_DECIMALS)
    vat_amount = serializers.DecimalField(AMOUNT_MAX_DIGITS, AMOUNT_DECIMALS)
    amount = serializers.DecimalField(AMOUNT_MAX_DIGITS, AMOUNT_DECIMALS)
    total = serializers.DecimalField(AMOUNT_MAX_DIGITS, AMOUNT_DECIMALS)

class BillSerializer(serializers.ModelSerializer):
    billing_address = BillingAddressSerializer(read_only=True)
    records = BillRecordSerializer(many=True, read_only=True)

    class Meta:
        model = Bill
        fields = ['owner', 'sum', 'vat_rate',
                'due_date', 'creation_date', 'starting_date', 'ending_date',
                'records', 'is_closed', 'billing_address']

# We do not want users to mutate the country / VAT number of an address, as it
# will change VAT on existing bills.
class UpdateBillingAddressSerializer(serializers.ModelSerializer):
    class Meta:
        model = BillingAddress
        fields = ['street', 'city', 'postal_code']