From d0a29e83794c63ce55ee0f3668c39b61a082f394 Mon Sep 17 00:00:00 2001 From: MuslemRahimi Date: Thu, 13 Feb 2025 11:21:32 +0100 Subject: [PATCH] update cron job --- app/cron_lemon_squeezy.py | 89 +++++++++++++++++++++++++++------------ 1 file changed, 61 insertions(+), 28 deletions(-) diff --git a/app/cron_lemon_squeezy.py b/app/cron_lemon_squeezy.py index 1deac0c..32b2b74 100644 --- a/app/cron_lemon_squeezy.py +++ b/app/cron_lemon_squeezy.py @@ -3,6 +3,8 @@ import asyncio import aiohttp from dotenv import load_dotenv from pocketbase import PocketBase +from dateutil.parser import isoparse + import time # Load API keys and credentials from environment variables load_dotenv() @@ -33,11 +35,12 @@ async def get_subscription_data(): async with aiohttp.ClientSession(headers=headers) as session: while True: - # If we have made 100 requests, wait for 60 seconds before continuing - if request_count > 0 and request_count % 100 == 0: + # Limit API request rate if necessary + if request_count > 0 and request_count % 200 == 0: print("Reached 100 API requests. Waiting for 60 seconds...") await asyncio.sleep(60) + # Note: Do not include the meta parameter url = f"{base_url}?page[number]={page}&page[size]=100" async with session.get(url) as response: if response.status != 200: @@ -46,11 +49,20 @@ async def get_subscription_data(): break data = await response.json() - # Append the subscription data from this page - if "data" in data: - all_subscriptions.extend(data["data"]) - # If a next-page link exists, increment the page counter; otherwise, break the loop + # Extract subscription data and check for custom meta in attributes + if "data" in data: + for item in data["data"]: + subscription = { + "id": item.get("id"), + "attributes": item.get("attributes", {}), + } + # Access your custom meta if it exists under a key like 'custom_meta' + if "custom_meta" in subscription["attributes"]: + subscription["custom_meta"] = subscription["attributes"]["custom_meta"] + all_subscriptions.append(subscription) + + # Handle pagination if a next-page link is available if "links" in data and data["links"].get("next"): page += 1 else: @@ -60,14 +72,20 @@ async def get_subscription_data(): return all_subscriptions + async def run(): + + # Fetch all users (assumes pb.collection(...).get_full_list() is synchronous) all_users = pb.collection("users").get_full_list() users_by_email = {user.email: user for user in all_users if hasattr(user, 'email')} - + + # Fetch all subscriptions (awaited) all_subscriptions = await get_subscription_data() print(f"Total Subscriptions: {len(all_subscriptions)}\n") - # Group subscriptions by email, prioritizing "active" status first, then latest updated_at + # Group subscriptions by email. + # For each email, we want to always choose an active subscription over a non-active one. + # If both subscriptions are of the same "activity" level, we choose the one with the later updated_at. subscriptions_by_email = {} for sub in all_subscriptions: attributes = sub.get('attributes', {}) @@ -76,48 +94,63 @@ async def run(): continue status = attributes.get('status', '').lower() - updated_at = attributes.get('updated_at') + updated_at_str = attributes.get('updated_at') + try: + updated_at = isoparse(updated_at_str) if updated_at_str else None + except Exception as e: + print(f"Error parsing updated_at ({updated_at_str}) for {user_email}: {e}") + updated_at = None + existing_sub = subscriptions_by_email.get(user_email) - - # First-time entry: always add if not existing_sub: subscriptions_by_email[user_email] = sub continue - + + # Get info from the already saved subscription existing_attrs = existing_sub.get('attributes', {}) existing_status = existing_attrs.get('status', '').lower() - existing_updated = existing_attrs.get('updated_at') - - # Prioritize "active" status + existing_updated_str = existing_attrs.get('updated_at') + try: + existing_updated = isoparse(existing_updated_str) if existing_updated_str else None + except Exception as e: + print(f"Error parsing existing updated_at ({existing_updated_str}) for {user_email}: {e}") + existing_updated = None + + # If the new sub is active, it should win over any non-active subscription. if status == 'active': if existing_status == 'active': - # Both active: keep the newer one - if updated_at > existing_updated: + # Both active: keep the one with the later updated_at. + if updated_at and existing_updated and updated_at > existing_updated: subscriptions_by_email[user_email] = sub else: - # Replace non-active with active + # Replace a non-active subscription with an active one. subscriptions_by_email[user_email] = sub else: + # New sub is not active. if existing_status != 'active': - # Neither is active: keep the newer one - if updated_at > existing_updated: + # Both are non-active: choose the one with the later updated_at. + if updated_at and existing_updated and updated_at > existing_updated: subscriptions_by_email[user_email] = sub - - # Process filtered subscriptions + # If one of the dates is missing, you might add extra logic here. + + # Process the subscriptions after grouping. for user_email, sub in subscriptions_by_email.items(): try: attributes = sub.get('attributes', {}) status = attributes.get('status', 'N/A').lower() user = users_by_email.get(user_email) - + if status in ['expired', 'refunded']: + # Example logic: downgrade a Pro user if not lifetime and subscription is expired/refunded. if user and getattr(user, 'tier', None) == 'Pro' and not getattr(user, 'lifetime', False): - pb.collection('users').update(user.id, { - 'tier': 'Free' - }) + # Uncomment the line below to perform the update: + # pb.collection('users').update(user.id, {'tier': 'Free'}) print(f"Downgraded: {user_email}") - except: - pass + print(attributes) + + except Exception as e: + print(f"Error processing user {user_email}: {e}") + if __name__ == "__main__":