Source code for kibitzr.stash

import contextlib
import logging


logger = logging.getLogger(__name__)


class Stash:

    FILENAME = 'stash.db'

    @contextlib.contextmanager
    def open(self):
        # pylint: disable=import-outside-toplevel
        import shelve
        with contextlib.closing(shelve.open(self.FILENAME)) as db:
            yield db

    def read(self):
        with self.open() as db:
            return dict(db)

    def write(self, data):
        with self.open() as db:
            for key, value in data.items():
                db[key] = value

    @classmethod
    def print_content(cls):
        for key, value in cls().read().items():
            print(f"{key}: {value}")


class LazyStash(Stash):
    def __init__(self):
        self._stashobj = None

    @property
    def _stash(self):
        if self._stashobj is None:
            self._stashobj = self.read()
        return self._stashobj

    def __getitem__(self, key):
        return self._stash[key]

    def get(self, key, default=None):
        try:
            return self._stash[key]
        except KeyError:
            return default