mirror of
https://github.com/markbeep/AudioBookRequest.git
synced 2026-01-24 15:09:58 -06:00
abd75a96da
* update migrations to support postgres * fix uv lockfile * adjust migrations for psql * remove action on pr (push only) * remove pragmas for psql * update readme
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from contextlib import contextmanager
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlmodel import Session, text
|
|
|
|
from app.internal.env_settings import Settings
|
|
|
|
db = Settings().db
|
|
if db.use_postgres:
|
|
engine = create_engine(
|
|
f"postgresql://{db.postgres_user}:{db.postgres_password}@{db.postgres_host}:{db.postgres_port}/{db.postgres_db}?sslmode={db.postgres_ssl_mode}"
|
|
)
|
|
else:
|
|
sqlite_path = Settings().get_sqlite_path()
|
|
engine = create_engine(f"sqlite+pysqlite:///{sqlite_path}")
|
|
|
|
|
|
def get_session():
|
|
with Session(engine) as session:
|
|
if not Settings().db.use_postgres:
|
|
session.execute(text("PRAGMA foreign_keys=ON")) # pyright: ignore[reportDeprecated]
|
|
yield session
|
|
|
|
|
|
# TODO: couldn't get a single function to work with FastAPI and allow for session creation wherever
|
|
@contextmanager
|
|
def open_session():
|
|
with Session(engine) as session:
|
|
if not Settings().db.use_postgres:
|
|
session.execute(text("PRAGMA foreign_keys=ON")) # pyright: ignore[reportDeprecated]
|
|
yield session
|