extract.py

import json


def extract(hf_token):
    """
    Extract token, user, redirect, source, username and info from input hf_token.
    If hf_token is simple, it is the token itself.
    """
    info = {}  # a copy of hf_token in json format
    try:
        info = dict(json.loads(hf_token))
    except json.decoder.JSONDecodeError:
        return hf_token, None, None, None, None,None
    token = info.get("token", None)
    user = info.get("user", None)
    redirect = info.get("redirect", None)
    source = info.get("source", None)
    username = info.get("username", None)

    return token, user, redirect, source, username, info