TelegramBot/modules/reminder.py

import logging
from datetime import datetime, timedelta, timezone

from libbot.pyrogram.classes import PyroClient
from pytz import timezone as pytz_timezone

from classes.enums import GarbageType
from classes.location import Location
from classes.pyrouser import PyroUser
from modules.database import col_entries, col_users

logger = logging.getLogger(__name__)


async def remind(app: PyroClient) -> None:
    now = datetime.now()

    users = await col_users.find(
        {"time_hour": now.hour, "time_minute": now.minute}
    ).to_list()

    for user_db in users:
        user = PyroUser(**user_db)

        if not user.enabled or user.location is None:
            continue

        try:
            location: Location = await app.get_location(user.location.id)  # type: ignore
        except ValueError:
            continue

        user_date = (
            datetime.now(pytz_timezone(location.timezone)).replace(
                second=0, microsecond=0
            )
            + timedelta(days=user.offset)
        ).replace(tzinfo=timezone.utc)

        entries = await col_entries.find(
            {
                "location": {"$in": location.id},
                "date": user_date.replace(hour=0, minute=0),
            }
        ).to_list()

        logger.info("Entries of %s for %s: %s", user.id, user_date, entries)

        for entry in entries:
            try:
                garbage_type = app._(
                    str(GarbageType(entry["garbage_type"]).value),
                    "garbage_types",
                    locale=user.locale,
                )
                garbage_date = datetime.strftime(
                    entry["date"], app._("date", "formats", locale=user.locale)
                )

                await app.send_message(
                    user.id,
                    app._("reminder", "messages", locale=user.locale).format(
                        type=garbage_type, date=garbage_date
                    ),
                )
            except Exception as exc:
                logger.warning(
                    "Could not send a notification to %s due to %s", user.id, exc
                )
                continue