forked from uncloud/uncloud
132 lines
4.7 KiB
Python
132 lines
4.7 KiB
Python
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']
|