uncloud-mravi/uncloud_pay/serializers.py

120 lines
4.3 KiB
Python
Raw Normal View History

2020-02-27 11:38:04 +00:00
from django.contrib.auth import get_user_model
from rest_framework import serializers
2020-05-07 10:05:26 +00:00
from uncloud_auth.serializers import UserSerializer
from django.utils.translation import gettext_lazy as _
from .models import *
2020-12-29 00:43:33 +00:00
import uncloud_pay.stripe as uncloud_stripe
2021-01-01 11:41:54 +00:00
from uncloud import AMOUNT_DECIMALS, AMOUNT_MAX_DIGITS
2020-03-02 21:26:40 +00:00
2020-12-25 16:29:17 +00:00
###
2020-12-28 22:35:34 +00:00
# 2020-12 Checked code
2021-01-01 12:25:52 +00:00
class StripeCreditCardSerializer(serializers.HyperlinkedModelSerializer):
2020-12-28 22:35:34 +00:00
class Meta:
model = StripeCreditCard
2020-12-29 00:43:33 +00:00
exclude = [ "card_id", "owner" ]
2020-12-28 22:35:34 +00:00
read_only_fields = [ "last4", "brand", "expiry_date" ]
2020-12-29 00:43:33 +00:00
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):
payment_intent = uncloud_stripe.charge_customer(data['owner'],
data['amount'])
data["external_reference"] = payment_intent["id"]
data["source"] = "stripe"
return data
2020-12-25 16:29:17 +00:00
2021-01-01 11:41:54 +00:00
class BalanceSerializer(serializers.Serializer):
balance = serializers.DecimalField(max_digits=AMOUNT_MAX_DIGITS, decimal_places=AMOUNT_DECIMALS)
2021-01-17 14:53:30 +00:00
class BillingAddressSerializer(serializers.ModelSerializer):
class Meta:
model = BillingAddress
exclude = [ "owner" ]
2020-12-25 16:29:17 +00:00
################################################################################
# Unchecked code
2020-03-06 08:32:25 +00:00
###
# Orders & Products.
class OrderSerializer(serializers.ModelSerializer):
2020-05-07 10:05:26 +00:00
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
2020-03-06 08:32:25 +00:00
class Meta:
model = Order
2020-05-08 14:47:32 +00:00
read_only_fields = ['replaced_by', 'depends_on']
2020-05-07 10:08:18 +00:00
fields = ['uuid', 'owner', 'description', 'creation_date', 'starting_date', 'ending_date',
2020-05-08 14:47:32 +00:00
'bill', 'recurring_period', 'recurring_price', 'one_time_price'] + read_only_fields
2020-03-06 08:32:25 +00:00
###
# Bills
# TODO: remove magic numbers for decimal fields
class BillRecordSerializer(serializers.Serializer):
2020-03-06 08:32:25 +00:00
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
2020-05-07 13:38:49 +00:00
fields = ['uuid', 'reference', 'owner', 'amount', 'vat_amount', 'total',
'due_date', 'creation_date', 'starting_date', 'ending_date',
'records', 'final', 'billing_address']
2020-04-15 13:17:38 +00:00
# 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 = ['uuid', 'street', 'city', 'postal_code']