diff --git a/modules/reminder.py b/modules/reminder.py index 134b0e2..cf2b939 100644 --- a/modules/reminder.py +++ b/modules/reminder.py @@ -39,7 +39,7 @@ async def remind(app: PyroClient) -> None: entries = await col_entries.find( { - "locations": {"$in": location.id}, + "locations": location.id, "date": user_date.replace(hour=0, minute=0), } ).to_list() diff --git a/plugins/commands/upcoming.py b/plugins/commands/upcoming.py index c5866c5..ac29e63 100644 --- a/plugins/commands/upcoming.py +++ b/plugins/commands/upcoming.py @@ -34,23 +34,15 @@ async def command_upcoming(app: PyroClient, message: Message): + 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 - ) + entries = [ + await GarbageEntry.from_record(entry) + async for entry in col_entries.find( + { + "locations": user.location.id, + "date": {"$gte": date_min, "$lte": date_max}, + } ) - return + ] entries_text = "\n\n".join( [