AmineTMIRI/backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update cron job

Browse Source
This commit is contained in:
MuslemRahimi 2025-02-13 11:21:32 +01:00
parent 4c57c0c520
commit d0a29e8379
1 changed files with 61 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
app/cron_lemon_squeezy.py
View File

@ -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__":