1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
import logging
from typing import NoReturn
import peewee
from peewee import SqliteDatabase
import inject
from cozy.db.book import Book
from cozy.db.settings import Settings as SettingsModel
from cozy.db.storage import Storage as StorageModel
from cozy.model.storage import InvalidPath, Storage
from cozy.report import reporter
log = logging.getLogger("model.storage_location")
class Settings:
_storages: list[Storage] = []
_db = inject.attr(SqliteDatabase)
def __init__(self):
self._db_object: SettingsModel = SettingsModel.get()
@property
def first_start(self) -> bool:
return self._db_object.first_start
@property
def last_played_book(self) -> Book | None:
try:
return self._db_object.last_played_book
except peewee.DoesNotExist:
log.warning(
"last_played_book references an non existent object. Setting last_played_book to None."
)
reporter.warning(
"settings_model",
"last_played_book references an non existent object. Setting last_played_book to None.",
)
self.last_played_book = None
return None
@last_played_book.setter
def last_played_book(self, new_value) -> None:
if new_value:
self._db_object.last_played_book = new_value._db_object
else:
self._db_object.last_played_book = None
self._db_object.save(
@property
def default_location(self) -> Storage | NoReturn:
for location in self.storage_locations:
if location.default:
return location
raise AssertionError("This should never happen")
@property
def storage_locations(self) -> list[Storage]:
if not self._storages:
self._load_all_storage_locations()
return self._storages
@property
def external_storage_locations(self) -> list[Storage]:
if not self._storages:
self._load_all_storage_locations()
return [storage for storage in self._storages if storage.external]
def invalidate(self) -> None:
self._storages.clear()
def _load_all_storage_locations(self) -> None:
self.invalidate()
for storage_db_obj in StorageModel.select(StorageModel.id):
try:
self._storages.append(Storage(self._db, storage_db_obj.id))
except InvalidPath:
log.error("Invalid path found in database, skipping: %s", storage_db_obj.path)
self._ensure_default_storage_is_present()
def _ensure_default_storage_is_present(self):
default_storage_present = any(storage.default for storage in self._storages)
if not default_storage_present and self._storages:
self._storages[0].default = True
|