97 lines
3.7 KiB
Python
97 lines
3.7 KiB
Python
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.") |