refactor

backend/igny8_core/auth/serializers.py

@@ -27,8 +27,8 @@ class SubscriptionSerializer(serializers.ModelSerializer):
         model = Subscription
         fields = [
             'id', 'account', 'account_name', 'account_slug',
-            'stripe_subscription_id', 'status',
-            'current_period_start', 'current_period_end',
+            'stripe_subscription_id', 'payment_method', 'external_payment_id',
+            'status', 'current_period_start', 'current_period_end',
             'cancel_at_period_end',
             'created_at', 'updated_at'
         ]
@@ -48,7 +48,11 @@ class AccountSerializer(serializers.ModelSerializer):
     
     class Meta:
         model = Account
-        fields = ['id', 'name', 'slug', 'owner', 'plan', 'plan_id', 'credits', 'status', 'subscription', 'created_at']
+        fields = [
+            'id', 'name', 'slug', 'owner', 'plan', 'plan_id',
+            'credits', 'status', 'payment_method',
+            'subscription', 'created_at'
+        ]
         read_only_fields = ['owner', 'created_at']
 
 
@@ -260,6 +264,12 @@ class RegisterSerializer(serializers.Serializer):
         allow_null=True,
         default=None
     )
+    plan_slug = serializers.CharField(max_length=50, required=False)
+    payment_method = serializers.ChoiceField(
+        choices=['stripe', 'paypal', 'bank_transfer'],
+        default='bank_transfer',
+        required=False
+    )
     
     def validate(self, attrs):
         if attrs['password'] != attrs['password_confirm']: