Move things around for readability in uncloud_pay models and serializer

uncloud/uncloud_pay/serializers.py

@@ -9,36 +9,62 @@ from uncloud_vm.models import VMProduct
 
 import uncloud_pay.stripe as stripe
 
-# TODO: remove magic numbers for decimal fields
-class BillRecordSerializer(serializers.Serializer):
-    order = serializers.CharField()
-    description = serializers.CharField()
-    recurring_period = serializers.CharField()
-    recurring_price = serializers.DecimalField(max_digits=10, decimal_places=2)
-    amount = serializers.DecimalField(max_digits=10, decimal_places=2)
+###
+# Users.
 
-class BillSerializer(serializers.ModelSerializer):
-    records = BillRecordSerializer(many=True, read_only=True)
+class UserSerializer(serializers.ModelSerializer):
     class Meta:
-        model = Bill
-        fields = ['owner', 'total', 'due_date', 'creation_date',
-                'starting_date', 'ending_date', 'records', 'final']
+        model = get_user_model()
+        fields = ['username', 'email', 'balance']
+
+    # Display current 'balance'
+    balance = serializers.SerializerMethodField('get_balance')
+    def __sum_balance(self, entries):
+        return reduce(lambda acc, entry: acc + entry.amount, entries, 0)
+
+    def get_balance(self, user):
+        return get_balance_for(user)
+
+###
+# Payments and Payment Methods.
 
 class PaymentSerializer(serializers.ModelSerializer):
     class Meta:
         model = Payment
         fields = ['owner', 'amount', 'source', 'timestamp']
 
+class PaymentMethodSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = PaymentMethod
+        fields = ['source', 'description', 'primary']
+
 class CreditCardSerializer(serializers.Serializer):
     number = serializers.IntegerField()
     exp_month = serializers.IntegerField()
     exp_year = serializers.IntegerField()
     cvc = serializers.IntegerField()
 
-class PaymentMethodSerializer(serializers.ModelSerializer):
+class CreatePaymentMethodSerializer(serializers.ModelSerializer):
+    credit_card = CreditCardSerializer()
+
     class Meta:
         model = PaymentMethod
-        fields = ['source', 'description', 'primary']
+        fields = ['source', 'description', 'primary', 'credit_card']
+
+    def create(self, validated_data):
+        credit_card = stripe.CreditCard(**validated_data.pop('credit_card'))
+        user = self.context['request'].user
+        customer = stripe.create_customer(user.username, user.email)
+        # TODO check customer error
+        customer_id = customer['response_object']['id']
+        stripe_card = stripe.create_card(customer_id, credit_card)
+        # TODO: check credit card error
+        validated_data['stripe_card_id'] = stripe_card['response_object']['id']
+class CreditCardSerializer(serializers.Serializer):
+    number = serializers.IntegerField()
+    exp_month = serializers.IntegerField()
+    exp_year = serializers.IntegerField()
+    cvc = serializers.IntegerField()
 
 class CreatePaymentMethodSerializer(serializers.ModelSerializer):
     credit_card = CreditCardSerializer()
@@ -58,15 +84,36 @@ class CreatePaymentMethodSerializer(serializers.ModelSerializer):
         validated_data['stripe_card_id'] = stripe_card['response_object']['id']
         payment_method = PaymentMethod.objects.create(**validated_data)
         return payment_method
+        payment_method = PaymentMethod.objects.create(**validated_data)
+        return payment_method
 
-class ProductSerializer(serializers.Serializer):
-    vms = VMProductSerializer(many=True, read_only=True)
+###
+# Bills
+
+# TODO: remove magic numbers for decimal fields
+class BillRecordSerializer(serializers.Serializer):
+    order = serializers.CharField()
+    description = serializers.CharField()
+    recurring_period = serializers.CharField()
+    recurring_price = serializers.DecimalField(max_digits=10, decimal_places=2)
+    amount = serializers.DecimalField(max_digits=10, decimal_places=2)
+
+class BillSerializer(serializers.ModelSerializer):
+    records = BillRecordSerializer(many=True, read_only=True)
+    class Meta:
+        model = Bill
+        fields = ['owner', 'total', 'due_date', 'creation_date',
+                'starting_date', 'ending_date', 'records', 'final']
+
+###
+# Orders & Products.
 
 class OrderRecordSerializer(serializers.ModelSerializer):
     class Meta:
         model = OrderRecord
         fields  = ['setup_fee', 'recurring_price', 'description']
 
+
 class OrderSerializer(serializers.ModelSerializer):
     records = OrderRecordSerializer(many=True, read_only=True)
     class Meta:
@@ -74,15 +121,5 @@ class OrderSerializer(serializers.ModelSerializer):
         fields = ['uuid', 'creation_date', 'starting_date', 'ending_date',
                 'bill', 'recurring_period', 'records', 'recurring_price', 'setup_fee']
 
-class UserSerializer(serializers.ModelSerializer):
-    class Meta:
-        model = get_user_model()
-        fields = ['username', 'email', 'balance']
-
-    # Display current 'balance'
-    balance = serializers.SerializerMethodField('get_balance')
-    def __sum_balance(self, entries):
-        return reduce(lambda acc, entry: acc + entry.amount, entries, 0)
-
-    def get_balance(self, user):
-        return get_balance_for(user)
+class ProductSerializer(serializers.Serializer):
+    vms = VMProductSerializer(many=True, read_only=True)