TelegramBot/plugins/commands/upcoming.py

from datetime import datetime, timedelta, timezone

from pymongo.errors import OperationFailure
from pyrogram import filters
from pyrogram.types import Message
from pytz import timezone as pytz_timezone

from classes.garbage_entry import GarbageEntry
from classes.pyroclient import PyroClient
from modules.database import col_entries


@PyroClient.on_message(
    ~filters.scheduled & filters.private & filters.command(["upcoming"], prefixes=["/"])  # type: ignore
)
async def command_upcoming(app: PyroClient, message: Message):
    user = await app.find_user(message.from_user)

    if user.location is None:
        await message.reply_text(
            app._("upcoming_empty_location", "messages", locale=user.locale)
        )
        return

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

    try:
        entries = [
            await GarbageEntry.from_record(entry)
            async for entry in col_entries.find(
                {
                    "locations": {"$in": user.location.id},
                    "date": {"$gte": date_min, "$lte": date_max},
                }
            )
        ]
    except OperationFailure:
        await message.reply_text(
            app._("upcoming_empty", "messages", locale=user.locale).format(
                name=user.location.name
            )
        )
        return

    entries_text = "\n\n".join(
        [
            f"**{entry.date.strftime(app._('date', 'formats', locale=user.locale))}**:\n{app._(str(entry.garbage_type.value), 'garbage_types')}"
            for entry in entries
        ]
    )

    if not entries:
        await message.reply_text(
            app._("upcoming_empty", "messages", locale=user.locale).format(
                name=user.location.name
            )
        )
        return

    await message.reply_text(
        app._("upcoming", "messages", locale=user.locale).format(entries=entries_text)
    )