Spaces:
Sleeping
Sleeping
Aurélie GABU commited on
Commit ·
9d1c071
1
Parent(s): 009706f
denying CI to be connected to DB
Browse files- App/database.py +11 -2
App/database.py
CHANGED
|
@@ -5,6 +5,9 @@ from sqlalchemy.orm import sessionmaker, declarative_base
|
|
| 5 |
|
| 6 |
load_dotenv()
|
| 7 |
|
|
|
|
|
|
|
|
|
|
| 8 |
DB_USER = os.getenv("DB_USER", "postgres")
|
| 9 |
DB_PASSWORD = os.getenv("DB_PASSWORD", "password")
|
| 10 |
DB_HOST = os.getenv("DB_HOST", "localhost")
|
|
@@ -13,6 +16,12 @@ DB_NAME = os.getenv("DB_NAME", "test_db")
|
|
| 13 |
|
| 14 |
DATABASE_URL = (f"postgresql+psycopg2://{DB_USER}:{DB_PASSWORD}"f"@{DB_HOST}:{DB_PORT}/{DB_NAME}")
|
| 15 |
|
| 16 |
-
engine = create_engine(DATABASE_URL)
|
| 17 |
-
SessionLocal = sessionmaker(autocommit = False, autoflush = False, bind = engine)
|
| 18 |
Base = declarative_base()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
|
| 6 |
load_dotenv()
|
| 7 |
|
| 8 |
+
# Détection si on est en train d'exécuter pytest
|
| 9 |
+
IS_TEST = os.getenv("PYTEST_CURRENT_TEST") is not None
|
| 10 |
+
|
| 11 |
DB_USER = os.getenv("DB_USER", "postgres")
|
| 12 |
DB_PASSWORD = os.getenv("DB_PASSWORD", "password")
|
| 13 |
DB_HOST = os.getenv("DB_HOST", "localhost")
|
|
|
|
| 16 |
|
| 17 |
DATABASE_URL = (f"postgresql+psycopg2://{DB_USER}:{DB_PASSWORD}"f"@{DB_HOST}:{DB_PORT}/{DB_NAME}")
|
| 18 |
|
|
|
|
|
|
|
| 19 |
Base = declarative_base()
|
| 20 |
+
|
| 21 |
+
if not IS_TEST:
|
| 22 |
+
engine = create_engine(DATABASE_URL)
|
| 23 |
+
SessionLocal = sessionmaker(autocommit = False, autoflush = False, bind = engine)
|
| 24 |
+
|
| 25 |
+
else:
|
| 26 |
+
engine = None
|
| 27 |
+
SessionLocal = None
|