Ajit Panday commited on
Commit
9dd8976
·
1 Parent(s): 8514019

Fix database migration configuration

Browse files
Files changed (2) hide show
  1. app/database.py +6 -1
  2. start.sh +5 -0
app/database.py CHANGED
@@ -1,9 +1,14 @@
1
  from sqlalchemy import create_engine
2
  from sqlalchemy.orm import sessionmaker
3
  from .settings import DATABASE_URL
 
 
 
 
 
4
 
5
  # Create SQLAlchemy engine
6
- engine = create_engine(DATABASE_URL)
7
 
8
  # Create session factory
9
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
1
  from sqlalchemy import create_engine
2
  from sqlalchemy.orm import sessionmaker
3
  from .settings import DATABASE_URL
4
+ import os
5
+
6
+ def get_db_url():
7
+ """Get database URL from environment variables"""
8
+ return os.getenv('DATABASE_URL', DATABASE_URL)
9
 
10
  # Create SQLAlchemy engine
11
+ engine = create_engine(get_db_url())
12
 
13
  # Create session factory
14
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
start.sh CHANGED
@@ -1,5 +1,10 @@
1
  #!/bin/bash
2
 
 
 
 
 
 
3
  # Run database migrations
4
  cd app
5
  alembic upgrade head
 
1
  #!/bin/bash
2
 
3
+ # Load environment variables
4
+ if [ -f .env ]; then
5
+ export $(cat .env | grep -v '^#' | xargs)
6
+ fi
7
+
8
  # Run database migrations
9
  cd app
10
  alembic upgrade head