SyncAPI/modules/utils.py

from os import makedirs, path
from typing import Any, Tuple, Union
from uuid import uuid4
from ujson import loads, dumps, JSONDecodeError
from traceback import print_exc

# Print to stdout and then to log
def logWrite(message: str, debug: bool = False) -> None:
    # save to log file and rotation is to be done
    # logAppend(f'{message}', debug=debug)
    print(f"{message}", flush=True)

def jsonLoad(filepath: str) -> Any:
    """Load json file

    ### Args:
        * filepath (`str`): Path to input file

    ### Returns:
        * `Any`: Some json deserializable
    """    
    with open(filepath, "r", encoding='utf8') as file:
        try:
            output = loads(file.read())
        except JSONDecodeError:
            logWrite(f"Could not load json file {filepath}: file seems to be incorrect!\n{print_exc()}")
            raise
        except FileNotFoundError:
            logWrite(f"Could not load json file {filepath}: file does not seem to exist!\n{print_exc()}")
            raise
        file.close()
    return output

def jsonSave(contents: Union[list, dict], filepath: str) -> None:
    """Save contents into json file

    ### Args:
        * contents (`Union[list, dict]`): Some json serializable
        * filepath (`str`): Path to output file
    """    
    try:
        with open(filepath, "w", encoding='utf8') as file:
            file.write(dumps(contents, ensure_ascii=False, indent=4))
            file.close()
    except Exception as exp:
        logWrite(f"Could not save json file {filepath}: {exp}\n{print_exc()}")
    return

def configGet(key: str, *args: str) -> Any:
    """Get value of the config key

    ### Args:
        * key (`str`): The last key of the keys path.
        * *args (`str`): Path to key like: dict[args][key].

    ### Returns:
        * `Any`: Value of provided key
    """    
    this_dict = jsonLoad("config.json")
    this_key = this_dict
    for dict_key in args:
        this_key = this_key[dict_key]
    return this_key[key]

def saveFile(filebytes: bytes, filename: Union[str, None] = None, dirname: Union[str, None] = None) -> Tuple[str, str]:
    """Save some bytedata into random file and return its ID

    ### Args:
        * filebytes (`bytes`): Bytes to write into file

    ### Returns:
        * `Tuple[str, str]`: Tuple where first item is an ID and the second is an absolute path to file
    """    
    pathlist = [configGet("data", "locations"), "files"]
    if dirname is not None:
        pathlist.append(dirname)
    makedirs(path.join(pathlist), exist_ok=True)
    filename = str(uuid4()) if filename is None else filename
    with open(path.join(pathlist+[filename], "wb")) as file:
        file.write(filebytes)
    return filename, path.join(pathlist+[filename])