wireguard-manager-gui/wg_dashboard_backend/database.py

import sqlalchemy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
import const

engine = sqlalchemy.create_engine(
    const.DATABASE_URL, connect_args={"check_same_thread": False}
)


SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

Base = declarative_base()

from const import DATABASE_URL
from migrate import DatabaseAlreadyControlledError
from migrate.versioning.shell import main
try:
    main(["version_control", DATABASE_URL, "migrations"])
except DatabaseAlreadyControlledError:
    pass
main(["upgrade", DATABASE_URL, "migrations"])