Vladless
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
import asyncio
|
|
|
|
from apscheduler.triggers.cron import CronTrigger
|
|
|
|
from database import async_session_maker, cancel_expired_pending_payments
|
|
from logger import logger
|
|
|
|
|
|
async def scheduled_audit_drain() -> None:
|
|
from audit import drain_audit_redis_to_db
|
|
|
|
try:
|
|
drained = await drain_audit_redis_to_db(async_session_maker)
|
|
logger.info("[AuditDrain] Ночной drain завершён, записано событий: {}", drained)
|
|
except Exception as error:
|
|
logger.error("[AuditDrain] Ошибка ночного drain: {}", error)
|
|
|
|
|
|
async def scheduled_stats_report() -> None:
|
|
from handlers.admin.stats.stats_handler import send_daily_stats_report
|
|
|
|
async with async_session_maker() as session:
|
|
await send_daily_stats_report(session)
|
|
|
|
|
|
async def sweep_stale_payments_job() -> None:
|
|
async with async_session_maker() as session:
|
|
await cancel_expired_pending_payments(session)
|
|
await session.commit()
|
|
|
|
|
|
def scheduled_audit_drain_process_runner() -> None:
|
|
asyncio.run(scheduled_audit_drain())
|
|
|
|
|
|
def scheduled_stats_report_process_runner() -> None:
|
|
asyncio.run(scheduled_stats_report())
|
|
|
|
|
|
def sweep_stale_payments_process_runner() -> None:
|
|
asyncio.run(sweep_stale_payments_job())
|
|
|
|
|
|
async def cleanup_expired_gifts_job() -> None:
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import update as sa_update
|
|
|
|
from database.models import Gift
|
|
|
|
async with async_session_maker() as session:
|
|
try:
|
|
result = await session.execute(
|
|
sa_update(Gift).where(Gift.expiry_time < datetime.utcnow(), Gift.is_used is False).values(is_used=True)
|
|
)
|
|
count = result.rowcount
|
|
await session.commit()
|
|
if count:
|
|
logger.info("[GiftCleanup] Просроченных подарков помечено использованными: {}", count)
|
|
except Exception as error:
|
|
logger.error("[GiftCleanup] Ошибка очистки подарков: {}", error)
|
|
|
|
|
|
def cleanup_expired_gifts_process_runner() -> None:
|
|
asyncio.run(cleanup_expired_gifts_job())
|
|
|
|
|
|
AUDIT_DRAIN_TRIGGER = CronTrigger(hour=0, minute=0, timezone="Europe/Moscow")
|
|
DAILY_STATS_REPORT_TRIGGER = CronTrigger(hour=0, minute=1, timezone="Europe/Moscow")
|
|
STALE_PAYMENTS_SWEEP_TRIGGER = CronTrigger(minute=0, timezone="Europe/Moscow")
|
|
EXPIRED_GIFTS_CLEANUP_TRIGGER = CronTrigger(hour=3, minute=0, timezone="Europe/Moscow")
|