v0.6.8-rc4

Author: Aegis Stack

CLAUDE.md

@@ -32,7 +32,7 @@ Each generated project includes:
 
 ## Installation
 
-**Current Version**: 0.6.8rc3
+**Current Version**: 0.6.8rc4
 
 ```bash
 pip install aegis-stack

aegis/__init__.py

@@ -2,4 +2,4 @@
 Aegis Stack CLI - Component generation and project management tools.
 """
 
-__version__ = "0.6.8rc3"
+__version__ = "0.6.8rc4"

aegis/templates/copier-aegis-project/{{ project_slug }}/Makefile.jinja

@@ -189,11 +189,7 @@ migrate-history: ## Show migration history
 	@docker compose exec webserver uv run alembic -c alembic/alembic.ini history --verbose
 
 migrate-fix: ## Auto-fix schema mismatches after upgrade (safe: only adds, never drops)
-	@echo "Checking for schema mismatches..."
 	@docker compose exec webserver uv run python -m app.cli.migrate_fix
-	@echo "Applying fix migration..."
-	@docker compose exec webserver uv run alembic -c alembic/alembic.ini upgrade head
-	@echo "Schema fix complete"
 
 migrate-reset: ## Reset database (WARNING: destructive)
 	@echo "WARNING: This will destroy all data in the database!"

aegis/templates/copier-aegis-project/{{ project_slug }}/app/cli/migrate_fix.py.jinja

@@ -77,12 +77,9 @@ def _get_additive_diffs() -> list[tuple]:
                 additive_diffs.append(diff)
 
         elif op_type == "add_table":
-            # Missing tables need a full migration — too complex for auto-fix
             table = diff[1]
             if table.name in managed_tables:
-                print(
-                    t("migrate.table_missing_warning", table=table.name)
-                )
+                additive_diffs.append(diff)
 
         # Skip: add_index, remove_table, remove_column, remove_index,
         # modify_type, modify_nullable, modify_default, etc.
@@ -122,7 +119,45 @@ def _generate_fix_migration(diffs: list[tuple]) -> Path | None:
     for diff in diffs:
         op_type = diff[0]
 
-        if op_type == "add_column":
+        if op_type == "add_table":
+            table = diff[1]
+            # Build CREATE TABLE with columns
+            cols = []
+            for col in table.columns:
+                col_str = f"sa.Column('{col.name}', {_sa_type_str(col.type)}"
+                col_str += f", nullable={col.nullable}"
+                if col.primary_key:
+                    col_str += ", primary_key=True"
+                if col.server_default is not None:
+                    col_str += f", server_default=sa.text('{col.server_default.arg}')"
+                col_str += ")"
+                cols.append(f"        {col_str},")
+            cols_block = "\n".join(cols)
+            upgrade_lines.append(
+                f"    op.create_table(\n        '{table.name}',\n{cols_block}\n    )"
+            )
+            # Add foreign keys
+            for fk in table.foreign_key_constraints:
+                local_cols = [c.name for c in fk.columns]
+                ref_table = fk.referred_table.name
+                ref_cols = [c.name for c in fk.referred_table.primary_key.columns]
+                fk_name = fk.name or f"fk_{table.name}_{'_'.join(local_cols)}_{ref_table}"
+                upgrade_lines.append(
+                    f"    op.create_foreign_key('{fk_name}', '{table.name}', '{ref_table}', {local_cols}, {ref_cols})"
+                )
+                downgrade_lines.append(
+                    f"    op.drop_constraint('{fk_name}', '{table.name}', type_='foreignkey')"
+                )
+            # Add indexes
+            for idx in table.indexes:
+                idx_cols = [c.name for c in idx.columns]
+                unique_str = ", unique=True" if idx.unique else ""
+                upgrade_lines.append(
+                    f"    op.create_index('{idx.name}', '{table.name}', {idx_cols}{unique_str})"
+                )
+            downgrade_lines.append(f"    op.drop_table('{table.name}')")
+
+        elif op_type == "add_column":
             table_name = diff[2]
             column = diff[3]
             col_str = f"sa.Column('{column.name}', {_sa_type_str(column.type)}"
@@ -212,10 +247,23 @@ def main() -> None:
     for diff in diffs:
         if diff[0] == "add_column":
             print(f"  + {t('migrate.add_column')} {diff[2]}.{diff[3].name}")
+        elif diff[0] == "add_table":
+            print(f"  + {t('migrate.add_table')} {diff[1].name}")
 
     migration_path = _generate_fix_migration(diffs)
     if migration_path:
         print(f"\n{t('migrate.generated_migration', name=migration_path.name)}")
+
+        # Prompt to apply
+        apply = input(f"\n{t('migrate.apply_now')} [Y/n] ").strip().lower()
+        if apply in ("", "y", "yes"):
+            from alembic import command as alembic_command
+
+            alembic_cfg = Config("alembic/alembic.ini")
+            alembic_command.upgrade(alembic_cfg, "head")
+            print(t("migrate.applied_migration"))
+        else:
+            print(t("migrate.apply_later"))
     else:
         print(f"\n{t('migrate.no_fixable_differences')}")
         sys.exit(0)

aegis/templates/copier-aegis-project/{{ project_slug }}/app/i18n/locales/en.py

@@ -473,7 +473,11 @@
     "migrate.schema_up_to_date": "Schema is up to date. No fix needed.",
     "migrate.found_differences": "Found {count} schema differences:",
     "migrate.add_column": "ADD COLUMN",
+    "migrate.add_table": "ADD TABLE",
     "migrate.generated_migration": "Generated fix migration: {name}",
+    "migrate.apply_now": "Apply migration now?",
+    "migrate.applied_migration": "Migration applied successfully.",
+    "migrate.apply_later": "Migration saved. Run 'make migrate' to apply later.",
     "migrate.no_fixable_differences": "No fixable differences found.",
     # ── AI ───────────────────────────────────────────────────────────
     "ai.header_inline": "Illiana: ",

aegis/templates/copier-aegis-project/{{ project_slug }}/app/i18n/locales/zh.py

@@ -472,7 +472,11 @@
     "migrate.schema_up_to_date": "数据库结构已是最新，无需修复。",
     "migrate.found_differences": "发现 {count} 处结构差异：",
     "migrate.add_column": "新增列",
+    "migrate.add_table": "新增表",
     "migrate.generated_migration": "已生成修复迁移：{name}",
+    "migrate.apply_now": "是否立即执行迁移？",
+    "migrate.applied_migration": "迁移已成功执行",
+    "migrate.apply_later": "迁移已保存。可稍后运行 'make migrate' 执行",
     "migrate.no_fixable_differences": "没有可自动修复的差异。",
     # ── AI ───────────────────────────────────────────────────────────
     "ai.header_inline": "Illiana：",

copier.yml

@@ -6,7 +6,7 @@
 # - Update support
 
 _min_copier_version: "9.0.0"
-_version: "0.6.8rc3"
+_version: "0.6.8rc4"
 
 # IMPORTANT: Template content is in subdirectory
 # This allows the template to be recognized as git-tracked (aegis-stack repo root has .git)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aegis-stack"
-version = "0.6.8rc3"
+version = "0.6.8rc4"
 description = "A production-ready FastAPI platform with modular components and a built-in control plane. Try: uvx aegis-stack init my-project"
 readme = "README.md"
 requires-python = ">=3.11,<3.15"

tests/core/test_migrate_fix.py

@@ -195,3 +195,158 @@ def test_upgrade_scenario_basic_to_current(self) -> None:
         # Missing tables
         assert "password_reset_token" in added_table_names
         assert "email_verification_token" in added_table_names
+
+
+class TestMigrateFixMigrationRendering:
+    """Test that add_table diffs render correct migration content."""
+
+    def _render_add_table_upgrade(self, table: sa.Table) -> tuple[list[str], list[str]]:
+        """Simulate the migration rendering logic from migrate_fix.py."""
+
+        def _sa_type_str(col_type: object) -> str:
+            type_name = type(col_type).__name__
+            return f"sa.{type_name}()"
+
+        upgrade_lines: list[str] = []
+        downgrade_lines: list[str] = []
+
+        cols = []
+        for col in table.columns:
+            col_str = f"sa.Column('{col.name}', {_sa_type_str(col.type)}"
+            col_str += f", nullable={col.nullable}"
+            if col.primary_key:
+                col_str += ", primary_key=True"
+            if col.server_default is not None:
+                col_str += f", server_default=sa.text('{col.server_default.arg}')"  # type: ignore[union-attr]
+            col_str += ")"
+            cols.append(f"        {col_str},")
+        cols_block = "\n".join(cols)
+        upgrade_lines.append(
+            f"    op.create_table(\n        '{table.name}',\n{cols_block}\n    )"
+        )
+
+        for fk in table.foreign_key_constraints:
+            local_cols = [c.name for c in fk.columns]
+            ref_table = fk.referred_table.name
+            ref_cols = [c.name for c in fk.referred_table.primary_key.columns]
+            fk_name = fk.name or f"fk_{table.name}_{'_'.join(local_cols)}_{ref_table}"
+            upgrade_lines.append(
+                f"    op.create_foreign_key('{fk_name}', '{table.name}', '{ref_table}', {local_cols}, {ref_cols})"
+            )
+            downgrade_lines.append(
+                f"    op.drop_constraint('{fk_name}', '{table.name}', type_='foreignkey')"
+            )
+
+        for idx in table.indexes:
+            idx_cols = [c.name for c in idx.columns]
+            unique_str = ", unique=True" if idx.unique else ""
+            upgrade_lines.append(
+                f"    op.create_index('{idx.name}', '{table.name}', {idx_cols}{unique_str})"
+            )
+
+        downgrade_lines.append(f"    op.drop_table('{table.name}')")
+
+        return upgrade_lines, downgrade_lines
+
+    def test_renders_create_table(self) -> None:
+        """add_table diff renders op.create_table with all columns."""
+        metadata = sa.MetaData()
+        table = sa.Table(
+            "password_reset_token",
+            metadata,
+            sa.Column("id", sa.Integer(), primary_key=True),
+            sa.Column("user_id", sa.Integer(), nullable=False),
+            sa.Column("token", sa.String(), nullable=False),
+            sa.Column(
+                "used", sa.Boolean(), nullable=False, server_default=sa.text("0")
+            ),
+        )
+
+        upgrade, downgrade = self._render_add_table_upgrade(table)
+
+        # Should have create_table
+        create_line = upgrade[0]
+        assert "op.create_table(" in create_line
+        assert "'password_reset_token'" in create_line
+        assert "'id'" in create_line
+        assert "'user_id'" in create_line
+        assert "'token'" in create_line
+        assert "'used'" in create_line
+        assert "primary_key=True" in create_line
+        assert "server_default=sa.text('0')" in create_line
+
+        # Downgrade drops the table
+        assert "op.drop_table('password_reset_token')" in downgrade[-1]
+
+    def test_renders_foreign_keys(self) -> None:
+        """add_table diff with FKs renders op.create_foreign_key."""
+        metadata = sa.MetaData()
+        sa.Table(
+            "user",
+            metadata,
+            sa.Column("id", sa.Integer(), primary_key=True),
+        )
+        sa.Table(
+            "password_reset_token",
+            metadata,
+            sa.Column("id", sa.Integer(), primary_key=True),
+            sa.Column(
+                "user_id",
+                sa.Integer(),
+                sa.ForeignKey("user.id"),
+                nullable=False,
+            ),
+            sa.Column("token", sa.String(), nullable=False),
+        )
+
+        token_table = metadata.tables["password_reset_token"]
+        upgrade, downgrade = self._render_add_table_upgrade(token_table)
+
+        # Should have create_foreign_key
+        fk_lines = [line for line in upgrade if "create_foreign_key" in line]
+        assert len(fk_lines) == 1
+        assert "'password_reset_token'" in fk_lines[0]
+        assert "'user'" in fk_lines[0]
+        assert "['user_id']" in fk_lines[0]
+        assert "['id']" in fk_lines[0]
+
+        # Downgrade should drop the constraint
+        drop_fk_lines = [line for line in downgrade if "drop_constraint" in line]
+        assert len(drop_fk_lines) == 1
+        assert "type_='foreignkey'" in drop_fk_lines[0]
+
+    def test_renders_indexes(self) -> None:
+        """add_table diff with indexes renders op.create_index."""
+        metadata = sa.MetaData()
+        table = sa.Table(
+            "org_invite",
+            metadata,
+            sa.Column("id", sa.Integer(), primary_key=True),
+            sa.Column("email", sa.String(), nullable=False),
+            sa.Column("token", sa.String(), nullable=False),
+        )
+        sa.Index("ix_org_invite_token", table.c.token, unique=True)
+
+        upgrade, _ = self._render_add_table_upgrade(table)
+
+        idx_lines = [line for line in upgrade if "create_index" in line]
+        assert len(idx_lines) == 1
+        assert "'ix_org_invite_token'" in idx_lines[0]
+        assert "'org_invite'" in idx_lines[0]
+        assert "unique=True" in idx_lines[0]
+
+    def test_empty_table_renders(self) -> None:
+        """Table with only a PK column still renders correctly."""
+        metadata = sa.MetaData()
+        table = sa.Table(
+            "simple",
+            metadata,
+            sa.Column("id", sa.Integer(), primary_key=True),
+        )
+
+        upgrade, downgrade = self._render_add_table_upgrade(table)
+
+        assert "op.create_table(" in upgrade[0]
+        assert "'simple'" in upgrade[0]
+        assert len(upgrade) == 1  # No FKs or indexes
+        assert "op.drop_table('simple')" in downgrade[-1]