Add python script for applying migrations
This commit is contained in:
55
run_migration.py
Normal file
55
run_migration.py
Normal file
@@ -0,0 +1,55 @@
|
||||
"""
|
||||
Database migration runner
|
||||
Execute SQL migration files using the Flask app's database connection
|
||||
"""
|
||||
import os
|
||||
import sys
|
||||
|
||||
# Add parent directory to path to import app
|
||||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||||
|
||||
from extensions import db as database
|
||||
|
||||
def run_migration(migration_file):
|
||||
"""Execute a SQL migration file"""
|
||||
try:
|
||||
# Read migration file
|
||||
migration_path = os.path.join('migrations', migration_file)
|
||||
if not os.path.exists(migration_path):
|
||||
print(f"ERROR: Migration file not found: {migration_path}")
|
||||
sys.exit(1)
|
||||
|
||||
with open(migration_path, 'r') as f:
|
||||
sql = f.read()
|
||||
|
||||
# Execute migration using the database connection
|
||||
conn = database.pool.getconn()
|
||||
cursor = conn.cursor()
|
||||
|
||||
try:
|
||||
cursor.execute(sql)
|
||||
conn.commit()
|
||||
print(f"✓ Successfully executed migration: {migration_file}")
|
||||
except Exception as e:
|
||||
conn.rollback()
|
||||
print(f"ERROR executing migration: {e}")
|
||||
sys.exit(1)
|
||||
finally:
|
||||
cursor.close()
|
||||
database.pool.putconn(conn)
|
||||
|
||||
except Exception as e:
|
||||
print(f"ERROR: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) < 2:
|
||||
print("Usage: python run_migration.py <migration_file>")
|
||||
print("Example: python run_migration.py 003_create_login_history.sql")
|
||||
sys.exit(1)
|
||||
|
||||
# Initialize the app to set up database connection
|
||||
from app import app
|
||||
with app.app_context():
|
||||
migration_file = sys.argv[1]
|
||||
run_migration(migration_file)
|
||||
Reference in New Issue
Block a user