37733-vm/core/management/commands/check_subscriptions.py
2026-01-24 04:08:52 +00:00

58 lines
2.5 KiB
Python

from django.core.management.base import BaseCommand
from django.utils import timezone
from core.models import Profile
from core.whatsapp import send_whatsapp_message
from django.utils.translation import gettext as _
from datetime import timedelta
class Command(BaseCommand):
help = 'Checks for expiring subscriptions and sends reminders via WhatsApp'
def handle(self, *args, **options):
today = timezone.now().date()
# Reminders for 7 days, 3 days, and 1 day before expiry
reminder_days = [7, 3, 1]
for days in reminder_days:
expiry_date = today + timedelta(days=days)
expiring_profiles = Profile.objects.filter(
subscription_expiry=expiry_date,
subscription_plan__in=['MONTHLY', 'ANNUAL']
)
for profile in expiring_profiles:
full_phone = profile.full_phone_number
if full_phone:
message = _(
"Hello %(user)s, your MASAR CARGO subscription (%(plan)s) will expire in %(days)s days on %(date)s. "
"Please renew to avoid account suspension."
) % {
'user': profile.user.username,
'plan': profile.get_subscription_plan_display(),
'days': days,
'date': profile.subscription_expiry.strftime('%d/%m/%Y')
}
self.stdout.write(f"Sending reminder to {profile.user.username} ({full_phone}) - {days} days left")
send_whatsapp_message(full_phone, message)
# Also check for profiles that expired TODAY and send a notification
expired_today = Profile.objects.filter(
subscription_expiry=today,
subscription_plan__in=['MONTHLY', 'ANNUAL']
)
for profile in expired_today:
full_phone = profile.full_phone_number
if full_phone:
message = _(
"Hello %(user)s, your MASAR CARGO subscription has EXPIRED today. "
"Your account is now suspended. Please contact support to renew."
) % {
'user': profile.user.username
}
self.stdout.write(f"Sending expiry notice to {profile.user.username} ({full_phone})")
send_whatsapp_message(full_phone, message)
self.stdout.write(self.style.SUCCESS('Successfully checked subscriptions'))