cache.py

import os
import re
import pathlib
import gzip
import json
import tbapy.models

cache_dir = 'cache/'
fn_dir = 'fn/'
api_dir = 'api/'


# https://github.com/django/django/blob/master/django/utils/text.py#L219
def get_valid_filename(s):
    s = str(s).strip().replace(' ', '_')
    return re.sub(r'(?u)[^-\w.]', '', s)


def call(fn, refresh=False, *args, **kwargs):
    filepath = f'{cache_dir}{fn_dir}{get_valid_filename(fn.__name__)}_{get_valid_filename(str(args))}_' \
        f'{get_valid_filename(kwargs)}'
    # pathlib.Path(filepath).mkdir(parents=True, exist_ok=True)

    if os.path.exists(f'{filepath}.json.gz') and not refresh:
        with gzip.open(f'{filepath}.json.gz', 'rb') as f:
            content = json.loads(f.read())

        return content
    else:
        data = fn(*args, **kwargs)
        with gzip.open(f'{filepath}.json.gz', 'wb+') as f:
            f.write(json.dumps(data).encode())

        if isinstance(data, tbapy.models._base_model_class):
            data = dict(data)

        return data