Overhauled entire project config, added notifications, email templates, optimized stripe subscriptions, redis caching, and webdriver utilities

backend/subscriptions/signals.py

@@ -0,0 +1,91 @@
+from django.dispatch import receiver
+from django.db.models.signals import post_migrate, post_save
+from .models import UserSubscription, StripePrice, SubscriptionPlan
+from django.core.cache import cache
+from config.settings import STRIPE_SECRET_KEY
+import stripe
+stripe.api_key = STRIPE_SECRET_KEY
+
+# Template for running actions after user have paid for a subscription
+
+
+@receiver(post_save, sender=SubscriptionPlan)
+def clear_cache_after_plan_updates(sender, instance, **kwargs):
+    # Clear cache
+    cache.delete('subscriptionplans')
+
+
+@receiver(post_save, sender=UserSubscription)
+def scan_after_payment(sender, instance, **kwargs):
+    # If the updated/created subscription is valid
+    if instance.valid and instance.user:
+        # TODO: Add any Celery task actions here for regular subscription payees
+        pass
+
+
+@receiver(post_migrate)
+def create_subscriptions(sender, **kwargs):
+    if sender.name == 'subscriptions':
+        print('Importing data from Stripe')
+        created_prices = 0
+        created_plans = 0
+        skipped_prices = 0
+        skipped_plans = 0
+        products = stripe.Product.list(active=True)
+        prices = stripe.Price.list(expand=["data.tiers"], active=True)
+
+        # Create the StripePrice
+        for price in prices['data']:
+            annual = (price['recurring']['interval'] ==
+                      'year') if price['recurring'] else False
+            STRIPE_PRICE, CREATED = StripePrice.objects.get_or_create(
+                stripe_price_id=price['id'],
+                price=price['unit_amount'] / 100,
+                annual=annual,
+                lookup_key=price['lookup_key'],
+                prorated=price['recurring']['usage_type'] == 'metered',
+                currency=price['currency']
+            )
+            if CREATED:
+                created_prices += 1
+            else:
+                skipped_prices += 1
+
+        # Create the SubscriptionPlan
+        for product in products['data']:
+            ANNUAL_PRICE = None
+            MONTHLY_PRICE = None
+            for price in prices['data']:
+                if price['product'] == product['id']:
+                    STRIPE_PRICE = StripePrice.objects.get(
+                        stripe_price_id=price['id'],
+                    )
+                    if STRIPE_PRICE.annual:
+                        ANNUAL_PRICE = STRIPE_PRICE
+                    else:
+                        MONTHLY_PRICE = STRIPE_PRICE
+            if ANNUAL_PRICE or MONTHLY_PRICE:
+                SUBSCRIPTION_PLAN, CREATED = SubscriptionPlan.objects.get_or_create(
+                    name=product['name'],
+                    description=product['description'],
+                    stripe_product_id=product['id'],
+                    annual_price=ANNUAL_PRICE,
+                    monthly_price=MONTHLY_PRICE,
+                    group_exclusive=product['metadata']['group_exclusive'] == 'True'
+                )
+                if CREATED:
+                    created_plans += 1
+                else:
+                    skipped_plans += 1
+            # Skip over plans with missing pricing rates
+            else:
+                print('Skipping plan' +
+                      product['name'] + 'with missing pricing data')
+
+            # Assign the StripePrice to the SubscriptionPlan
+            SUBSCRIPTION_PLAN.save()
+
+        print('Created', created_plans, 'new plans')
+        print('Skipped', skipped_plans, 'existing plans')
+        print('Created', created_prices, 'new prices')
+        print('Skipped', skipped_prices, 'existing prices')