extract.py

import json


def extract(info_str):
    """Extract user, source, username, and all info"""
    info = {}  # a copy of hf_token in json format
    try:
        info = dict(json.loads(info_str))
    except json.decoder.JSONDecodeError:
        return "__fake__", None, None, None
    user = info.get("user", "__fake__")
    source = info.get("source", None)
    username = info.get("username", None)
    return user, source, username, info