Make a basic version of subscriber_sync

Author: diettam

djpaddle/management/commands/djpaddle_sync_subscribers_from_paddle.py

@@ -0,0 +1,18 @@
+"""
+sync_subscribers_from_paddle command.
+"""
+from django.core.management.base import BaseCommand
+
+from ...models import Subscription
+
+
+class Command(BaseCommand):
+    """Sync subscribers from paddle."""
+
+    help = "Sync subscribers from paddle."
+
+    def handle(self, *args, **options):
+        """Call sync_from_paddle_data for each subscriber returned by api_list."""
+        for sub_data in Subscription.api_list():
+            sub = Subscription.sync_from_paddle_data(sub_data)
+            print("Synchronized {0}".format(str(sub)))

djpaddle/models.py

@@ -2,6 +2,7 @@
 from django.db.models.signals import post_save
 from django.utils.translation import gettext_lazy as _
 from django.dispatch import receiver
+from django.utils import timezone
 
 from . import settings, signals, api
 from .fields import PaddleCurrencyCodeField
@@ -123,6 +124,7 @@ class Subscription(PaddleBaseModel):
         (STATUS_PAUSED, _("paused")),
         (STATUS_DELETED, _("deleted")),
     )
+    PADDLE_URI_LIST = 'subscription/users'
 
     id = models.CharField(max_length=32, primary_key=True)
     subscriber = models.ForeignKey(
@@ -151,6 +153,51 @@ class Subscription(PaddleBaseModel):
     class Meta:
         ordering = ["created_at"]
 
+    @classmethod
+    def api_list(cls):
+        return api.retrieve(uri=cls.PADDLE_URI_LIST)
+
+    @classmethod
+    def sync_from_paddle_data(cls, data):
+        pk = data.get('subscription_id', None)
+        # First, find and drop current sub with this data.
+        cls.objects.filter(pk=pk).delete()
+        kwargs = {}
+
+        try:
+            kwargs['subscriber'] = settings.get_subscriber_model().objects.get(
+                email=data["user_email"]
+            )
+        except settings.get_subscriber_model().DoesNotExist:
+            pass
+
+        try:
+            kwargs['plan'] = Plan.objects.get(pk=data.get("plan_id"))
+        except Plan.DoesNotExist:
+            print("Skipping, plan not found.")
+            return
+
+        # Now, create object with this pk.
+        sub = Subscription.objects.create(
+            id=pk,
+            cancel_url=data.get("cancel_url"),
+            checkout_id="", # ???
+            currency=data.get('last_payment').get("currency"),
+            email=data.get("user_email"),
+            event_time=timezone.now(), # ???
+            marketing_consent=data.get('marketing_consent'),
+            # next_bill_date can be null if user won't pay again...
+            next_bill_date=data.get("next_payment", {}).get("date", timezone.now()),
+            passthrough="", # ???
+            quantity=data.get("last_payment", {}).get("amount", 0),
+            source="", # ???
+            status=data.get("state"),
+            unit_price=0.00, # ???
+            update_url = data.get('update_url'),
+            **kwargs
+        )
+        return sub
+
     @classmethod
     def _sanitize_webhook_payload(cls, payload):
         data = {}
@@ -160,10 +207,14 @@ def _sanitize_webhook_payload(cls, payload):
         # transform `user_id` to subscriber ref
         data["subscriber"] = None
         subscriber_id = payload.pop("user_id", None)
-        if subscriber_id not in ["", None]:
-            data["subscriber"], created = settings.get_subscriber_model().objects.get_or_create(
-                email=payload["email"]
-            )
+        try:
+            if subscriber_id not in ["", None]:
+                data[
+                    "subscriber"], created = settings.get_subscriber_model().objects.get(
+                    email=payload["email"]
+                )
+        except settings.get_subscriber_model().DoesNotExist:
+            pass
 
         # transform `subscription_plan_id` to plan ref
         data["plan"] = None