Mateusz Gruszczyński
33 lines
850 B
Python
33 lines
850 B
Python
from __future__ import annotations
|
|
|
|
import fcntl
|
|
from contextlib import contextmanager
|
|
from pathlib import Path
|
|
|
|
from flask import Flask
|
|
|
|
from .database_init import initialize_database
|
|
from .database_migrations import migrate_database
|
|
|
|
|
|
@contextmanager
|
|
def _startup_lock(app: Flask):
|
|
lock_path = Path(app.instance_path) / ".database-startup.lock"
|
|
lock_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
with lock_path.open("w") as lock_file:
|
|
fcntl.flock(lock_file.fileno(), fcntl.LOCK_EX)
|
|
try:
|
|
yield
|
|
finally:
|
|
fcntl.flock(lock_file.fileno(), fcntl.LOCK_UN)
|
|
|
|
|
|
def prepare_database(app: Flask) -> list[str]:
|
|
with app.app_context(), _startup_lock(app):
|
|
initialize_database()
|
|
return migrate_database()
|
|
|
|
|
|
__all__ = ["initialize_database", "migrate_database", "prepare_database"]
|