10 Commits

6 changed files with 25 additions and 13 deletions

View File

@ -6,6 +6,11 @@ ENV PYTHONDONTWRITEBYTECODE=1
# Turns off buffering for easier container logging # Turns off buffering for easier container logging
ENV PYTHONUNBUFFERED=1 ENV PYTHONUNBUFFERED=1
WORKDIR /app
# Creates new user
RUN adduser -u 1000 --disabled-password --gecos "" appuser && chown -R appuser /app
# Install deps # Install deps
RUN apt update && apt full-upgrade -y RUN apt update && apt full-upgrade -y
RUN apt install curl gcc -y RUN apt install curl gcc -y
@ -16,11 +21,8 @@ RUN apt install libmariadb3 libmariadb-dev -y
COPY requirements.txt . COPY requirements.txt .
RUN python -m pip install -r requirements.txt RUN python -m pip install -r requirements.txt
WORKDIR /app
COPY . /app COPY . /app
RUN adduser -u 1000 --disabled-password --gecos "" appuser && chown -R appuser /app
USER appuser USER appuser
CMD ["python", "main.py"] CMD ["python", "main.py"]

View File

@ -17,12 +17,14 @@
- /get_accounts - получить список аккаунтов - /get_accounts - получить список аккаунтов
- /get_account - получить логин и пароль аккаунта - /get_account - получить логин и пароль аккаунта
- /delete_account - удалить аккаунт - /delete_account - удалить аккаунт
- /delete_all - удалить все аккаунты и мастер пароль (Спать хочу -> подтверждения нету) - /delete_all - удалить все аккаунты и мастер пароль
- /reset_master_pass- удалить все аккаунты и изменить мастер пароль - /reset_master_pass- удалить все аккаунты и изменить мастер пароль
- /cancel - отмена текущего действия
- /help - помощь
### Настройка ### Настройка
Настройка питизводится через переменные среды. Их можно прописать в файле .env, если не хотите задавать их каждый раз в случае, если вы не работаете с Docker Настройка производится через переменные среды. Их можно прописать в файле .env, если не хотите задавать их каждый раз в случае, если вы не используете Docker
#### Переменные среды #### Переменные среды
@ -31,4 +33,3 @@
- DB_PASS - пароль пользователя базы данных - DB_PASS - пароль пользователя базы данных
- DB_NAME - название базы данных - DB_NAME - название базы данных
- TG_TOKEN - токен телеграма, его следует указать в файле .env в формате TG_TOKEN={токен} - TG_TOKEN - токен телеграма, его следует указать в файле .env в формате TG_TOKEN={токен}
- /cancel - отмена текущего действия

View File

@ -2,7 +2,6 @@ import functools
import mariadb import mariadb
import telebot import telebot
from sqlalchemy.future import Engine
from . import handlers from . import handlers

View File

@ -13,7 +13,7 @@ Message = telebot.types.Message
def _send_tmp_message( 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: ) -> None:
bot_mes = bot.send_message(chat_id, text) bot_mes = bot.send_message(chat_id, text, parse_mode="MarkdownV2")
time.sleep(timeout) time.sleep(timeout)
bot.delete_message(chat_id, bot_mes.id) bot.delete_message(chat_id, bot_mes.id)
@ -93,7 +93,10 @@ def reset_master_pass(bot: telebot.TeleBot, engine: Engine, mes: Message) -> Non
_base(bot, mes) _base(bot, mes)
if database.get.get_master_pass(engine, mes.from_user.id) is None: if database.get.get_master_pass(engine, mes.from_user.id) is None:
return _send_tmp_message(bot, mes.chat.id, "Мастер пароль не задан") return _send_tmp_message(bot, mes.chat.id, "Мастер пароль не задан")
bot_mes = bot.send_message(mes.chat.id, "Отправьте мастер пароль") bot_mes = bot.send_message(
mes.chat.id,
"Отправьте новый мастер пароль, осторожно, все текущие аккаунты будут удалены навсегда",
)
bot.register_next_step_handler( bot.register_next_step_handler(
mes, functools.partial(_reset_master_pass2, bot, engine, bot_mes) mes, functools.partial(_reset_master_pass2, bot, engine, bot_mes)
) )
@ -273,7 +276,12 @@ def _get_account3(
login, passwd = cryptography.other_accounts.decrypt_account_info( login, passwd = cryptography.other_accounts.decrypt_account_info(
enc_login, enc_pass, text.encode("utf-8"), salt enc_login, enc_pass, text.encode("utf-8"), salt
) )
_send_tmp_message(bot, mes.chat.id, f"Логин:\n{login}\nПароль:\n{passwd}", 30) _send_tmp_message(
bot,
mes.chat.id,
f"Логин:\n`{login}`\nПароль:\n`{passwd}`\nНажмите на логин или пароль, чтобы скопировать",
30,
)
del text, mes, passwd, login del text, mes, passwd, login
gc.collect() gc.collect()
@ -312,9 +320,10 @@ def help(bot: telebot.TeleBot, mes: telebot.types.Message) -> None:
/get_accounts - получить список аккаунтов /get_accounts - получить список аккаунтов
/get_account - получить логин и пароль аккаунта /get_account - получить логин и пароль аккаунта
/delete_account - удалить аккаунт /delete_account - удалить аккаунт
/delete_all - удалить все аккаунты и мастер пароль (Спать хочу -> подтверждения нету) /delete_all - удалить все аккаунты и мастер пароль
/reset_master_pass - удалить все аккаунты и изменить мастер пароль /reset_master_pass - удалить все аккаунты и изменить мастер пароль
/cancel - отмена текущего действия""" /cancel - отмена текущего действия
/help - помощь"""
bot.send_message(mes.chat.id, message) bot.send_message(mes.chat.id, message)

View File

@ -14,3 +14,4 @@ def change_master_pass(
) )
with sqlmodel.Session(engine) as session: with sqlmodel.Session(engine) as session:
session.exec(statement) session.exec(statement)
session.commit()

View File

@ -22,7 +22,7 @@ def delete_master_pass(engine: Engine, user_id: int) -> None:
def delete_account(engine: Engine, user_id: int, name: str) -> None: def delete_account(engine: Engine, user_id: int, name: str) -> None:
statement = sqlmodel.delete(models.Account).where( statement = sqlmodel.delete(models.Account).where(
models.Account.user_id == user_id and models.Account.name == name models.Account.user_id == user_id, models.Account.name == name
) )
with sqlmodel.Session(engine) as session: with sqlmodel.Session(engine) as session:
session.exec(statement) session.exec(statement)