From 3651a07be56addcf7f9f17151f438de67042444c Mon Sep 17 00:00:00 2001 From: Bruk Berhane Asfaw Date: Mon, 17 Oct 2022 17:15:29 +0300 Subject: [PATCH] update AuthTokenManager to support custom values --- knox/models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/knox/models.py b/knox/models.py index b054f289..33ec7239 100644 --- a/knox/models.py +++ b/knox/models.py @@ -17,7 +17,8 @@ def create( self, user, expiry=knox_settings.TOKEN_TTL, - prefix=knox_settings.TOKEN_PREFIX + prefix=knox_settings.TOKEN_PREFIX, + **kwargs ): token = prefix + crypto.create_token_string() digest = crypto.hash_token(token) @@ -25,7 +26,7 @@ def create( expiry = timezone.now() + expiry instance = super(AuthTokenManager, self).create( token_key=token[:CONSTANTS.TOKEN_KEY_LENGTH], digest=digest, - user=user, expiry=expiry) + user=user, expiry=expiry, **kwargs) return instance, token