mirror of https://github.com/enpaul/kodak.git
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
import logging
|
|
from typing import Tuple
|
|
from typing import Type
|
|
|
|
import peewee
|
|
|
|
from kodak import constants
|
|
from kodak import exceptions
|
|
from kodak.configuration import KodakConfig
|
|
from kodak.database._shared import INTERFACE as interface
|
|
from kodak.database._shared import KodakModel
|
|
from kodak.database.access import AccessRecord
|
|
from kodak.database.alias import AliasRecord
|
|
from kodak.database.image import ImageRecord
|
|
|
|
|
|
MODELS: Tuple[Type[KodakModel], ...] = (ImageRecord, AliasRecord, AccessRecord)
|
|
|
|
|
|
def initialize(config: KodakConfig):
|
|
"""Initialize the database interface
|
|
|
|
Defining the database as an
|
|
`unconfigured proxy object <http://docs.peewee-orm.com/en/latest/peewee/database.html#setting-the-database-at-run-time>`_
|
|
allows it to be configured at runtime based on the config values.
|
|
|
|
:param config: Populated configuration container object
|
|
"""
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
if config.database.backend == constants.DatabaseBackend.SQLITE:
|
|
logger.debug("Using SQLite database backend")
|
|
logger.debug(f"Applying SQLite pragmas: {config.database.sqlite.pragmas}")
|
|
database = peewee.SqliteDatabase(
|
|
config.database.sqlite.path, pragmas=config.database.sqlite.pragmas
|
|
)
|
|
elif config.database.backend == constants.DatabaseBackend.MARIADB:
|
|
logger.debug("Using MariaDB database backend")
|
|
logger.debug(
|
|
"Configuring MariaDB:"
|
|
f" {config.database.mariadb.username}@{config.database.mariadb.hostname}:{config.database.mariadb.port},"
|
|
f" with database '{config.database.mariadb.schema}'"
|
|
)
|
|
database = peewee.MySQLDatabase(
|
|
config.database.mariadb.schema,
|
|
host=config.database.mariadb.hostname,
|
|
port=config.database.mariadb.port,
|
|
user=config.database.mariadb.username,
|
|
password=config.database.mariadb.password,
|
|
charset="utf8mb4",
|
|
)
|
|
else:
|
|
raise exceptions.ConfigurationError(
|
|
f"Invalid storage backend in configuration: {config.database.backend}"
|
|
)
|
|
|
|
interface.initialize(database)
|
|
|
|
with interface.atomic():
|
|
interface.create_tables(MODELS)
|