helper.py

import os
import json
import shutil
from datetime import datetime
from jwcrypto.jwt import JWT
from jwcrypto.jwk import JWK
import base64
from math import floor
import uuid
import sys

def rmDir(path):
  if os.path.exists(path):
    shutil.rmtree(path)

def copyDir(src, dst):
  if not os.path.exists(dst):
    os.makedirs(dst)
  for item in os.listdir(src):
    s = os.path.join(src, item)
    d = os.path.join(dst, item)
    if os.path.isdir(s):
      copyDir(s, d)
    else:
      shutil.copy2(s, d)

def readJson(path):
  with open(path, 'r') as f:
    return json.load(f)

def saveJson(path, data):
  with open(path, 'w') as f:
    json.dump(data, f, indent=3)

def generateToken(issuer, key):
  timestamp = floor(datetime.utcnow().timestamp())
  data = {
    'iss': issuer,
    'iat': timestamp,
    'exp': timestamp + 300,
    'jti': str(uuid.uuid4())
  }
  key_b64 = (
    base64
    .urlsafe_b64encode(key.encode('ascii'))
    .decode()
  )
  jwk = JWK(
    k=key_b64,
    kty='oct'
  )
  token = JWT(header={'alg': 'HS256'}, claims=data)
  token.make_signed_token(jwk)
  return token.serialize()

def checkResponse(response):
  try:
    response.raise_for_status()
  except:
    print(response.json(), file=sys.stderr)
    raise