From 16f71b0b09933b80420346df3f72de7fa6662aab Mon Sep 17 00:00:00 2001 From: Houkime <> Date: Wed, 21 Dec 2022 13:51:39 +0000 Subject: [PATCH] refactor(tokens-repo): use tokens-repo to create recovery token --- selfprivacy_api/actions/api_tokens.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/selfprivacy_api/actions/api_tokens.py b/selfprivacy_api/actions/api_tokens.py index 61c695d..33ab286 100644 --- a/selfprivacy_api/actions/api_tokens.py +++ b/selfprivacy_api/actions/api_tokens.py @@ -2,11 +2,11 @@ from datetime import datetime from typing import Optional from pydantic import BaseModel +from mnemonic import Mnemonic from selfprivacy_api.utils.auth import ( delete_token, - generate_recovery_token, get_recovery_token_status, get_tokens_info, is_recovery_token_exists, @@ -17,6 +17,12 @@ from selfprivacy_api.utils.auth import ( get_token_name, ) +from selfprivacy_api.repositories.tokens.json_tokens_repository import ( + JsonTokensRepository, +) + +TOKEN_REPO = JsonTokensRepository() + class TokenInfoWithIsCaller(BaseModel): """Token info""" @@ -112,5 +118,6 @@ def get_new_api_recovery_key( if uses_left <= 0: raise InvalidUsesLeft("Uses must be greater than 0") - key = generate_recovery_token(expiration_date, uses_left) - return key + key = TOKEN_REPO.create_recovery_key(expiration_date, uses_left) + mnemonic_phrase = Mnemonic(language="english").to_mnemonic(bytes.fromhex(key.key)) + return mnemonic_phrase