diff options
Diffstat (limited to 'api/auth/token.py')
-rw-r--r-- | api/auth/token.py | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/api/auth/token.py b/api/auth/token.py index d75c91b..e94b014 100644 --- a/api/auth/token.py +++ b/api/auth/token.py @@ -8,32 +8,32 @@ import time # get valid token hashes for a given user_id def valid_tokens(user_id): tokens = json.loads( - cursor.execute( - "select valid_tokens from users where user_id = ?", [user_id] - ).fetchone()[0] + cursor.execute( + "select valid_tokens from users where user_id = ?", [user_id] + ).fetchone()[0] ) # return only tokens that aren't expired return [ - token for token in tokens - if token["expirationDate"] > int(time.time() * 1000) + token for token in tokens + if token["expirationDate"] > int(time.time() * 1000) ] def validate_token(user_id, token): tokens = valid_tokens(user_id) return hashlib.sha256(str(token).encode()).hexdigest() in [ - t["token"] for t in tokens - if t["expirationDate"] > int(time.time() * 1000) + t["token"] for t in tokens + if t["expirationDate"] > int(time.time() * 1000) ] def modify_tokens(user_id, formatted_token, remove): temp_tokens = valid_tokens(user_id) temp_tokens.remove(formatted_token - ) if remove else temp_tokens.append(formatted_token) + ) if remove else temp_tokens.append(formatted_token) cursor.execute( - "update users set valid_tokens = ? where user_id = ?", - [json.dumps(temp_tokens), user_id] + "update users set valid_tokens = ? where user_id = ?", + [json.dumps(temp_tokens), user_id] ) connection.commit() @@ -48,13 +48,13 @@ def revoke_token(user_id, formatted_token): def hash_token(token): return { - "token": hashlib.sha256(str(token["token"]).encode()).hexdigest(), - "expirationDate": token["expirationDate"] + "token": hashlib.sha256(str(token["token"]).encode()).hexdigest(), + "expirationDate": token["expirationDate"] } def generate_token(): return { - "token": secrets.token_hex(128), - "expirationDate": int(time.time() * 1000) + (24 * 60 * 60 * 1000) + "token": secrets.token_hex(128), + "expirationDate": int(time.time() * 1000) + (24 * 60 * 60 * 1000) } |