The Monster Cache

from os import path, makedirs
from json import load, dump
from copy import deepcopy
from importlib.util import module_from_spec, spec_from_file_location

more_cfg = {
    'level_3': {
        'level_4': {
            'item3': 10,
            'item4': 20,
        }
    }
}


class Loader:
    @staticmethod
    def load(file):
        if path.exists(file):
            spec = spec_from_file_location("module.name", file)
            module = module_from_spec(spec)
            spec.loader.exec_module(module)
            return module.config

    @staticmethod
    def json(file):
        if file:
            with open(file) as f:
                return load(f)


class Cache:
    def __init__(self, *args, **kwargs):
        self.nodes = args[0] if args else kwargs
        self.loader = Loader()

    def get(self, uri=None):
        if not uri:
            return self.nodes

        def walk(_uri, nodes):
            parts = _uri.split('/', 1)
            key = parts.pop(0)

            if key in nodes:
                node = nodes[key]

                if not parts:
                    return node
                else:
                    return walk(parts[0], node)

        return walk(uri, self.nodes)

    def set(self, uri, *args, **kwargs):

        def walk(_uri, nodes):
            parts = _uri.split('/', 1)
            key = parts.pop(0)

            if key in nodes and parts:
                return walk(parts[0], nodes[key])
            elif len(_uri.split('/')) == 1:
                value = args[0] if args else kwargs
                nodes[key] = value

        return walk(uri, self.nodes)

    def dump(self, indent=None):
        """ Dumps the contents of the cache to the screen.
        The output from dump goes stdout and is used to view the cache contents.
        Default indentation is a dot for each level.
        :param indent:
            indent (str): String to be use for indenting levels.
        :return:
            Nothing.
        """
        indent = indent if indent else '.'

        print('-------------------------------------------------------------------------------------------------------')

        if self.nodes:
            def walk(_cfg, count):
                count += 1
                for key, value in _cfg.items():
                    if isinstance(value, dict):
                        print(indent * count, key)
                        walk(value, count)
                    else:
                        if isinstance(value, str):
                            value = f'"{value}"'
                        print(indent * count, key, value)
            walk(self.nodes, 0)
        else:
            print(' (No Data)')

        print('-------------------------------------------------------------------------------------------------------')

    def save(self, file=None):
        if file:
            dirname = path.dirname(file)

            if dirname and not path.exists(dirname):
                makedirs(dirname)

            with open(file, 'w') as f:
                dump(self.nodes, f, indent=3)

    def load(self, file=None):
        file_type = path.splitext(file)[1].lstrip('.').lower()

        if file_type == 'py':
            self.nodes = self.loader.load(file)
        if file_type == 'json':
            self.nodes = self.loader.json(file)

    def copy(self):
        return Cache(deepcopy(self.nodes))

    def remove(self, uri):
        """ Remove entree from cache.
        Removes an entree from the cache if it exists.
        :param uri:
            uri (str): URI that points to the entree to remove.
        :return:
            Nothing.
        """

        uri = uri.lstrip('/')
        if self.exists(uri):
            node = self.get('/'.join(uri.split('/')[:-1]))
            del node[uri.split('/')[-1]]

    def exists(self, uri):
        """ Test if URI exists in the cache.

        :param uri:
        :return:
        """
        return True if self.get(uri) else False

    def destroy(self):
        """ Destroy cache.
        Deletes all entries in the cache.
        :return:
            Nothing.
        """
        del self.nodes
        self.nodes = {}


c = Cache()
c.load('data.py')

print(c.get('level_1/level_2/item2'))

c.set('level_1/level_2/more_cfg', more_cfg)

if c.exists('level_1/level_2/more_cfg'):
    print(c.get('level_1/level_2/more_cfg'))

c.save('data.json')

cache_copy = c.copy()
cache_copy.dump()

c.remove('level_1/level_2/item1')
c.remove('level_1/level_2/item2')
c.dump()

cache_copy.dump()

my_cache = Cache()
my_cache.load('data.json')
my_cache.dump()