From 1f224c44bcaa103e1ba16b0eaaaec2d8aa28c749 Mon Sep 17 00:00:00 2001 From: Conrad Date: Mon, 27 Jan 2025 13:09:16 +0100 Subject: [PATCH] feat: added sslmode flag --- creyPY/fastapi/db/common.py | 2 ++ creyPY/fastapi/db/session.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) 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)