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")