Script for token generation

tokens.py

# $ export JWT_SECRET=something
# $ python3 -m pip install pyjwt
import jwt, os
import datetime

SECRET = os.getenv("JWT_SECRET")
NOW = datetime.datetime.now(tz=datetime.timezone.utc)

def user_token(uid, role, seconds=60):
  dt = datetime.timedelta(seconds=seconds)
  return jwt.encode({"exp": NOW + dt, "uid":uid, "role":role, "aud":"grocery.shop"}, SECRET, algorithm="HS256")

def payment_token(payment, seconds=90):
  dt = datetime.timedelta(seconds=seconds)
  return jwt.encode({"exp": NOW + dt, "payment":payment, "aud":"grocery.shop"}, SECRET, algorithm="HS256")

print("== Admin token ===")
print(user_token(1, "admin", seconds=360), end="\n\n")

print("== User #1 token =")
print(user_token(101, "customer", seconds=360), end="\n\n")

print("== User #2 token =")
print(user_token(102, "customer", seconds=360), end="\n\n")

print("== Payment token =")
print(payment_token("DEAFBEE", seconds=360))