From 27a7c24bc35f415977c611c1e00a2909d4c5cbc5 Mon Sep 17 00:00:00 2001 From: Houkime <> Date: Mon, 12 Dec 2022 10:42:16 +0000 Subject: [PATCH] refactor(tokens-repo): separate token storing --- .../repositories/tokens/json_tokens_repository.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/selfprivacy_api/repositories/tokens/json_tokens_repository.py b/selfprivacy_api/repositories/tokens/json_tokens_repository.py index 86e756a..e011b62 100644 --- a/selfprivacy_api/repositories/tokens/json_tokens_repository.py +++ b/selfprivacy_api/repositories/tokens/json_tokens_repository.py @@ -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"""