tiramisu/tiramisu/storage/cache/dictionary.py

# -*- coding: utf-8 -*-
# Copyright (C) 2018-2021 Team tiramisu (see AUTHORS for all contributors)
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by the
# Free Software Foundation, either version 3 of the License, or (at your
# option) any later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
# ____________________________________________________________


class Cache:
    __slots__ = ('_cache',)

    def __init__(self):
        self._cache = {}

    def setcache(self, path, index, val, time, validated):
        self._cache.setdefault(path, {})[index] = (val, int(time), validated)

    def getcache(self, path, index):
        values = self._cache.get(path)
        if values is None:
            return
        return values.get(index)

    def delcache(self, path):
        if path in  self._cache:
            del self._cache[path]

    def get_cached(self):
        return self._cache

    def reset_all_cache(self):
        self._cache.clear()