mirror of
https://github.com/senju1337/senju.git
synced 2025-12-24 07:39:29 +00:00
feat: add get_latest_haiku_or_default to store
Resf: OPS-65
This commit is contained in:
parent
2c7394edfa
commit
69db48af8b
2 changed files with 15 additions and 1 deletions
|
|
@ -59,3 +59,7 @@ class Haiku:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
return haiku
|
return haiku
|
||||||
|
|
||||||
|
|
||||||
|
DEFAULT_HAIKU: Haiku = Haiku(["Purple petals rise", "Defying fragile beauty",
|
||||||
|
"Fiercely breathing life"])
|
||||||
|
|
|
||||||
|
|
@ -7,7 +7,8 @@ from typing import Optional
|
||||||
from tinydb import TinyDB
|
from tinydb import TinyDB
|
||||||
from tinydb.queries import QueryImpl
|
from tinydb.queries import QueryImpl
|
||||||
|
|
||||||
from senju.haiku import Haiku
|
from senju import haiku
|
||||||
|
from senju.haiku import DEFAULT_HAIKU, Haiku
|
||||||
|
|
||||||
DEFAULT_DB_PATH: Path = Path("/var/lib/senju.json")
|
DEFAULT_DB_PATH: Path = Path("/var/lib/senju.json")
|
||||||
|
|
||||||
|
|
@ -51,3 +52,12 @@ class StoreManager:
|
||||||
except IndexError as e:
|
except IndexError as e:
|
||||||
self.logger.error(f"The database seems to be empty: {e}")
|
self.logger.error(f"The database seems to be empty: {e}")
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
def get_latest_haiku_or_default(self) -> Haiku:
|
||||||
|
id = self.get_id_of_latest_haiku()
|
||||||
|
if id is None:
|
||||||
|
return DEFAULT_HAIKU
|
||||||
|
haiku = self.load_haiku(id)
|
||||||
|
if haiku is None:
|
||||||
|
return DEFAULT_HAIKU
|
||||||
|
return haiku
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue