qwen-api-fastapi / generate_token.py
convaiinnovations's picture
Upload folder using huggingface_hub
e2addc4 verified
Raw
History Blame Contribute Delete
606 Bytes
import jwt
import os
from datetime import datetime, timedelta
from dotenv import load_dotenv
load_dotenv()
SECRET_KEY = os.getenv("JWT_SECRET_KEY")
ALGORITHM = "HS256"
def create_token(user_id: str):
if not SECRET_KEY:
print("❌ Error: JWT_SECRET_KEY not found in .env")
return
payload = {
"sub": user_id,
"exp": datetime.utcnow() + timedelta(hours=24)
}
token = jwt.encode(payload, SECRET_KEY, algorithm=ALGORITHM)
print(f"✅ Token generated for {user_id}:\n{token}")
return token
if __name__ == "__main__":
create_token("admin_user")