Merge 458449b into 4691632

aa_stripe/management/commands/sync_all_cards_for_all_customers.py

@@ -0,0 +1,95 @@
+# -*- coding: utf-8 -*-
+from time import sleep
+
+import stripe
+from django.conf import settings
+from django.core.management.base import BaseCommand
+
+from aa_stripe.models import StripeCard, StripeCustomer
+from aa_stripe.settings import stripe_settings
+
+
+class Command(BaseCommand):
+    help = "Sync all cards for all customers with Stripe API"
+
+    def add_arguments(self, parser):
+        min_customer_id = StripeCustomer.objects.order_by("id").first().id
+        max_customer_id = StripeCustomer.objects.order_by("-id").first().id
+        parser.add_argument(
+            '--max_customer_id',
+            nargs='?',
+            const=max_customer_id,
+            default=max_customer_id,
+            type=int,
+            dest='max_id',
+            help='Id of the last customer for whom to update cards',
+        )
+        parser.add_argument(
+            '--min_customer_id',
+            nargs='?',
+            const=min_customer_id,
+            default=min_customer_id,
+            type=int,
+            dest='min_id',
+            help='Id of the first customer for whom to update cards',
+        )
+
+    def handle(self, *args, **options):
+        stripe.api_key = stripe_settings.API_KEY
+
+        counts = {"created": 0, "updated": 0, "deleted": 0}
+        processed_users_set = set()
+
+        for customer in StripeCustomer.objects.filter(
+                is_active=True, id__range=(options['min_id'], options['max_id'])).order_by("-id"):
+            if customer.user.id in processed_users_set:
+                continue
+
+            customer_from_stripe = stripe.Customer.retrieve(customer.stripe_customer_id)
+            actual_cards = customer_from_stripe.sources.all(object="card")
+            actual_cards_map = {c.id: c for c in actual_cards}
+            actual_cards_set = set(actual_cards_map)
+            our_cards = StripeCard.objects.filter(customer=customer)
+            our_cards_set = set(our_cards.values_list('stripe_card_id', flat=True))
+            our_deleted_cards = StripeCard.objects.deleted().filter(customer=customer)
+            our_deleted_cards_set = set(our_deleted_cards.values_list('stripe_card_id', flat=True))
+
+            stripe_deleted_cards = our_cards_set - actual_cards_set
+            for card_id in stripe_deleted_cards:
+                card = our_cards.get(stripe_card_id=card_id)
+                card.is_deleted = True
+                card.save()
+
+            undelete_cards = actual_cards_set & our_deleted_cards_set
+            for card_id in undelete_cards:
+                card = our_deleted_cards.get(stripe_card_id=card_id)
+                card.is_deleted = False
+                card.save()
+
+            update_cards = our_cards_set & actual_cards_set
+            for card_id in update_cards:
+                card = our_cards.get(stripe_card_id=card_id)
+                card.update_from_stripe_card(actual_cards_map[card_id])
+
+            created_cards = actual_cards_set - (our_cards_set | our_deleted_cards_set)
+            for card_id in created_cards:
+                card = StripeCard(customer=customer)
+                card.update_from_stripe_card(actual_cards_map[card_id])
+
+            stripe_defaut_source = customer_from_stripe.default_source
+            our_defaut_source = customer.default_card.stripe_card_id
+            if stripe_defaut_source != our_defaut_source:
+                customer.default_card = StripeCard.objects.get(stripe_card_id=stripe_defaut_source)
+                customer.save()
+
+            counts["created"] += len(created_cards)
+            counts["deleted"] += len(stripe_deleted_cards)
+            counts["updated"] += len(update_cards) + len(undelete_cards)
+            processed_users_set.add(customer.user.id)
+
+            print("Processed customer with id: {}".format(customer.id))
+            if not settings.TESTING:
+                sleep(0.25)
+
+        if options.get("verbosity") > 1:
+            print("Cards created: {created}, updated: {updated}, deleted: {deleted}".format(**counts))

aa_stripe/models.py

@@ -108,6 +108,13 @@ def _retrieve_from_stripe(self, set_deleted=False):
             if set_deleted:
                 self.is_deleted = True
 
+    def _set_fields_from_stripe_object(self, stripe_object):
+        self.stripe_card_id = stripe_object["id"]
+        self.last4 = stripe_object["last4"]
+        self.exp_month = stripe_object["exp_month"]
+        self.exp_year = stripe_object["exp_year"]
+        self.stripe_response = stripe_object
+
     def create_at_stripe(self):
         if self.is_created_at_stripe:
             raise StripeMethodNotAllowed()
@@ -116,11 +123,10 @@ def create_at_stripe(self):
         customer = stripe.Customer.retrieve(self.customer.stripe_customer_id)
         card = customer.sources.create(source=self.stripe_js_response["source"])
 
-        self.stripe_card_id = card["id"]
-        self.last4 = card["last4"]
-        self.exp_month = card["exp_month"]
-        self.exp_year = card["exp_year"]
-        self.stripe_response = card
+        return self.update_from_stripe_card(card)
+
+    def update_from_stripe_card(self, card):
+        self._set_fields_from_stripe_object(card)
         self.is_created_at_stripe = True
         self.save()
         return self