aboutsummaryrefslogtreecommitdiff
path: root/api/auth/token.py
diff options
context:
space:
mode:
Diffstat (limited to 'api/auth/token.py')
-rw-r--r--api/auth/token.py28
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)
}