PagerMaid-Pyro/pagermaid/dependence/_scheduler.py

import contextlib
import datetime

import pytz
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from typing import TYPE_CHECKING
from pagermaid.config import Config

if TYPE_CHECKING:
    from pagermaid.enums import Message

scheduler = AsyncIOScheduler(timezone=Config.TIME_ZONE)


async def delete_message(message: "Message") -> bool:
    with contextlib.suppress(Exception):
        await message.delete()
        return True
    return False


def add_delete_message_job(message: "Message", delete_seconds: int = 60):
    scheduler.add_job(
        delete_message,
        "date",
        id=f"{message.chat.id}|{message.id}|delete_message",
        name=f"{message.chat.id}|{message.id}|delete_message",
        args=[message],
        run_date=datetime.datetime.now(pytz.timezone(Config.TIME_ZONE))
        + datetime.timedelta(seconds=delete_seconds),
        replace_existing=True,
    )