Spaces:
Sleeping
Sleeping
Ajit Panday commited on
Commit ·
9dd8976
1
Parent(s): 8514019
Fix database migration configuration
Browse files- app/database.py +6 -1
- 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(
|
| 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
|