Initial commit

2025-05-14 08:11:59 +00:00
commit 77584b5b51
41 changed files with 5205 additions and 0 deletions
@@ -0,0 +1,97 @@
+import logging
+import asyncio
+from aiogram import Bot
+from apscheduler .schedulers .asyncio import AsyncIOScheduler
+from datetime import datetime
+
+from config .settings import Settings
+from .subscription_service import SubscriptionService
+
+from db .database import get_db_connection_manager
+from bot .middlewares .i18n import JsonI18n
+from .panel_api_service import PanelApiService
+
+
+async def send_expiration_warnings (bot :Bot ,settings :Settings ,i18n :JsonI18n ,panel_service :PanelApiService ):
+    """
+    Sends expiration warnings to users whose subscriptions are ending soon.
+    This function is called by the scheduler.
+    """
+    logging .info (f"Scheduler job: Checking for expiring subscriptions at {datetime.now()}...")
+
+
+    sub_service =SubscriptionService (get_db_connection_manager ,settings ,panel_service )
+
+    expiring_subs =await sub_service .get_subscriptions_ending_soon (settings .SUBSCRIPTION_EXPIRATION_NOTIFICATION_DAYS )
+
+    if not expiring_subs :
+        logging .info ("No subscriptions found ending soon for notification.")
+        return
+
+    logging .info (f"Found {len(expiring_subs)} subscriptions ending soon for notification.")
+    for sub_info in expiring_subs :
+        user_id =sub_info ['user_id']
+        if not user_id :
+            logging .warning (f"Skipping notification for subscription without user_id: {sub_info}")
+            continue
+
+        user_lang =sub_info .get ('language_code')if sub_info .get ('language_code')else getattr (settings ,'DEFAULT_LANGUAGE','en')
+        first_name =sub_info .get ('first_name','User')
+        end_date_str =sub_info ['end_date_str']
+
+        days_left_float =sub_info .get ('days_left')
+        days_left_display ='N/A'
+        if days_left_float is not None :
+
+            days_left_display =max (0 ,int (round (days_left_float )))
+
+        _ =lambda key ,**kwargs :i18n .gettext (user_lang ,key ,**kwargs )
+        message_text =_ (
+        "subscription_ending_soon_notification",
+        user_name =first_name ,
+        end_date =end_date_str ,
+        days_left =days_left_display
+        )
+        try :
+            await bot .send_message (user_id ,message_text )
+
+            await sub_service .update_last_notification_sent (user_id ,end_date_str )
+            logging .info (f"Sent expiration warning to user {user_id} for subscription ending {end_date_str}")
+        except Exception as e :
+
+            logging .error (f"Failed to send expiration warning to user {user_id}: {e}")
+        await asyncio .sleep (0.1 )
+
+
+async def schedule_subscription_notifications (bot :Bot ,settings :Settings ,i18n :JsonI18n ,scheduler :AsyncIOScheduler ):
+    """Schedules the daily job to send expiration warnings."""
+
+
+    async def job_wrapper ():
+        panel_service =PanelApiService (settings )
+        try :
+
+            await send_expiration_warnings (bot ,settings ,i18n ,panel_service )
+        except Exception as e :
+            logging .error (f"Error in scheduled job 'send_expiration_warnings': {e}",exc_info =True )
+        finally :
+            await panel_service .close_session ()
+
+
+    try :
+        notification_hour =int (settings .SUBSCRIPTION_NOTIFICATION_HOUR_UTC )
+        notification_minute =int (settings .SUBSCRIPTION_NOTIFICATION_MINUTE_UTC )
+    except (ValueError ,TypeError ):
+        logging .warning ("SUBSCRIPTION_NOTIFICATION_HOUR_UTC or MINUTE_UTC is invalid. Defaulting to 9:00 UTC.")
+        notification_hour =9
+        notification_minute =0
+
+    scheduler .add_job (
+    job_wrapper ,
+    'cron',
+    hour =notification_hour ,
+    minute =notification_minute ,
+    name ="daily_subscription_expiration_warnings",
+    misfire_grace_time =60 *15
+    )
+    logging .info (f"Subscription expiration warning job scheduled daily at {notification_hour:02d}:{notification_minute:02d} UTC.")