Created command to fiz inactive subscription with paid payment status

renzon · renzon · commit be18554be550 · 2021-06-06T13:02:18.000-03:00
close #3770
diff --git a/pythonpro/memberkit/management/commands/fix_inactive_subscriptions.py b/pythonpro/memberkit/management/commands/fix_inactive_subscriptions.py
@@ -0,0 +1,36 @@
+from django.core.management.base import BaseCommand
+from django.db.models import OuterRef, Subquery
+from django_pagarme import facade as pagarme_facade
+from django_pagarme.models import PagarmePayment, PagarmeNotification
+
+from pythonpro.memberkit.models import Subscription
+
+
+class Command(BaseCommand):
+    help = 'Ativa assinatura inativas que possuem última notificação de pagamento como paga'
+
+    def add_arguments(self, parser):
+        pass
+
+    def handle(self, *args, **options):
+        last_notification = PagarmeNotification.objects.filter(payment=OuterRef('pk')).order_by('-creation')
+        qs = PagarmePayment.objects.filter(
+            notifications__status=pagarme_facade.PAID,
+            subscription__isnull=False
+        ).only(
+            'subscription'
+        ).select_related(
+            'subscription'
+        ).select_related(
+            'subscription__subscriber'
+        ).filter(
+            subscription__status=Subscription.Status.INACTIVE
+        ).annotate(
+            payment_status=Subquery(last_notification.values('status')[:1])
+        ).filter(payment_status=pagarme_facade.PAID)
+        for payment in qs:
+            subscription = payment.subscription
+            self.stdout.write(self.style.SUCCESS(f'Processando {subscription}'))
+            subscription.observation += '\n\nAtivada via comando automático do servidor.'
+            subscription.status = Subscription.Status.ACTIVE
+            subscription.save()
diff --git a/pythonpro/memberkit/models.py b/pythonpro/memberkit/models.py
@@ -43,3 +43,6 @@ class Status(models.TextChoices):
     responsible = models.ForeignKey(get_user_model(), on_delete=models.DO_NOTHING, null=True,
                                     related_name='created_subscriptions')
     observation = models.TextField(verbose_name='Observação', blank=True, default='')
+
+    def __str__(self):
+        return f'Assinatura: {self.id} de {self.subscriber}'
diff --git a/pythonpro/memberkit/tests/test_commands.py b/pythonpro/memberkit/tests/test_commands.py
@@ -25,4 +25,53 @@ def test_synchronize_paid_subscriptions(db):
     baker.make(PagarmeNotification, payment=payment, status=pagarme_facade.PAID)
     baker.make(PagarmeNotification, payment=payment, status=pagarme_facade.REFUNDED)
     management.call_command('synchronize_paid_subscriptions')
-    assert 2 == Subscription.objects.count()
+    assert 2 == Subscription.objects.filter(status=Subscription.Status.INACTIVE).count()
+
+
+def test_fix_inactive_subscriptions(db):
+    payment_config = baker.make(PagarmeItemConfig)
+    inactive_with_paid_payment = _make_subscriptions_with_payment(payment_config, pagarme_facade.PAID)
+    unpaid_statuses = set(pagarme_facade._impossible_states)
+    unpaid_statuses.discard(pagarme_facade.PAID)
+    for status in unpaid_statuses:
+        # only one will have PAID status
+        _make_subscriptions_with_payment(payment_config, status)
+
+    # Make one refunded Subscription
+    payment = baker.make(PagarmePayment, user=baker.make(User))
+    payment.items.set([payment_config])
+    baker.make(PagarmeNotification, payment=payment, status=pagarme_facade.PAID)
+    baker.make(PagarmeNotification, payment=payment, status=pagarme_facade.REFUNDED)
+    refunded_subscription = baker.make(Subscription, payment=payment, status=Subscription.Status.INACTIVE)
+
+    # Make one subscriptions with no payment at all
+    baker.make(Subscription, status=Subscription.Status.INACTIVE)
+    active_without_payment = baker.make(Subscription, status=Subscription.Status.ACTIVE)
+
+    active_with_paid_payment = _make_subscriptions_with_payment(payment_config, pagarme_facade.PAID)
+    active_with_paid_payment.status = Subscription.Status.ACTIVE
+    active_with_paid_payment.save()
+
+    previous_subscriptions = Subscription.objects.count()
+    assert 2 == Subscription.objects.filter(status=Subscription.Status.ACTIVE).count()
+
+    management.call_command('fix_inactive_subscriptions')
+
+    assert previous_subscriptions == Subscription.objects.count()
+    assert 3 == Subscription.objects.filter(status=Subscription.Status.ACTIVE).count()
+    assert not Subscription.objects.filter(status=Subscription.Status.ACTIVE, id=refunded_subscription.id).exists()
+
+    appended_observation_message = '\n\nAtivada via comando automático do servidor.'
+    active_without_payment.refresh_from_db()
+    assert not active_without_payment.observation.endswith(appended_observation_message)
+    active_with_paid_payment.refresh_from_db()
+    assert not active_with_paid_payment.observation.endswith(appended_observation_message)
+    inactive_with_paid_payment.refresh_from_db()
+    assert inactive_with_paid_payment.observation.endswith(appended_observation_message)
+
+
+def _make_subscriptions_with_payment(payment_config, status):
+    payment = baker.make(PagarmePayment, user=baker.make(User))
+    payment.items.set([payment_config])
+    baker.make(PagarmeNotification, payment=payment, status=status)
+    return baker.make(Subscription, payment=payment, status=Subscription.Status.INACTIVE)
