senju/senju/store_manager.py

from typing import Optional
from tinydb import TinyDB
from pathlib import Path
from logging import Logger

from tinydb.queries import QueryImpl

from senju.haiku import Haiku

DEFAULT_DB_PATH: Path = Path("/var/lib/senju.json")


class StoreManager:
    __slots__ = "_db", "logger"
    _db: TinyDB
    logger: Logger

    def __init__(self, path_to_db: Path = DEFAULT_DB_PATH) -> None:
        self._db = TinyDB(path_to_db)
        self.logger = Logger(__name__)

    def _query(self, query: QueryImpl) -> list[dict]:
        return self._db.search(query)

    def _load(self, id: int) -> Optional[dict]:
        try:
            return self._db.get(doc_id=id)
        except IndexError as e:
            self.logger.warning(f"could not get item with id {id}: {e}")
            return None

    def _save(self, data: dict) -> int:
        return self._db.insert(data)

    def load_haiku(self, key: int) -> Optional[Haiku]:
        raw_haiku: dict | None = self._load(key)
        if raw_haiku is None:
            return None
        h = Haiku(**raw_haiku)
        return h

    def save_haiku(self, data: Haiku) -> int:
        return self._save(data.__dict__)

    def get_id_of_latest_haiku(self) -> Optional[int]:
        try:
            id = self._db.all()[-1].doc_id
            return id
        except IndexError as e:
            self.logger.error(f"The database seems to be empty: {e}")
            return None