Spaces:
Sleeping
Sleeping
Merge pull request #5 from RaphaelRIVIERE/feature/cicd
Browse filesfix: resolve CI test failure due to DB engine import-time crash
- app/db/session.py +10 -10
app/db/session.py
CHANGED
|
@@ -8,19 +8,19 @@ from sqlalchemy.orm import Session
|
|
| 8 |
|
| 9 |
load_dotenv()
|
| 10 |
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
)
|
| 17 |
-
|
| 18 |
-
engine = create_engine(DATABASE_URL)
|
| 19 |
-
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
| 20 |
|
| 21 |
def get_db() -> Generator[Session, None, None]:
|
|
|
|
|
|
|
| 22 |
db = SessionLocal()
|
| 23 |
try:
|
| 24 |
yield db
|
| 25 |
finally:
|
| 26 |
-
db.close()
|
|
|
|
| 8 |
|
| 9 |
load_dotenv()
|
| 10 |
|
| 11 |
+
def get_database_url() -> str:
|
| 12 |
+
return (
|
| 13 |
+
f"postgresql+psycopg2://"
|
| 14 |
+
f"{os.getenv('DB_USER')}:{os.getenv('DB_PASSWORD')}"
|
| 15 |
+
f"@{os.getenv('DB_HOST')}:{os.getenv('DB_PORT')}"
|
| 16 |
+
f"/{os.getenv('DB_NAME')}"
|
| 17 |
+
)
|
|
|
|
|
|
|
| 18 |
|
| 19 |
def get_db() -> Generator[Session, None, None]:
|
| 20 |
+
engine = create_engine(get_database_url())
|
| 21 |
+
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
| 22 |
db = SessionLocal()
|
| 23 |
try:
|
| 24 |
yield db
|
| 25 |
finally:
|
| 26 |
+
db.close()
|