Compare commits
2 Commits
4122e15308
...
b56ebd0b61
Author | SHA1 | Date | |
---|---|---|---|
b56ebd0b61 | |||
adf9865fbe |
@ -9,6 +9,9 @@ from .. import cryptography, database
|
|||||||
from .utils import (
|
from .utils import (
|
||||||
accounts_to_json,
|
accounts_to_json,
|
||||||
base_handler,
|
base_handler,
|
||||||
|
check_account_name,
|
||||||
|
check_login,
|
||||||
|
check_passwd,
|
||||||
get_all_accounts,
|
get_all_accounts,
|
||||||
json_to_accounts,
|
json_to_accounts,
|
||||||
send_tmp_message,
|
send_tmp_message,
|
||||||
@ -20,15 +23,19 @@ Message = telebot.types.Message
|
|||||||
def get_accounts(
|
def get_accounts(
|
||||||
bot: telebot.TeleBot, engine: Engine, mes: telebot.types.Message
|
bot: telebot.TeleBot, engine: Engine, mes: telebot.types.Message
|
||||||
) -> None:
|
) -> None:
|
||||||
|
base_handler(bot, mes)
|
||||||
accounts = database.get.get_accounts(engine, mes.from_user.id)
|
accounts = database.get.get_accounts(engine, mes.from_user.id)
|
||||||
bot.delete_message(mes.chat.id, mes.id)
|
if accounts:
|
||||||
|
accounts = [f"`{account}`" for account in accounts]
|
||||||
return send_tmp_message(
|
return send_tmp_message(
|
||||||
bot,
|
bot,
|
||||||
mes.chat.id,
|
mes.chat.id,
|
||||||
"Ваши аккаунты:\n" + "\n".join(accounts) if accounts else "У вас нет аккаунтов",
|
"Ваши аккаунты:\n"
|
||||||
timeout=30,
|
+ "\n".join(accounts)
|
||||||
)
|
+ "\nНажмите на название, чтобы скопировать",
|
||||||
|
30,
|
||||||
|
)
|
||||||
|
send_tmp_message(bot, mes.chat.id, "У вас нет аккаунтов")
|
||||||
|
|
||||||
|
|
||||||
def delete_all(
|
def delete_all(
|
||||||
@ -135,6 +142,8 @@ def _add_account2(
|
|||||||
if text == "/cancel":
|
if text == "/cancel":
|
||||||
return send_tmp_message(bot, mes.chat.id, "Успешная отмена")
|
return send_tmp_message(bot, mes.chat.id, "Успешная отмена")
|
||||||
|
|
||||||
|
if not check_account_name(text):
|
||||||
|
return send_tmp_message(bot, mes.chat.id, "Не корректное название аккаунта")
|
||||||
if text in database.get.get_accounts(engine, mes.from_user.id):
|
if text in database.get.get_accounts(engine, mes.from_user.id):
|
||||||
return send_tmp_message(
|
return send_tmp_message(
|
||||||
bot, mes.chat.id, "Аккаунт с таким именем уже существует"
|
bot, mes.chat.id, "Аккаунт с таким именем уже существует"
|
||||||
@ -159,6 +168,8 @@ def _add_account3(
|
|||||||
text = mes.text.strip()
|
text = mes.text.strip()
|
||||||
if text == "/cancel":
|
if text == "/cancel":
|
||||||
return send_tmp_message(bot, mes.chat.id, "Успешная отмена")
|
return send_tmp_message(bot, mes.chat.id, "Успешная отмена")
|
||||||
|
if not check_login(text):
|
||||||
|
return send_tmp_message(bot, mes.chat.id, "Не корректный логин")
|
||||||
|
|
||||||
data["login"] = text
|
data["login"] = text
|
||||||
|
|
||||||
@ -180,6 +191,8 @@ def _add_account4(
|
|||||||
text = mes.text.strip()
|
text = mes.text.strip()
|
||||||
if text == "/cancel":
|
if text == "/cancel":
|
||||||
return send_tmp_message(bot, mes.chat.id, "Успешная отмена")
|
return send_tmp_message(bot, mes.chat.id, "Успешная отмена")
|
||||||
|
if not check_passwd(text):
|
||||||
|
return send_tmp_message(bot, mes.chat.id, "Не корректный пароль")
|
||||||
|
|
||||||
data["passwd"] = text
|
data["passwd"] = text
|
||||||
|
|
||||||
|
@ -75,3 +75,23 @@ def accounts_to_json(accounts: list[tuple[str, str, str]]) -> io.StringIO:
|
|||||||
file = io.StringIO(_accounts_list_to_json(accounts))
|
file = io.StringIO(_accounts_list_to_json(accounts))
|
||||||
file.name = "passwords.json"
|
file.name = "passwords.json"
|
||||||
return file
|
return file
|
||||||
|
|
||||||
|
|
||||||
|
def _base_check(val: str) -> bool:
|
||||||
|
"Returns false if finds new lines or backtick (`)"
|
||||||
|
return not ("\n" in val or "`" in val)
|
||||||
|
|
||||||
|
|
||||||
|
def check_account_name(name: str) -> bool:
|
||||||
|
"Returns true if account name is valid"
|
||||||
|
return _base_check(name)
|
||||||
|
|
||||||
|
|
||||||
|
def check_login(login: str) -> bool:
|
||||||
|
"Returns true if login is valid"
|
||||||
|
return _base_check(login)
|
||||||
|
|
||||||
|
|
||||||
|
def check_passwd(passwd: str) -> bool:
|
||||||
|
"Returns true if password is valid"
|
||||||
|
return _base_check(passwd)
|
||||||
|
Reference in New Issue
Block a user