diff --git a/creyPY/fastapi/db/common.py b/creyPY/fastapi/db/common.py index 67c43dd..6c767fb 100644 --- a/creyPY/fastapi/db/common.py +++ b/creyPY/fastapi/db/common.py @@ -10,4 +10,6 @@ password = os.getenv("POSTGRES_PASSWORD", "root") port = os.getenv("POSTGRES_PORT", "5432") name = os.getenv("POSTGRES_DB", "fastapi") +ssl_mode = os.getenv("SSL_MODE", "require") + SQLALCHEMY_DATABASE_URL = f"postgresql+psycopg://{user}:{password}@{host}:{port}/" diff --git a/creyPY/fastapi/db/session.py b/creyPY/fastapi/db/session.py index 5ae5c6d..9a8979d 100644 --- a/creyPY/fastapi/db/session.py +++ b/creyPY/fastapi/db/session.py @@ -4,10 +4,10 @@ from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker from sqlalchemy.orm.session import Session -from .common import SQLALCHEMY_DATABASE_URL, name +from .common import SQLALCHEMY_DATABASE_URL, name, ssl_mode engine = create_engine( - SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True, connect_args={"sslmode": "require"} + SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True, connect_args={"sslmode": ssl_mode} ) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)