mirror of
https://github.com/enpaul/kodak.git
synced 2024-11-13 01:56:56 +00:00
Add initial database schema and infrastructure components
This commit is contained in:
parent
474d846f47
commit
e265dda12d
58
imagemonk/database/__init__.py
Normal file
58
imagemonk/database/__init__.py
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
import logging
|
||||||
|
from typing import Tuple
|
||||||
|
|
||||||
|
import peewee
|
||||||
|
|
||||||
|
from imagemonk import constants
|
||||||
|
from imagemonk.configuration import ImageMonkConfig
|
||||||
|
from imagemonk.database._shared import ImageMonkModel
|
||||||
|
from imagemonk.database._shared import INTERFACE as interface
|
||||||
|
from imagemonk.database.image import ImageRecord
|
||||||
|
from imagemonk.database.thumbnail import ThumbnailRecord
|
||||||
|
|
||||||
|
|
||||||
|
MODELS: Tuple[ImageMonkModel, ...] = (ImageRecord, ThumbnailRecord)
|
||||||
|
|
||||||
|
|
||||||
|
def initialize(config: ImageMonkConfig):
|
||||||
|
"""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.SupportedDatabaseBackend.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.SupportedDatabaseBackend.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 ValueError(
|
||||||
|
f"Invalid storage backend in configuration: {config.database.backend}"
|
||||||
|
)
|
||||||
|
|
||||||
|
interface.initialize(database)
|
||||||
|
|
||||||
|
with interface.atomic():
|
||||||
|
interface.create_tables(MODELS)
|
15
imagemonk/database/_shared.py
Normal file
15
imagemonk/database/_shared.py
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
import datetime
|
||||||
|
import uuid
|
||||||
|
|
||||||
|
import peewee
|
||||||
|
|
||||||
|
|
||||||
|
INTERFACE = peewee.DatabaseProxy()
|
||||||
|
|
||||||
|
|
||||||
|
class ImageMonkModel(peewee.Model):
|
||||||
|
class Meta: # pylint: disable=too-few-public-methods,missing-class-docstring
|
||||||
|
database = INTERFACE
|
||||||
|
|
||||||
|
uuid = peewee.UUIDField(null=False, unique=True, default=uuid.uuid4)
|
||||||
|
created = peewee.DateTimeField(null=False, default=datetime.datetime.utcnow)
|
30
imagemonk/database/image.py
Normal file
30
imagemonk/database/image.py
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
import json
|
||||||
|
import uuid
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
import peewee
|
||||||
|
|
||||||
|
from imagemonk.database._shared import ImageMonkModel
|
||||||
|
|
||||||
|
|
||||||
|
class ImageRecord(ImageMonkModel):
|
||||||
|
"""Database record for"""
|
||||||
|
|
||||||
|
width = peewee.IntegerField(null=False)
|
||||||
|
height = peewee.IntegerField(null=False)
|
||||||
|
format = peewee.CharField(null=False)
|
||||||
|
deleted = peewee.BooleanField(null=False, default=False)
|
||||||
|
public = peewee.BooleanField(null=False, default=False)
|
||||||
|
owner = peewee.UUIDField(null=False)
|
||||||
|
sha256 = peewee.CharField(null=False)
|
||||||
|
_readable = peewee.CharField(null=False, default="[]")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def readable(self) -> List[uuid.UUID]:
|
||||||
|
"""List of UUIDs corresponding to accounts that can read the file"""
|
||||||
|
return [uuid.UUID(item) for item in json.loads(self._readable)]
|
||||||
|
|
||||||
|
@readable.setter
|
||||||
|
def readable(self, value: List[uuid.UUID]):
|
||||||
|
"""Update the list of UUIDs for accounts that can read the file"""
|
||||||
|
self._readable = json.dumps([str(item) for item in value])
|
11
imagemonk/database/thumbnail.py
Normal file
11
imagemonk/database/thumbnail.py
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
import peewee
|
||||||
|
|
||||||
|
from imagemonk.database._shared import ImageMonkModel
|
||||||
|
from imagemonk.database.image import ImageRecord
|
||||||
|
|
||||||
|
|
||||||
|
class ThumbnailRecord(ImageMonkModel):
|
||||||
|
|
||||||
|
parent = peewee.ForeignKeyField(ImageRecord)
|
||||||
|
width = peewee.IntegerField(null=False)
|
||||||
|
height = peewee.IntegerField(null=False)
|
Loading…
Reference in New Issue
Block a user