mirror of
https://github.com/enpaul/kodak.git
synced 2024-11-14 10:36:55 +00:00
126 lines
3.6 KiB
Python
126 lines
3.6 KiB
Python
import json
|
|
import os
|
|
from dataclasses import dataclass
|
|
from dataclasses import field
|
|
from pathlib import Path
|
|
from typing import Any
|
|
from typing import Dict
|
|
from typing import Optional
|
|
from typing import Tuple
|
|
|
|
from dehance import constants
|
|
|
|
|
|
def _default_sqlite_pragmas() -> Dict[str, Any]:
|
|
"""Default SQLite pragmas
|
|
|
|
.. note:: This needs to be a separate callout function to avoid setting a mutable type in the
|
|
dataclass signature.
|
|
|
|
.. note:: These settings are taken directly from
|
|
`the Peewee docs <http://docs.peewee-orm.com/en/latest/peewee/database.html#recommended-settings>`_.
|
|
"""
|
|
return {
|
|
"journal_mode": "wal",
|
|
"cache_size": -1 * 64000,
|
|
"foreign_keys": 1,
|
|
"ignore_check_constraints": 0,
|
|
"synchronous": 0,
|
|
}
|
|
|
|
|
|
@dataclass
|
|
class _DBSqliteConfig:
|
|
path: Path = Path.cwd() / "dehance.db"
|
|
pragmas: Dict[str, Any] = field(default_factory=_default_sqlite_pragmas)
|
|
|
|
@classmethod
|
|
def build(cls):
|
|
return cls(
|
|
path=Path(os.environ.get(constants.ENV_CONF_DB_SQLITE_PATH, cls.path)),
|
|
pragmas=json.loads(os.environ[constants.ENV_CONF_DB_SQLITE_PRAGMAS])
|
|
if constants.ENV_CONF_DB_SQLITE_PRAGMAS in os.environ
|
|
else _default_sqlite_pragmas(),
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class _DBMariaConfig:
|
|
|
|
hostname: str = "localhost"
|
|
username: str = "root"
|
|
password: Optional[str] = None
|
|
port: int = 3306
|
|
schema: str = "dehance"
|
|
|
|
@classmethod
|
|
def build(cls):
|
|
return cls(
|
|
hostname=os.getenv(constants.ENV_CONF_DB_MARIA_HOSTNAME, cls.hostname),
|
|
username=os.getenv(constants.ENV_CONF_DB_MARIA_USERNAME, cls.username),
|
|
password=os.environ.get(constants.ENV_CONF_DB_MARIA_PASSWORD, cls.password),
|
|
port=int(os.environ.get(constants.ENV_CONF_DB_MARIA_PORT, cls.port)),
|
|
schema=os.getenv(constants.ENV_CONF_DB_MARIA_SCHEMA, cls.schema),
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class _DBConfig:
|
|
|
|
backend: constants.SupportedDatabaseBackend = (
|
|
constants.SupportedDatabaseBackend.SQLITE
|
|
)
|
|
sqlite: _DBSqliteConfig = field(default_factory=_DBSqliteConfig.build)
|
|
mariadb: _DBMariaConfig = field(default_factory=_DBMariaConfig.build)
|
|
|
|
@classmethod
|
|
def build(cls):
|
|
return cls(
|
|
backend=constants.SupportedDatabaseBackend[
|
|
os.environ[constants.ENV_CONF_DB_BACKEND]
|
|
]
|
|
if constants.ENV_CONF_DB_BACKEND in os.environ
|
|
else cls.backend
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class _UploadConfig:
|
|
|
|
size_limit: int = 1 * 1024 * 1024
|
|
formats: Tuple[str] = ("jpg", "jpeg")
|
|
|
|
@classmethod
|
|
def build(cls):
|
|
return cls(
|
|
size_limit=(int(os.environ[constants.ENV_CONF_FS_UPLOAD_MAX_SIZE]) * 1024)
|
|
if constants.ENV_CONF_FS_UPLOAD_MAX_SIZE in os.environ
|
|
else cls.size_limit,
|
|
formats=(
|
|
item.strip().lower()
|
|
for item in os.environ[constants.ENV_CONF_FS_UPLOAD_FORMATS].split(",")
|
|
)
|
|
if constants.ENV_CONF_FS_UPLOAD_MAX_SIZE in os.environ
|
|
else cls.formats,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class DehanceConfig:
|
|
database: _DBConfig = field(default_factory=_DBConfig.build)
|
|
upload: _UploadConfig = field(default_factory=_UploadConfig.build)
|
|
storage_path: Path = Path.cwd()
|
|
|
|
@classmethod
|
|
def build(cls):
|
|
return cls(
|
|
storage_path=Path(
|
|
os.getenv(constants.ENV_CONF_FS_STORAGE_PATH, cls.storage_path)
|
|
).resolve()
|
|
)
|
|
|
|
|
|
def load() -> DehanceConfig:
|
|
|
|
return DehanceConfig.build()
|