diff --git a/billing/tasks.py b/billing/tasks.py index 104d6fb..1a2e790 100644 --- a/billing/tasks.py +++ b/billing/tasks.py @@ -4,16 +4,22 @@ 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 billing.models import Topup, Payment from django.utils.timezone import localtime from datetime import datetime +from enum import Enum logger = logging.getLogger(__name__) +class NotificationType(Enum): + TOPUP = "TOPUP" + PAYMENT = "PAYMENT" + + @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( @@ -33,8 +39,9 @@ def update_expired_topups(timestamp: int): if topup.user and topup.user.mobile and not topup.expiry_notification_sent: send_sms_task.defer( mobile=topup.user.mobile, + type=NotificationType.TOPUP, amount=topup.amount, - topup_id=str(topup.id), + model_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 @@ -43,7 +50,6 @@ def update_expired_topups(timestamp: int): 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 @@ -53,29 +59,94 @@ def update_expired_topups(timestamp: int): } +@app.periodic( + cron="*/1 * * * * *", periodic_id="notify_expired_payments", queue="heavy_tasks" +) +@app.task +def update_expired_payments(timestamp: int): + expired_payments_qs = Payment.objects.filter( + expires_at__lte=timezone.now(), + expiry_notification_sent=False, + paid=False, + ).select_related("user") + if not expired_payments_qs.exists(): + logger.info("No expired payments found.") + return {"total_expired_payments": 0} + + with transaction.atomic(): + count = expired_payments_qs.count() + logger.info(f"Found {count} payments to expire.") + + for payment in expired_payments_qs: + if ( + payment.user + and payment.user.mobile + and not payment.expiry_notification_sent + ): + send_sms_task.defer( + mobile=payment.user.mobile, + type=NotificationType.PAYMENT, + amount=payment.amount, + model_id=str(payment.id), + created_at=localtime(payment.created_at).isoformat(), + user=f"{payment.user.first_name + ' ' + payment.user.last_name}" + if payment.user.last_name and payment.user.first_name + else "User", + ) + payment.expiry_notification_sent = True + payment.save() + else: + payment.expiry_notification_sent = True + payment.save() + return + + return { + "total_expired_payments": count, + } + + @app.task def send_sms_task( - user: str, mobile: str, amount: float, topup_id: str, created_at: str + user: str, + mobile: str, + amount: float, + model_id: str, + created_at: str, + type: NotificationType = NotificationType.TOPUP, ): - # Parse the ISO formatted date string try: dt = datetime.fromisoformat(created_at) formatted_date = dt.strftime("%d %b %Y, %I:%M %p") except Exception: - formatted_date = created_at # fallback to original if parsing fails - - message = ( - f"Dear {user}, \n\nYour topup of {amount} MVR [created at {formatted_date}] has expired. " - "Please make a new topup to update your wallet. \n\n- SAR Link" - ) + formatted_date = created_at + message: str = "" + if type == NotificationType.TOPUP: + message = ( + f"Dear {user}, \n\nYour topup of {amount} MVR [created at {formatted_date}] has expired. " + "Please make a new topup to update your wallet. \n\n- SAR Link" + ) + elif type == NotificationType.PAYMENT: + message = f"Dear {user}, \n\nYour payment of {amount} MVR [created at {formatted_date}] has expired. \n\n- SAR Link" send_sms(mobile, message) - logger.info(f"SMS sent to {mobile} for expired topup of {amount} MVR.") + logger.info(f"SMS sent to {mobile} for expired {type.value} 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.") + if type == NotificationType.TOPUP: + try: + topup = Topup.objects.get(id=model_id) + topup.expiry_notification_sent = True + topup.save() + logger.info(f"Marked topup {model_id} as notified.") + except Topup.DoesNotExist: + logger.error( + f"Topup id: {model_id} not found when trying to mark as notified." + ) + else: + try: + topup = Payment.objects.get(id=model_id) + topup.expiry_notification_sent = True + topup.save() + logger.info(f"Marked payment {model_id} as notified.") + except Payment.DoesNotExist: + logger.error( + f"Payment id: {model_id} not found when trying to mark as notified." + )