From d75fede3d1b7dfbd90f588f9470d0de95529a4a0 Mon Sep 17 00:00:00 2001 From: vikynoah Date: Thu, 2 Jan 2025 23:20:17 +0100 Subject: [PATCH] fix: Force postgresql SSL mode (#27) * fix: Force postgresql SSL mode * changes --- creyPY/fastapi/db/async_session.py | 5 ++++- creyPY/fastapi/db/session.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/creyPY/fastapi/db/async_session.py b/creyPY/fastapi/db/async_session.py index 56ed9ae..55d2c5a 100644 --- a/creyPY/fastapi/db/async_session.py +++ b/creyPY/fastapi/db/async_session.py @@ -5,7 +5,10 @@ from sqlalchemy.orm import sessionmaker from .common import SQLALCHEMY_DATABASE_URL, name -async_engine = create_async_engine(SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True) +async_engine = create_async_engine(SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True, + connect_args={ + 'sslmode': 'require' + }) AsyncSessionLocal = sessionmaker( bind=async_engine, diff --git a/creyPY/fastapi/db/session.py b/creyPY/fastapi/db/session.py index b400468..7c746ed 100644 --- a/creyPY/fastapi/db/session.py +++ b/creyPY/fastapi/db/session.py @@ -6,7 +6,10 @@ from sqlalchemy.orm.session import Session from .common import SQLALCHEMY_DATABASE_URL, name -engine = create_engine(SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True) +engine = create_engine(SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True, + connect_args={ + 'sslmode': 'require' + }) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)