refactor(tokens-repo): separate token storing

pull/26/head
Houkime 2022-12-12 10:42:16 +00:00
parent 4e60d1d37a
commit 27a7c24bc3
1 changed files with 5 additions and 1 deletions

View File

@ -43,6 +43,11 @@ class JsonTokensRepository(AbstractTokensRepository):
"""Create new token"""
new_token = Token.generate(device_name)
self.__store_token(new_token)
return new_token
def __store_token(self, new_token: Token):
with WriteUserData(UserDataFiles.TOKENS) as tokens_file:
tokens_file["tokens"].append(
{
@ -51,7 +56,6 @@ class JsonTokensRepository(AbstractTokensRepository):
"date": new_token.created_at.strftime(DATETIME_FORMAT),
}
)
return new_token
def delete_token(self, input_token: Token) -> None:
"""Delete the token"""