TelegramBot/modules/reminder.py

import logging
from datetime import datetime, timedelta

from bson import json_util
from libbot.pyrogram.classes import PyroClient

from classes.enums import GarbageType
from classes.location import Location
from classes.pyrouser import PyroUser
from modules.database import col_users
from modules.database_api import col_entries
from modules.utils import from_utc

logger = logging.getLogger(__name__)


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

    logger.debug("Performing reminder lookup for %s (UTCNOW)", utcnow)

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

    logger.debug(
        "Found following reminders for %s (UTC NOW): %s",
        utcnow,
        json_util.dumps(users, indent=None),
    )

    for user_db in users:
        logger.debug("Processing user %s...", json_util.dumps(user_db, indent=None))

        user = await PyroUser.from_dict(**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 = from_utc(
            datetime.utcnow() + timedelta(days=user.offset),
            user.location.timezone.zone,
        ).replace(hour=0, minute=0, second=0, microsecond=0)

        entries = await col_entries.find(
            {
                "locations": location.id,
                "date": user_date,
            }
        ).to_list()

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

        for entry in entries:
            logger.debug(
                "Sending %s notification about %s",
                user.id,
                json_util.dumps(entry, indent=None),
            )

            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