Vorkout/connect
3
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix(api): on duplicate password update

Browse Source
This commit is contained in:
Vladislav Syrochkin 2025-06-30 17:52:31 +05:00
parent ad1369c3e3
commit 4c0beb24f9
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/api/db/logic/keyring.py
View File

@ -3,12 +3,12 @@ from enum import Enum
from typing import Optional from typing import Optional
from sqlalchemy import insert, select, update from sqlalchemy import insert, select, update
from sqlalchemy.dialects.mysql import insert as mysql_insert
from sqlalchemy.ext.asyncio import AsyncConnection from sqlalchemy.ext.asyncio import AsyncConnection
from api.db.tables.account import account_keyring_table, KeyStatus, KeyType from api.db.tables.account import account_keyring_table, KeyStatus, KeyType
from api.schemas.account.account_keyring import AccountKeyring from api.schemas.account.account_keyring import AccountKeyring
from api.utils.hasher import hasher from api.utils.hasher import hasher
from api.utils.key_id_gen import KeyIdGenerator
async def get_key_by_id(connection: AsyncConnection, key_id: str) -> Optional[AccountKeyring]: async def get_key_by_id(connection: AsyncConnection, key_id: str) -> Optional[AccountKeyring]:
@ -73,15 +73,19 @@ async def create_key(connection: AsyncConnection, key: AccountKeyring, key_id: i
async def create_password_key(connection: AsyncConnection, password: str | None, owner_id: int): async def create_password_key(connection: AsyncConnection, password: str | None, owner_id: int):
if password is None: if password is None:
password = hasher.generate_password() password = hasher.generate_password()
stmt = insert(account_keyring_table).values( hashed_password = hasher.hash_data(password)
stmt = mysql_insert(account_keyring_table).values(
owner_id=owner_id, owner_id=owner_id,
key_type=KeyType.PASSWORD.value, key_type=KeyType.PASSWORD.value,
key_id=KeyIdGenerator(), key_id="PASSWORD",
key_value=hasher.hash_data(password), key_value=hashed_password,
created_at=datetime.now(timezone.utc), created_at=datetime.now(timezone.utc),
expiry=datetime.now(timezone.utc) + timedelta(days=365), expiry=datetime.now(timezone.utc) + timedelta(days=365),
status=KeyStatus.ACTIVE, status=KeyStatus.ACTIVE,
) )
stmt.on_duplicate_key_update(
key_value=hashed_password
)
await connection.execute(stmt) await connection.execute(stmt)
await connection.commit() await connection.commit()