Merge pull request #287 from brukberhane/develop

update AuthTokenManager to support custom values
jazzband · Jun 17, 2024 · 0113d42 · 0113d42
2 parents 01c1de4 + 3651a07
commit 0113d42
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/knox/models.py b/knox/models.py
@@ -17,15 +17,16 @@ 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)
         if expiry is not None:
             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