Spaces:
Runtime error
Runtime error
Harsh Upadhyay
commited on
Commit
·
f654b35
1
Parent(s):
89a2809
added db connection error handling.
Browse files- backend/app/database.py +6 -1
backend/app/database.py
CHANGED
@@ -7,9 +7,14 @@ from sqlalchemy.sql import func
|
|
7 |
import os
|
8 |
from sqlalchemy.exc import IntegrityError
|
9 |
from werkzeug.security import check_password_hash, generate_password_hash
|
|
|
10 |
|
11 |
# SQLAlchemy setup
|
12 |
-
DATABASE_URL = os.environ.get('DATABASE_URL')
|
|
|
|
|
|
|
|
|
13 |
engine = create_engine(DATABASE_URL)
|
14 |
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
15 |
Base = declarative_base()
|
|
|
7 |
import os
|
8 |
from sqlalchemy.exc import IntegrityError
|
9 |
from werkzeug.security import check_password_hash, generate_password_hash
|
10 |
+
from backend.config import Config # Add this import
|
11 |
|
12 |
# SQLAlchemy setup
|
13 |
+
DATABASE_URL = os.environ.get('DATABASE_URL') or Config.SQLALCHEMY_DATABASE_URI
|
14 |
+
|
15 |
+
if not DATABASE_URL or DATABASE_URL.strip() == "":
|
16 |
+
raise ValueError("DATABASE_URL is not set or is empty. Please set it as an environment variable or secret.")
|
17 |
+
|
18 |
engine = create_engine(DATABASE_URL)
|
19 |
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
20 |
Base = declarative_base()
|