import logging from datetime import datetime, timezone from convopyro import listen_message from pyrogram import filters from pyrogram.types import ForceReply, Message, ReplyKeyboardRemove from classes.pyroclient import PyroClient from modules import custom_filters logger = logging.getLogger(__name__) @PyroClient.on_message( ~filters.scheduled & filters.private & filters.command(["set_time"], prefixes=["/"]) & ~custom_filters.context # type: ignore ) async def command_set_time(app: PyroClient, message: Message): user = await app.find_user(message.from_user) if user.location is None: await message.reply_text( app._("location_empty", "messages", locale=user.locale) ) return await message.reply_text( app._("set_time", "messages", locale=user.locale), reply_markup=ForceReply( placeholder=app._("set_time", "force_replies", locale=user.locale) ), ) while True: app.contexts.append(message.from_user.id) answer = await listen_message(app, message.chat.id, 300) app.contexts.remove(message.from_user.id) if answer is None or answer.text == "/cancel": await message.reply_text( app._("cancelled", "messages", locale=user.locale), reply_markup=ReplyKeyboardRemove(), ) return try: datetime.strptime(answer.text, "%H:%M") except ValueError: await answer.reply_text( app._("set_time_invalid", "messages", locale=user.locale).format( cancel_notice=app._("cancel", "messages", locale=user.locale) ) ) continue break now = datetime.now() parsed_time = datetime.strptime(answer.text, "%H:%M").replace( year=now.year, month=now.month, day=now.day, second=0, microsecond=0, tzinfo=timezone.utc, ) user_time = parsed_time.astimezone(user.location.timezone or timezone.utc) await user.update_time(hour=user_time.hour, minute=user_time.minute) logger.info( "User %s has selected notification time of %s (%s UTC)", user.id, parsed_time.strftime("%H:%M"), user_time.strftime("%H:%M"), ) garbage_time = parsed_time.strftime(app._("time", "formats", locale=user.locale)) await answer.reply_text( app._("set_time_finished", "messages", locale=user.locale).format( offset=user.offset, time=garbage_time, toggle_notice=( "" if user.enabled else app._("toggle", "messages", locale=user.locale) ), ), reply_markup=ReplyKeyboardRemove(), )