Improve Tribute auto-renew

bot/main_bot.py

@@ -32,6 +32,7 @@ from bot.handlers.admin import admin_router_aggregate
 from bot.filters.admin_filter import AdminFilter
 
 from bot.services.notification_service import schedule_subscription_notifications
+from bot.services.auto_renew_service import schedule_tribute_autorenew
 from bot.services.yookassa_service import YooKassaService
 from bot.services.panel_api_service import PanelApiService
 from bot.services.subscription_service import SubscriptionService
@@ -116,6 +117,12 @@ async def on_startup_configured(dispatcher: Dispatcher):
                 panel_service,
                 async_session_factory,
             )
+            schedule_tribute_autorenew(
+                settings,
+                scheduler,
+                panel_service,
+                async_session_factory,
+            )
             scheduler.start()
             dispatcher["scheduler"] = scheduler
             logging.info("STARTUP: APScheduler started.")

bot/services/auto_renew_service.py

@@ -0,0 +1,75 @@
+import logging
+from datetime import datetime, timezone
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from sqlalchemy.orm import sessionmaker
+
+from config.settings import Settings
+from bot.services.subscription_service import SubscriptionService
+from bot.services.panel_api_service import PanelApiService
+from db.dal import subscription_dal
+
+
+async def auto_extend_tribute_subscriptions(
+        settings: Settings, panel_service: PanelApiService,
+        async_session_factory: sessionmaker) -> None:
+    logging.info(
+        f"Scheduler job 'auto_extend_tribute_subscriptions' started at {datetime.now(timezone.utc)} UTC."
+    )
+    async with async_session_factory() as session:
+        try:
+            sub_service = SubscriptionService(settings, panel_service)
+            subs = await subscription_dal.get_active_subscriptions_for_autorenew(
+                session, 'tribute', require_skip_flag=False)
+            if not subs:
+                logging.info("No Tribute subscriptions to auto-extend.")
+                return
+            for sub in subs:
+                if not sub.skip_notifications:
+                    from db.dal import payment_dal
+                    has_pay = await payment_dal.user_has_successful_payment_for_provider(
+                        session, sub.user_id, 'tribute')
+                    if not has_pay:
+                        continue
+                    await subscription_dal.set_skip_notifications_for_provider(
+                        session, sub.user_id, 'tribute', True)
+
+                months = sub.duration_months or 1
+                bonus_days = months * 30
+                await sub_service.extend_active_subscription_days(
+                    session, sub.user_id, bonus_days,
+                    reason='tribute_autorenew')
+            await session.commit()
+            logging.info(
+                f"Auto-extended {len(subs)} Tribute subscriptions and committed session.")
+        except Exception as e:
+            logging.error(
+                f"Error during auto_extend_tribute_subscriptions: {e}",
+                exc_info=True)
+            await session.rollback()
+            logging.info("Session rolled back due to error in auto_extend_tribute_subscriptions.")
+
+
+def schedule_tribute_autorenew(
+        settings: Settings, scheduler: AsyncIOScheduler,
+        panel_service: PanelApiService,
+        async_session_factory: sessionmaker) -> None:
+
+    async def job_wrapper():
+        try:
+            await auto_extend_tribute_subscriptions(
+                settings, panel_service, async_session_factory)
+        except Exception as e:
+            logging.error(
+                f"Unhandled error in scheduled job 'auto_extend_tribute_subscriptions': {e}",
+                exc_info=True)
+
+    scheduler.add_job(
+        job_wrapper,
+        'cron',
+        hour=0,
+        minute=5,
+        name='daily_tribute_autorenew',
+        misfire_grace_time=60 * 15,
+        replace_existing=True)
+    logging.info(
+        "Tribute subscription auto-renew job scheduled daily at 00:05 UTC.")