Harsh Upadhyay commited on
Commit
f654b35
·
1 Parent(s): 89a2809

added db connection error handling.

Browse files
Files changed (1) hide show
  1. 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()