import logging from django.db import transaction from django.utils import timezone from procrastinate.contrib.django import app from api.notifications import send_sms from billing.models import Topup from django.utils.timezone import localtime logger = logging.getLogger(__name__) @app.periodic( cron="*/1 * * * * *", periodic_id="notify_expired_topups", queue="heavy_tasks" ) # every 1 minute @app.task def update_expired_topups(timestamp: int): expired_topups_qs = Topup.objects.filter( expires_at__lte=timezone.now(), expiry_notification_sent=False, paid=False, ).select_related("user") if not expired_topups_qs.exists(): logger.info("No expired topups found.") return {"total_expired_topups": 0} with transaction.atomic(): count = expired_topups_qs.count() logger.info(f"Found {count} topups to expire.") for topup in expired_topups_qs: if topup.user and topup.user.mobile and not topup.expiry_notification_sent: send_sms_task.defer( mobile=topup.user.mobile, amount=topup.amount, topup_id=str(topup.id), created_at=localtime(topup.created_at).isoformat(), user=f"{topup.user.first_name + ' ' + topup.user.last_name}" if topup.user.last_name and topup.user.first_name else "User", ) topup.expiry_notification_sent = True topup.save() else: # Mark as notified even if we can't send SMS (no mobile number) topup.expiry_notification_sent = True topup.save() return return { "total_expired_topups": count, } @app.task def send_sms_task( user: str, mobile: str, amount: float, topup_id: str, created_at: str ): message = ( f"Dear {user}, \n\nYour topup of {amount} MVR [created at {created_at}] has expired. " "Please make a new topup to update your wallet. \n\n- SAR Link" ) send_sms(mobile, message) logger.info(f"SMS sent to {mobile} for expired topup of {amount} MVR.") # Mark the topup as notified after successful SMS sending try: topup = Topup.objects.get(id=topup_id) topup.expiry_notification_sent = True topup.save() logger.info(f"Marked topup {topup_id} as notified.") except Topup.DoesNotExist: logger.error(f"Topup {topup_id} not found when trying to mark as notified.")