rriviere commited on
Commit
70df73c
·
unverified ·
2 Parent(s): d4b25ae36fe10e

Merge pull request #5 from RaphaelRIVIERE/feature/cicd

Browse files

fix: resolve CI test failure due to DB engine import-time crash

Files changed (1) hide show
  1. 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
- DATABASE_URL = (
12
- f"postgresql+psycopg2://"
13
- f"{os.getenv('DB_USER')}:{os.getenv('DB_PASSWORD')}"
14
- f"@{os.getenv('DB_HOST')}:{os.getenv('DB_PORT')}"
15
- f"/{os.getenv('DB_NAME')}"
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()