Spaces:
Running
Running
File size: 1,707 Bytes
9d3cbac 58a7ac0 9d3cbac 58a7ac0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from datetime import datetime
from zoneinfo import ZoneInfo
from typing import Any
from jose import jwt
import hashlib
import json
import uuid
import os
SECRET_KEY = os.environ.get("SECRET_KEY")
def beijing():
"""get beijing time"""
return datetime.now(tz=ZoneInfo("Asia/Shanghai"))
def decode_jwt(token: str) -> dict[str, Any]:
"""get payload in the jwt token"""
assert SECRET_KEY, "Please set the environment variable SECRET_KEY"
return jwt.decode(token, SECRET_KEY, algorithms=["HS256"])
def parse_token(token: str | None) -> tuple[str, str]:
"""parse user token to get uid and username
:param token (str): jwt token
:return (str, str): (uid, username)
"""
if not token:
uid, username = "no_uid", "Anonymous"
else:
payload: dict = decode_jwt(token)
uid = payload.get("uid", "no_uid")
username = payload.get("username", "Anonymous")
print(f"UID: {uid}")
print(f"Username: {username}")
return uid, username
def md5(text: list[str | bytes] | str | bytes | None = None) -> str:
"""generate the md5 hash code of the given text, if text is None,
return a random md5"""
code = hashlib.md5()
if text:
if isinstance(text, str):
text = text.encode("utf-8")
code.update(text)
elif isinstance(text, list):
for t in text:
if isinstance(t, str):
t = t.encode("utf-8")
code.update(t)
else:
code.update(text)
else:
code.update(uuid.uuid4().bytes)
return code.hexdigest()
def json_to_str(obj: dict | list) -> str:
return json.dumps(obj, separators=(",", ":"))
|