Set one payment method as primary, allow updates

uncloud_django_based/uncloud/uncloud_pay/models.py

@@ -4,9 +4,7 @@ from django.contrib.auth import get_user_model
 from django.core.validators import MinValueValidator
 from django.utils.translation import gettext_lazy as _
 from django.utils import timezone
-from django.dispatch import receiver
 from django.core.exceptions import ObjectDoesNotExist
-import django.db.models.signals as signals
 
 import uuid
 from functools import reduce
@@ -106,7 +104,7 @@ class PaymentMethod(models.Model):
                 ),
             default='stripe')
     description = models.TextField()
-    primary = models.BooleanField(default=True)
+    primary = models.BooleanField(default=False, editable=False)
 
     # Only used for "Stripe" source
     stripe_payment_method_id = models.CharField(max_length=32, blank=True, null=True)
@@ -149,22 +147,24 @@ class PaymentMethod(models.Model):
         else:
             raise Exception('This payment method is unsupported/cannot be charged.')
 
+    def set_as_primary_for(self, user):
+        methods = PaymentMethod.objects.filter(owner=user, primary=True)
+        for method in methods:
+            print(method)
+            method.primary = False
+            method.save()
+
+        self.primary = True
+        self.save()
 
     def get_primary_for(user):
         methods = PaymentMethod.objects.filter(owner=user)
         for method in methods:
-            # Do we want to do something with non-primary method?
-            if method.active and method.primary:
+            if method.primary:
                 return method
 
         return None
 
-    class Meta:
-        # TODO: limit to one primary method per user.
-        # unique_together is no good since it won't allow more than one
-        # non-primary method.
-        pass
-
 ###
 # Bills.