Massive code cleanup

src/bot/handlers.py

@@ -44,7 +44,8 @@ def delete_all(bot: telebot.TeleBot, engine: Engine, mes: Message) -> None:
     base_handler(bot, mes)
     bot_mes = bot.send_message(
         mes.chat.id,
-        "Вы действительно хотите удалить все ваши аккаунты? Это действие нельзя отменить. Отправьте YES для подтверждения",
+        "Вы действительно хотите удалить все ваши аккаунты? Это действие нельзя отменить. "
+        "Отправьте YES для подтверждения",
     )
     bot.register_next_step_handler(
         mes, functools.partial(_delete_all, bot, engine, bot_mes)
@@ -95,7 +96,8 @@ def reset_master_pass(bot: telebot.TeleBot, engine: Engine, mes: Message) -> Non
         return send_tmp_message(bot, mes.chat.id, "Мастер пароль не задан")
     bot_mes = bot.send_message(
         mes.chat.id,
-        "Отправьте новый мастер пароль, осторожно, все текущие аккаунты будут удалены навсегда",
+        "Отправьте новый мастер пароль, осторожно, все текущие аккаунты "
+        "будут удалены навсегда",
     )
     bot.register_next_step_handler(
         mes, functools.partial(_reset_master_pass2, bot, engine, bot_mes)
@@ -290,7 +292,8 @@ def _get_account3(
     send_tmp_message(
         bot,
         mes.chat.id,
-        f"Логин:\n`{login}`\nПароль:\n`{passwd}`\nНажмите на логин или пароль, чтобы скопировать",
+        f"Логин:\n`{login}`\nПароль:\n`{passwd}`\nНажмите на логин или пароль, "
+        "чтобы скопировать",
         30,
     )
 
@@ -341,7 +344,8 @@ def help(bot: telebot.TeleBot, mes: Message) -> None:
 /cancel - отмена текущего действия
 /help - помощь
 /export - получить пароли в json формате
-/import - импортировать пароли из json в файле в таком же формате, как из /export
+/import - импортировать пароли из json в файле в таком же формате, \
+как из /export
 /gen_password - создать 10 надёжных паролей"""
     bot.send_message(mes.chat.id, message)
 
@@ -472,8 +476,8 @@ def _import3(
 
 
 def gen_password(bot: telebot.TeleBot, mes: Message) -> None:
-    # Generate 10 passwords and put 'em in the backticks
     base_handler(bot, mes)
+    # Generate 10 passwords and put 'em in the backticks
     passwords = (f"`{gen_passwd()}`" for _ in range(10))
     text = (
         "Пароли:\n"

src/bot/utils.py

@@ -2,7 +2,7 @@ import io
 import string
 import time
 from random import SystemRandom
-from typing import Self, Type
+from typing import Self, Type, Iterator
 
 import pydantic
 import telebot
@@ -37,7 +37,7 @@ class _Accounts(pydantic.BaseModel):
     accounts: list[_Account] = pydantic.Field(default_factory=list)
 
 
-def _accounts_list_to_json(accounts: list[tuple[str, str, str]]) -> str:
+def _accounts_list_to_json(accounts: Iterator[tuple[str, str, str]]) -> str:
     accounts = _Accounts(accounts=[_Account.from_tuple(i) for i in accounts])
     return accounts.json()
 
@@ -48,7 +48,10 @@ def json_to_accounts(json_: str) -> list[tuple[str, str, str]]:
 
 
 def send_tmp_message(
-    bot: telebot.TeleBot, chat_id: telebot.types.Message, text: str, timeout: int = 5
+    bot: telebot.TeleBot,
+    chat_id: telebot.types.Message,
+    text: str,
+    timeout: int = 5,
 ) -> None:
     bot_mes = bot.send_message(chat_id, text, parse_mode="MarkdownV2")
     time.sleep(timeout)
@@ -65,20 +68,16 @@ def base_handler(
 
 def get_all_accounts(
     engine: Engine, user_id: int, master_pass: str
-) -> list[tuple[str, str, str]]:
-    accounts: list[tuple[str, str, str]] = []
-    for account_name in database.get.get_accounts(engine, user_id):
-        salt, enc_login, enc_passwd = database.get.get_account_info(
-            engine, user_id, account_name
-        )
+) -> Iterator[tuple[str, str, str]]:
+    for account in database.get.get_all_accounts(engine, user_id):
+        name, salt, enc_login, enc_passwd = account
         login, passwd = cryptography.other_accounts.decrypt_account_info(
             enc_login, enc_passwd, master_pass, salt
         )
-        accounts.append((account_name, login, passwd))
-    return accounts
+        yield (name, login, passwd)
 
 
-def accounts_to_json(accounts: list[tuple[str, str, str]]) -> io.StringIO:
+def accounts_to_json(accounts: Iterator[tuple[str, str, str]]) -> io.StringIO:
     file = io.StringIO(_accounts_list_to_json(accounts))
     file.name = "passwords.json"
     return file