refactor(migration): improve error handling and type safety in migration module

- Replace generic error decorator with typed migration-specific error handling
- Add proper type hints and validation for migration operations
- Rename database dependency key from "database" to "db" for consistency
- Remove redundant db_migration option from CLI
- Improve error messages and logging for migration operations

Author: Charlie-BU

pyproject.toml

@@ -35,7 +35,7 @@ dependencies = [
 
 [project.optional-dependencies]
 "templating" = ["jinja2 == 3.0.1"]
-"database" = ["alembic >= 1.16.3"]
+"db" = ["alembic >= 1.16.3"]
 
 [project.urls]
 Documentation = "https://robyn.tech/"
@@ -86,7 +86,7 @@ orjson = "^3.9.15"
 
 [tool.poetry.extras]
 templating = ["jinja2"]
-database = ["alembic"]
+db = ["alembic"]
 
 [tool.poetry.group.dev]
 optional = true

robyn/cli.py

@@ -4,14 +4,17 @@
 import sys
 import webbrowser
 import argparse
+import importlib.util
 from pathlib import Path
 from typing import Optional
+from pip._internal.cli.main import main as pip_main
 
 from InquirerPy.base.control import Choice
 from InquirerPy.resolver import prompt
 
 from robyn.env_populator import load_vars
 from robyn.robyn import get_version
+from robyn.migrate import configure_parser, execute_command
 
 from .argument_parser import Config
 from .reloader import create_rust_file, setup_reloader
@@ -52,21 +55,10 @@ def create_robyn_app():
             "default": Choice("no-db", name="No DB"),
             "name": "project_type",
         },
-        {
-            "type": "list",
-            "message": "Need Database Migration? (Y/N)",
-            "choices": [
-                Choice("Y", name="Y"),
-                Choice("N", name="N"),
-            ],
-            "default": Choice("N", name="N"),
-            "name": "db_migration",
-        },
     ]
     result = prompt(questions=questions)
     project_dir_path = Path(str(result["directory"])).resolve()
     docker = result["docker"]
-    db_migration = result["db_migration"]
     project_type = str(result["project_type"])
 
     final_project_dir_path = (CURRENT_WORKING_DIR / project_dir_path).resolve()
@@ -83,34 +75,6 @@ def create_robyn_app():
     if docker == "N":
         os.remove(f"{final_project_dir_path}/Dockerfile")
 
-    # If database migration is needed, install alembic
-    if db_migration == "Y":
-        print("Installing alembic...")
-        try:
-            # Check if alembic is already installed
-            import importlib.util
-            alembic_spec = importlib.util.find_spec('alembic')
-            
-            if alembic_spec is None:
-                # Install alembic using pip API
-                try:
-                    import pip
-                    print("Installing alembic using pip API...")
-                    from pip._internal.cli.main import main as pip_main
-                    pip_main(['install', 'alembic', '--quiet'])
-                    print("Successfully installed alembic.")
-                except ImportError:
-                    # If pip API is not available, use subprocess
-                    print("Installing alembic using subprocess...")
-                    subprocess.run([sys.executable, "-m", "pip", "install", "alembic", "-q"], check=True,
-                                  stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-                    print("Successfully installed alembic.")
-            else:
-                print("Alembic is already installed.")
-        except (subprocess.CalledProcessError, ImportError) as e:
-            print(f"Failed to install alembic: {str(e)}")
-            print("Please install it manually using 'pip install alembic'.")
-
     print(f"New Robyn project created in '{final_project_dir_path}' ")
 
 
@@ -143,7 +107,6 @@ def start_app_normally(config: Config):
 
 def handle_db_command():
     """Handle database migration commands."""
-    import importlib.util
     alembic_spec = importlib.util.find_spec("alembic")
 
     if alembic_spec is None:
@@ -153,7 +116,6 @@ def handle_db_command():
         if install_choice == 'y':
             try:
                 try:
-                    from pip._internal.cli.main import main as pip_main
                     print("Installing alembic...")
                     pip_main(['install', 'alembic', '--quiet'])
                     print("Successfully installed alembic.")
@@ -175,13 +137,6 @@ def handle_db_command():
             print("Please install alembic manually using 'pip install alembic' before using database commands.")
             sys.exit(1)
 
-    try:
-        from robyn.migrate import configure_parser, execute_command
-    except ImportError as e:
-        print(f"ERROR: Failed to import migrate module: {str(e)}")
-        print("This might be due to an incomplete installation or a version mismatch.")
-        print("Try reinstalling Robyn or updating your dependencies.")
-        sys.exit(1)
     parser = argparse.ArgumentParser(
         usage=argparse.SUPPRESS,  # omit usage hint
         description='Robyn database migration commands.'