Compare commits

...

3 Commits

Author SHA1 Message Date
c7675c231f Fixed sending a final message in /import 2023-01-03 21:07:37 +03:00
ae88fccf13 Updated type hint of the handler in the register_state function 2023-01-03 12:34:54 +03:00
e29eefe40b Changes in helper_functions.delete_message
Removed checking if sleep_time is 0
Moved sleeping outside of try block
2023-01-03 12:29:08 +03:00
2 changed files with 6 additions and 6 deletions

View File

@ -18,7 +18,7 @@ states: dict[tuple[int, int], Handler] = {}
def register_state( def register_state(
message: Message, message: Message,
handler: Callable[[Message], Any], handler: Handler,
) -> None: ) -> None:
states[(message.chat.id, message.from_user.id)] = handler states[(message.chat.id, message.from_user.id)] = handler
@ -40,9 +40,8 @@ async def delete_message(
*, *,
sleep_time: int = 0, sleep_time: int = 0,
) -> bool: ) -> bool:
await asyncio.sleep(sleep_time)
try: try:
if sleep_time != 0:
await asyncio.sleep(sleep_time)
await bot.delete_message(mes.chat.id, mes.id) await bot.delete_message(mes.chat.id, mes.id)
except telebot.apihelper.ApiException: except telebot.apihelper.ApiException:
return False return False

View File

@ -650,11 +650,12 @@ async def _import3(
failed.append(account.name) failed.append(account.name)
if failed: if failed:
mes_text = "Не удалось добавить:\n" + "\n".join(failed) await send_deleteable_message(
bot, mes.chat.id, "Не удалось добавить:\n" + "\n".join(failed)
)
else: else:
mes_text = "Успех" await send_tmp_message(bot, mes.chat.id, "Успех")
await send_tmp_message(bot, mes.chat.id, mes_text, 10)
del text, mes, accounts del text, mes, accounts
gc.collect() gc.collect()