diff options
Diffstat (limited to 'jb/api/magic_token.py')
| -rw-r--r-- | jb/api/magic_token.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/jb/api/magic_token.py b/jb/api/magic_token.py index dc05686..189db02 100644 --- a/jb/api/magic_token.py +++ b/jb/api/magic_token.py @@ -10,6 +10,7 @@ from jb.models.auth import AmtAccountLink MAGIC_TOKEN_PREFIX = "auth:magic:" AMT_ACCOUNT_LINK_TOKEN_PREFIX = "auth:amt-account-link:" MAGIC_TOKEN_TTL: int = 3 * 60 * 60 # 3hrs, in seconds +INVITE_AMT_LINK_TOKEN_TTL: int = 45 * 24 * 60 * 60 # 45 days def redis_token_key(token: str, prefix: str = MAGIC_TOKEN_PREFIX) -> str: @@ -56,7 +57,10 @@ def consume_magic_token(token: str, redis_config: RedisConfig | None = None) -> def create_amt_account_link_token( - email: str, amt_worker_id: str, redis_config: RedisConfig | None = None + email: str, + amt_worker_id: str, + redis_config: RedisConfig | None = None, + ttl_seconds: int = MAGIC_TOKEN_TTL, ) -> str: """Bind an email and AMT worker ID to an opaque, short-lived token.""" if redis_config is None: @@ -72,7 +76,7 @@ def create_amt_account_link_token( redis_client.set( redis_token_key(token, AMT_ACCOUNT_LINK_TOKEN_PREFIX), data.model_dump_json(), - ex=MAGIC_TOKEN_TTL, + ex=ttl_seconds, ) return token |