Compare commits

...

2 Commits

Author SHA1 Message Date
0d3965d5d2 Renamed columns in tables
MasterPass passwd -> password_hash
Account enc_pass -> enc_password
2022-11-30 17:05:04 +03:00
04bb306751 Added _base_handler call to the cancel function in handlers 2022-11-30 16:53:59 +03:00
5 changed files with 10 additions and 9 deletions

View File

@ -7,14 +7,14 @@ import pydantic
class _Account(pydantic.BaseModel): class _Account(pydantic.BaseModel):
name: str name: str
login: str login: str
passwd: str password: str
@classmethod @classmethod
def from_tuple(cls: Type[Self], tuple_: tuple[str, str, str]) -> Self: def from_tuple(cls: Type[Self], tuple_: tuple[str, str, str]) -> Self:
return cls(name=tuple_[0], login=tuple_[1], passwd=tuple_[2]) return cls(name=tuple_[0], login=tuple_[1], passwd=tuple_[2])
def as_tuple(self: Self) -> tuple[str, str, str]: def as_tuple(self: Self) -> tuple[str, str, str]:
return (self.name, self.login, self.passwd) return (self.name, self.login, self.password)
class _Accounts(pydantic.BaseModel): class _Accounts(pydantic.BaseModel):

View File

@ -365,6 +365,7 @@ def help(bot: telebot.TeleBot, mes: Message) -> None:
def cancel(bot: telebot.TeleBot, mes: Message) -> None: def cancel(bot: telebot.TeleBot, mes: Message) -> None:
_base_handler(bot, mes)
_send_tmp_message(bot, mes.chat.id, "Нет активного действия") _send_tmp_message(bot, mes.chat.id, "Нет активного действия")

View File

@ -19,7 +19,7 @@ def add_account(
name=name, name=name,
salt=salt, salt=salt,
enc_login=enc_login, enc_login=enc_login,
enc_pass=enc_password, enc_password=enc_password,
) )
try: try:
with sqlmodel.Session(engine) as session: with sqlmodel.Session(engine) as session:
@ -41,7 +41,7 @@ def add_master_pass(
master_pass = models.MasterPass( master_pass = models.MasterPass(
user_id=user_id, user_id=user_id,
salt=salt, salt=salt,
passwd=password_hash, password_hash=password_hash,
) )
try: try:
with sqlmodel.Session(engine) as session: with sqlmodel.Session(engine) as session:

View File

@ -19,7 +19,7 @@ def get_master_pass(
result = session.exec(statement).first() result = session.exec(statement).first()
if result is None: if result is None:
return return
return (result.salt, result.passwd) return (result.salt, result.password_hash)
def get_accounts(engine: Engine, user_id: int) -> list[str]: def get_accounts(engine: Engine, user_id: int) -> list[str]:
@ -43,7 +43,7 @@ def get_all_accounts(
with sqlmodel.Session(engine) as session: with sqlmodel.Session(engine) as session:
result = session.exec(statement) result = session.exec(statement)
yield from ( yield from (
(account.name, account.salt, account.enc_login, account.enc_pass) (account.name, account.salt, account.enc_login, account.enc_password)
for account in result for account in result
) )
@ -61,4 +61,4 @@ def get_account_info(
result = session.exec(statement).first() result = session.exec(statement).first()
if result is None: if result is None:
return return
return (result.salt, result.enc_login, result.enc_pass) return (result.salt, result.enc_login, result.enc_password)

View File

@ -10,7 +10,7 @@ class MasterPass(sqlmodel.SQLModel, table=True):
salt: bytes = sqlmodel.Field( salt: bytes = sqlmodel.Field(
sa_column=sqlmodel.Column(sqlmodel.BINARY(64), nullable=False) sa_column=sqlmodel.Column(sqlmodel.BINARY(64), nullable=False)
) )
passwd: bytes = sqlmodel.Field( password_hash: bytes = sqlmodel.Field(
sa_column=sqlmodel.Column(sqlmodel.BINARY(128), nullable=False) sa_column=sqlmodel.Column(sqlmodel.BINARY(128), nullable=False)
) )
@ -27,6 +27,6 @@ class Account(sqlmodel.SQLModel, table=True):
enc_login: bytes = sqlmodel.Field( enc_login: bytes = sqlmodel.Field(
sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False) sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False)
) )
enc_pass: bytes = sqlmodel.Field( enc_password: bytes = sqlmodel.Field(
sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False) sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False)
) )