SelfPrivacy
/
selfprivacy-rest-api
7
2
Fork 1
Code Issues 15 Pull Requests 3 Projects Releases 1 Wiki Activity

refactor(tokens-repo): use tokens-repo to create recovery token

redis/token-repo
Houkime 2022-12-21 13:51:39 +00:00
parent 39277419ac
commit 16f71b0b09
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
selfprivacy_api/actions/api_tokens.py
View File

@ -2,11 +2,11 @@
from datetime import datetime from datetime import datetime
from typing import Optional from typing import Optional
from pydantic import BaseModel from pydantic import BaseModel
from mnemonic import Mnemonic
from selfprivacy_api.utils.auth import ( from selfprivacy_api.utils.auth import (
delete_token, delete_token,
generate_recovery_token,
get_recovery_token_status, get_recovery_token_status,
get_tokens_info, get_tokens_info,
is_recovery_token_exists, is_recovery_token_exists,
@ -17,6 +17,12 @@ from selfprivacy_api.utils.auth import (
get_token_name, get_token_name,
) )
from selfprivacy_api.repositories.tokens.json_tokens_repository import (
JsonTokensRepository,
)
TOKEN_REPO = JsonTokensRepository()
class TokenInfoWithIsCaller(BaseModel): class TokenInfoWithIsCaller(BaseModel):
"""Token info""" """Token info"""
@ -112,5 +118,6 @@ def get_new_api_recovery_key(
if uses_left <= 0: if uses_left <= 0:
raise InvalidUsesLeft("Uses must be greater than 0") raise InvalidUsesLeft("Uses must be greater than 0")
key = generate_recovery_token(expiration_date, uses_left) key = TOKEN_REPO.create_recovery_key(expiration_date, uses_left)
return key mnemonic_phrase = Mnemonic(language="english").to_mnemonic(bytes.fromhex(key.key))
return mnemonic_phrase