83 lines
2.5 KiB
Python
83 lines
2.5 KiB
Python
import logging
|
|
from datetime import datetime
|
|
|
|
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
|
|
from modules.utils import to_utc
|
|
|
|
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
|
|
|
|
parsed_time = datetime.strptime(answer.text, "%H:%M").replace(
|
|
year=1970, month=1, day=1, second=0, microsecond=0
|
|
)
|
|
user_time = to_utc(parsed_time, user.location.timezone.zone)
|
|
|
|
await user.update_time(hour=user_time.hour, minute=user_time.minute)
|
|
|
|
logger.info(
|
|
"User %s has selected notification time of %s",
|
|
user.id,
|
|
user_time.strftime("%H:%M"),
|
|
)
|
|
|
|
garbage_time = parsed_time.strftime(app._("time", "formats"))
|
|
|
|
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(),
|
|
)
|