Refactor migration script: streamline the database migration process by removing Alembic dependencies, updating logging for clarity, and initializing database tables directly. Enhance error handling for improved debugging during migration execution.

scripts/migrate.py

@@ -1,86 +1,42 @@
-"""Database migration script."""
+#!/usr/bin/env python3
+"""
+Database migration script for the TTS API.
+"""
 import os
 import sys
 import logging
-from alembic.config import Config
-from alembic import command
+from pathlib import Path
 
-# Configure logging first
+# Set up logging
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
 )
 logger = logging.getLogger(__name__)
 
-# --- Rely on PYTHONPATH set in Dockerfile --- 
-# project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-# if project_root not in sys.path:
-#     sys.path.insert(0, project_root)
-
-# Debug: Print relevant info
-logger.info(f"Current directory: {os.getcwd()}")
-logger.info(f"Python Path (sys.path): {sys.path}")
-
-models_db_path = "/app/app/models/database.py"
-logger.info(f"Checking existence of {models_db_path}: {os.path.exists(models_db_path)}")
-
-try:
-    # Direct import, assuming PYTHONPATH is correct
-    from app.models.database import Base
-    logger.info("Successfully imported Base from app.models.database")
-except ImportError as e:
-    logger.error(f"Error importing Base: {e}")
-    # Log directory contents for deeper debugging
-    logger.info("Contents of /app:")
-    if os.path.exists("/app"):
-        try:
-            logger.info(os.listdir("/app"))
-        except Exception as list_err:
-            logger.error(f"Could not list /app: {list_err}")
-    logger.info("Contents of /app/app:")
-    if os.path.exists("/app/app"):
-        try:
-            logger.info(os.listdir("/app/app"))
-        except Exception as list_err:
-            logger.error(f"Could not list /app/app: {list_err}")
-    logger.info("Contents of /app/app/models:")
-    if os.path.exists("/app/app/models"):
-        try:
-            logger.info(os.listdir("/app/app/models"))
-        except Exception as list_err:
-            logger.error(f"Could not list /app/app/models: {list_err}")
-    raise
-
 def run_migrations():
-    """Run database migrations using Alembic."""
+    """Run database migrations."""
     try:
-        # Project root inside the container is /app
-        project_root = "/app"
+        # Log current environment
+        logger.info(f"Current directory: {os.getcwd()}")
+        logger.info(f"Python Path (sys.path): {sys.path}")
         
-        # Create Alembic configuration using path inside container
-        alembic_cfg_path = os.path.join(project_root, "alembic.ini")
-        logger.info(f"Using Alembic config: {alembic_cfg_path}")
-        if not os.path.exists(alembic_cfg_path):
-            logger.error(f"Alembic config not found at {alembic_cfg_path}")
-            raise FileNotFoundError(f"Alembic config not found at {alembic_cfg_path}")
-            
-        alembic_cfg = Config(alembic_cfg_path)
+        # Import the database configuration
+        from app.models import init_db
         
-        # Set the script location relative to project root
-        script_location = os.path.join(project_root, "alembic")
-        alembic_cfg.set_main_option("script_location", script_location)
-        logger.info(f"Set Alembic script location to: {script_location}")
+        # Initialize database tables
+        init_db()
+        logger.info("Database tables created successfully")
         
-        # Run the migration
-        logger.info("Starting database migration...")
-        command.upgrade(alembic_cfg, "head")
-        logger.info("Database migration completed successfully.")
+        return True
         
+    except ImportError as e:
+        logger.error(f"Error importing database configuration: {e}")
+        return False
     except Exception as e:
-        logger.error(f"Error during database migration: {str(e)}")
-        import traceback
-        logger.error(traceback.format_exc())
-        raise
+        logger.error(f"Error running migrations: {e}")
+        return False
 
 if __name__ == "__main__":
-    run_migrations() 
+    success = run_migrations()
+    sys.exit(0 if success else 1)