inputlayer-client-dev 0.1.0.dev915__tar.gz → 0.1.0.dev919__tar.gz

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: inputlayer-client-dev
-Version: 0.1.0.dev915
+Version: 0.1.0.dev919
 Summary: Python Object-Logic Mapper for InputLayer knowledge graph engine
 License-Expression: Apache-2.0
 License-File: LICENSE
@@ -51,7 +51,7 @@ pip install inputlayer-client-dev[all]         # everything
 
 Requirements: Python 3.10+ and a running InputLayer server.
 
-This also installs the `il` CLI for managing schema migrations.
+This also installs the `inputlayer-migrate` tool for schema migrations; with the InputLayer product CLI installed, use it as `il migration <verb>`.
 
 ## Quick Start
 
@@ -425,16 +425,16 @@ The SDK includes a Django-style migration system for production schema managemen
 
 ```bash
 # Generate a migration from your models
-il makemigrations --models myapp.models
+il migration generate --models myapp.models
 
 # Apply pending migrations
-il migrate --url ws://localhost:8080/ws --kg production
+il migration apply --url ws://localhost:8080/ws --kg production
 
 # Check status
-il showmigrations --url ws://localhost:8080/ws --kg production
+il migration status --url ws://localhost:8080/ws --kg production
 
 # Rollback
-il revert --url ws://localhost:8080/ws --kg production 0001_initial
+il migration revert --url ws://localhost:8080/ws --kg production 0001_initial
 ```
 
 The autodetector diffs your current Python models against the last migration's state and generates the minimal set of operations (create/drop relations, create/drop/replace rules, create/drop indexes). Each migration file is self-contained with a full state snapshot.
@@ -519,10 +519,10 @@ The autodetector diffs your current Python models against the last migration's s
 
 | Command | Description |
 |---------|-------------|
-| `il makemigrations --models <module>` | Generate migration from model diff |
-| `il migrate --url <ws> --kg <name>` | Apply pending migrations |
-| `il revert --url <ws> --kg <name> <target>` | Revert to a target migration |
-| `il showmigrations --url <ws> --kg <name>` | Show applied/pending status |
+| `il migration generate --models <module>` | Generate migration from model diff |
+| `il migration apply --url <ws> --kg <name>` | Apply pending migrations |
+| `il migration revert --url <ws> --kg <name> <target>` | Revert to a target migration |
+| `il migration status --url <ws> --kg <name>` | Show applied/pending status |
 
 ### Aggregation Functions
 

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/README.md

@@ -17,7 +17,7 @@ pip install inputlayer-client-dev[all]         # everything
 
 Requirements: Python 3.10+ and a running InputLayer server.
 
-This also installs the `il` CLI for managing schema migrations.
+This also installs the `inputlayer-migrate` tool for schema migrations; with the InputLayer product CLI installed, use it as `il migration <verb>`.
 
 ## Quick Start
 
@@ -391,16 +391,16 @@ The SDK includes a Django-style migration system for production schema managemen
 
 ```bash
 # Generate a migration from your models
-il makemigrations --models myapp.models
+il migration generate --models myapp.models
 
 # Apply pending migrations
-il migrate --url ws://localhost:8080/ws --kg production
+il migration apply --url ws://localhost:8080/ws --kg production
 
 # Check status
-il showmigrations --url ws://localhost:8080/ws --kg production
+il migration status --url ws://localhost:8080/ws --kg production
 
 # Rollback
-il revert --url ws://localhost:8080/ws --kg production 0001_initial
+il migration revert --url ws://localhost:8080/ws --kg production 0001_initial
 ```
 
 The autodetector diffs your current Python models against the last migration's state and generates the minimal set of operations (create/drop relations, create/drop/replace rules, create/drop indexes). Each migration file is self-contained with a full state snapshot.
@@ -485,10 +485,10 @@ The autodetector diffs your current Python models against the last migration's s
 
 | Command | Description |
 |---------|-------------|
-| `il makemigrations --models <module>` | Generate migration from model diff |
-| `il migrate --url <ws> --kg <name>` | Apply pending migrations |
-| `il revert --url <ws> --kg <name> <target>` | Revert to a target migration |
-| `il showmigrations --url <ws> --kg <name>` | Show applied/pending status |
+| `il migration generate --models <module>` | Generate migration from model diff |
+| `il migration apply --url <ws> --kg <name>` | Apply pending migrations |
+| `il migration revert --url <ws> --kg <name> <target>` | Revert to a target migration |
+| `il migration status --url <ws> --kg <name>` | Show applied/pending status |
 
 ### Aggregation Functions
 

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "inputlayer-client-dev"
-version = "0.1.0.dev915"
+version = "0.1.0.dev919"
 description = "Python Object-Logic Mapper for InputLayer knowledge graph engine"
 readme = "README.md"
 license = "Apache-2.0"
@@ -34,7 +34,7 @@ dev = [
 ]
 
 [project.scripts]
-il = "inputlayer.migrations.cli:main"
+inputlayer-migrate = "inputlayer.migrations.cli:main"
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/inputlayer"]

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/src/inputlayer/__init__.py

@@ -68,8 +68,6 @@ from inputlayer.knowledge_graph import (
     WhyResult,
 )
 
-# Migrations
-from inputlayer.migrations import Migration
 
 # Notifications
 from inputlayer.notifications import NotificationEvent
@@ -116,8 +114,6 @@ __all__ = [
     "KnowledgeGraph",
     "KnowledgeGraphExistsError",
     "KnowledgeGraphNotFoundError",
-    # Migrations
-    "Migration",
     # Notifications
     "NotificationEvent",
     "PermissionError",

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/src/inputlayer/migrations/__init__.py

@@ -2,8 +2,6 @@
 
 from __future__ import annotations
 
-from typing import Any
-
 from inputlayer.migrations.operations import (
     CreateIndex,
     CreateRelation,
@@ -19,22 +17,6 @@ from inputlayer.migrations.operations import (
 from inputlayer.migrations.state import ModelState
 
 
-class Migration:
-    """Base class for migration files.
-
-    Subclass as ``M`` in each migration file::
-
-        class M(Migration):
-            dependencies = ["0001_initial"]
-            operations = [ops.CreateRelation(...)]
-            state = {"relations": {...}, "rules": {...}, "indexes": {}}
-    """
-
-    dependencies: list[str] = []  # noqa: RUF012
-    operations: list[Operation] = []  # noqa: RUF012
-    state: dict[str, Any] = {}  # noqa: RUF012
-
-
 __all__ = [
     "CreateIndex",
     "CreateRelation",
@@ -42,7 +24,6 @@ __all__ = [
     "DropIndex",
     "DropRelation",
     "DropRule",
-    "Migration",
     "ModelState",
     "Operation",
     "ReplaceRule",

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/src/inputlayer/migrations/cli.py

@@ -205,6 +205,20 @@ def _add_connection_args(parser: argparse.ArgumentParser) -> None:
     )
 
 
+def _add_migrations_dir(sub: argparse.ArgumentParser) -> None:
+    """Accept --migrations-dir after the subcommand as well as before it.
+
+    Delegating callers (the `il` CLI) always place flags after the
+    subcommand. SUPPRESS keeps the subparser from clobbering the value the
+    global flag already parsed.
+    """
+    sub.add_argument(
+        "--migrations-dir",
+        default=argparse.SUPPRESS,
+        help="Directory for migration files (default: ./migrations)",
+    )
+
+
 def build_parser() -> argparse.ArgumentParser:
     """Build the CLI argument parser."""
     parser = argparse.ArgumentParser(
@@ -226,22 +240,26 @@ def build_parser() -> argparse.ArgumentParser:
         help="Python module path containing models (e.g. myapp.models)",
     )
     make.add_argument("--name", default=None, help="Custom migration name suffix")
+    _add_migrations_dir(make)
     make.set_defaults(func=_cmd_makemigrations)
 
     # migrate
     mig = subparsers.add_parser("migrate", help="Apply pending migrations")
     _add_connection_args(mig)
+    _add_migrations_dir(mig)
     mig.set_defaults(func=_cmd_migrate)
 
     # revert
     rev = subparsers.add_parser("revert", help="Revert migrations to a target")
     _add_connection_args(rev)
+    _add_migrations_dir(rev)
     rev.add_argument("target", help="Migration name to revert to (e.g. 0001_initial)")
     rev.set_defaults(func=_cmd_revert)
 
     # showmigrations
     show = subparsers.add_parser("showmigrations", help="Show migration status")
     _add_connection_args(show)
+    _add_migrations_dir(show)
     show.set_defaults(func=_cmd_showmigrations)
 
     return parser
@@ -249,6 +267,9 @@ def build_parser() -> argparse.ArgumentParser:
 
 def main(argv: list[str] | None = None) -> int:
     """CLI entry point."""
+    from inputlayer.exceptions import InputLayerError
+    from inputlayer.migrations.errors import MigrationError
+
     parser = build_parser()
     args = parser.parse_args(argv)
 
@@ -256,7 +277,11 @@ def main(argv: list[str] | None = None) -> int:
         parser.print_help()
         return 1
 
-    return args.func(args)
+    try:
+        return args.func(args)
+    except (MigrationError, InputLayerError) as exc:
+        print(f"error: {exc}")
+        return 1
 
 
 if __name__ == "__main__":

inputlayer_client_dev-0.1.0.dev919/src/inputlayer/migrations/errors.py

@@ -0,0 +1,27 @@
+"""Migration errors and the engine-result check.
+
+The WebSocket protocol reports engine failures as a result-set with a
+single "error" column rather than an exception, so anything that must
+know whether a statement actually executed has to check for that shape
+explicitly. The migration stack must: silently swallowed errors are how
+the recorder shipped broken (write-only applied-state) in the first
+place.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+
+class MigrationError(Exception):
+    """Raised when a migration fails to load, apply, or revert."""
+
+
+def check_engine_result(result: Any, context: str) -> Any:
+    """Raise MigrationError if an execute() result is an engine error frame."""
+    columns = list(getattr(result, "columns", None) or [])
+    if columns == ["error"]:
+        rows = getattr(result, "rows", None) or []
+        detail = str(rows[0][0]) if rows and rows[0] else "unknown engine error"
+        raise MigrationError(f"engine error while {context}: {detail}")
+    return result

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/src/inputlayer/migrations/executor.py

@@ -2,26 +2,35 @@
 
 from __future__ import annotations
 
+from inputlayer.migrations.errors import MigrationError, check_engine_result
 from inputlayer.migrations.loader import MigrationInfo
 from inputlayer.migrations.recorder import KGExecutor, MigrationRecorder
 
-
-class MigrationError(Exception):
-    """Raised when a migration fails to apply or revert."""
+__all__ = ["MigrationError", "apply_migration", "migrate", "revert_migration", "revert_to"]
 
 
 def apply_migration(kg: KGExecutor, migration: MigrationInfo) -> None:
-    """Apply a single migration's forward operations."""
+    """Apply a single migration's forward operations.
+
+    Raises MigrationError on the first failing operation. Operations are
+    not transactional: earlier operations of this migration stay applied,
+    but the migration is NOT recorded as applied, so the failure is
+    visible and a re-run picks up where the schema actually is.
+    """
     for op in migration.operations:
         for cmd in op.forward_commands():
-            kg.execute(cmd)
+            check_engine_result(
+                kg.execute(cmd), f"applying {migration.name} ({op.describe()})"
+            )
 
 
 def revert_migration(kg: KGExecutor, migration: MigrationInfo) -> None:
     """Revert a single migration's operations in reverse order."""
     for op in reversed(migration.operations):
         for cmd in op.backward_commands():
-            kg.execute(cmd)
+            check_engine_result(
+                kg.execute(cmd), f"reverting {migration.name} ({op.describe()})"
+            )
 
 
 def migrate(

inputlayer_client_dev-0.1.0.dev919/src/inputlayer/migrations/loader.py

@@ -0,0 +1,135 @@
+"""Migration loader - discover and load JSON migration files from a directory."""
+
+from __future__ import annotations
+
+import re
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any
+
+from inputlayer.migrations.errors import MigrationError
+
+
+@dataclass
+class MigrationInfo:
+    """Metadata about a loaded migration file."""
+
+    name: str           # e.g. "0001_initial"
+    number: int         # e.g. 1
+    filename: str       # e.g. "0001_initial.py"
+    dependencies: list[str]
+    operations: list    # list[Operation]
+    state: dict[str, Any]
+
+    @property
+    def module_name(self) -> str:
+        return self.name
+
+
+_MIGRATION_RE = re.compile(r"^(\d{4})_.+\.json$")
+
+
+def _load_json_migration(entry: Path, name: str, number: int) -> MigrationInfo:
+    """Load a language-neutral JSON migration (the current format)."""
+    import json
+
+    from inputlayer.migrations.operations import operation_from_dict
+    from inputlayer.migrations.writer import MIGRATION_FORMAT
+
+    try:
+        document = json.loads(entry.read_text(encoding="utf-8"))
+    except json.JSONDecodeError as exc:
+        raise MigrationError(f"{entry.name}: invalid JSON: {exc}") from exc
+    if not isinstance(document, dict):
+        raise MigrationError(f"{entry.name}: migration document must be a JSON object")
+    fmt = document.get("format")
+    if not isinstance(fmt, int) or fmt > MIGRATION_FORMAT:
+        raise MigrationError(
+            f"{entry.name}: unsupported migration format {fmt!r} "
+            f"(this SDK reads up to {MIGRATION_FORMAT} - upgrade the SDK)"
+        )
+    state = dict(document.get("state", {}))
+    # JSON has no tuples; restore (col, type) pairs so state comparisons in
+    # the autodetector see no phantom changes against in-memory model state.
+    if "relations" in state:
+        state["relations"] = {
+            rel: [tuple(col) for col in cols] for rel, cols in state["relations"].items()
+        }
+    try:
+        operations = [operation_from_dict(d) for d in document.get("operations", [])]
+    except (KeyError, TypeError, ValueError) as exc:
+        raise MigrationError(f"{entry.name}: invalid operation: {exc}") from exc
+    return MigrationInfo(
+        name=name,
+        number=number,
+        filename=entry.name,
+        dependencies=list(document.get("dependencies", [])),
+        operations=operations,
+        state=state,
+    )
+
+
+def load_migrations(directory: str | Path) -> list[MigrationInfo]:
+    """Discover and load all migration files from a directory.
+
+    Migrations are language-neutral JSON documents; that is the only
+    format (pre-1.0, no legacy loading). Non-matching files - including
+    stray .py files from older SDKs - are ignored. Returns migrations
+    sorted by number.
+    """
+    directory = Path(directory)
+    if not directory.is_dir():
+        return []
+
+    migrations: list[MigrationInfo] = []
+    for entry in sorted(directory.iterdir()):
+        if not entry.is_file():
+            continue
+        match = _MIGRATION_RE.match(entry.name)
+        if not match:
+            continue
+
+        number = int(match.group(1))
+        name = entry.stem  # e.g. "0001_initial"
+
+        migrations.append(_load_json_migration(entry, name, number))
+
+    migrations.sort(key=lambda m: m.number)
+
+    # Duplicate names or numbers are always a mistake (a stale legacy .py
+    # twin of a regenerated .json, or a merge collision) and previously
+    # caused the same migration to be applied twice - destructively for
+    # Drop/Replace operations. Refuse loudly.
+    by_name: dict[str, str] = {}
+    by_number: dict[int, str] = {}
+    for m in migrations:
+        if m.name in by_name:
+            raise MigrationError(
+                f"duplicate migration name '{m.name}': {by_name[m.name]} and "
+                f"{m.filename} - delete the stale one"
+            )
+        if m.number in by_number:
+            raise MigrationError(
+                f"duplicate migration number {m.number:04d}: "
+                f"{by_number[m.number]} and {m.filename} - renumber one of them"
+            )
+        by_name[m.name] = m.filename
+        by_number[m.number] = m.filename
+
+    return migrations
+
+
+def get_latest_state(directory: str | Path) -> dict[str, Any]:
+    """Get the state from the most recent migration, or empty if none exist."""
+    migrations = load_migrations(directory)
+    if not migrations:
+        return {"relations": {}, "rules": {}, "indexes": {}}
+    return migrations[-1].state
+
+
+def get_next_number(directory: str | Path) -> int:
+    """Get the next migration number."""
+    migrations = load_migrations(directory)
+    if not migrations:
+        return 1
+    return migrations[-1].number + 1

inputlayer_client_dev-0.1.0.dev919/src/inputlayer/migrations/recorder.py

@@ -0,0 +1,83 @@
+"""Migration recorder - track applied migrations in the DB."""
+
+from __future__ import annotations
+
+from datetime import datetime, timezone
+from typing import Any, Protocol
+
+from inputlayer.migrations.errors import check_engine_result
+
+
+class KGExecutor(Protocol):
+    """Minimal interface for executing IQL commands."""
+
+    def execute(self, iql: str) -> object: ...
+
+
+# NOTE: must be a valid relation name (lowercase letter first). The previous
+# dunder name was rejected by schema declarations but silently auto-created
+# by inserts, leaving recorded state in a relation that queries could not
+# read - migration status/revert never worked against a real engine.
+MIGRATION_RELATION = "inputlayer_migrations"
+
+
+class MigrationRecorder:
+    """Track which migrations have been applied using an internal relation."""
+
+    def __init__(self, kg: KGExecutor) -> None:
+        self._kg = kg
+
+    def _execute(self, iql: str, context: str) -> Any:
+        return check_engine_result(self._kg.execute(iql), context)
+
+    def ensure_schema(self) -> None:
+        """Create the migration tracking relation if it doesn't exist."""
+        # The engine reports "already registered" as an error frame; that is
+        # the one expected, harmless outcome, so only genuinely new failures
+        # surface. Anything else (bad name, auth, parse) must not be
+        # swallowed - that is how the recorder shipped write-only once.
+        result = self._kg.execute(
+            f"+{MIGRATION_RELATION}(name: string, applied_at: string)"
+        )
+        columns = list(getattr(result, "columns", None) or [])
+        if columns == ["error"]:
+            rows = getattr(result, "rows", None) or []
+            detail = str(rows[0][0]) if rows and rows[0] else ""
+            if "exist" not in detail and "registered" not in detail:
+                check_engine_result(result, "creating the migration tracking relation")
+
+    def get_applied(self) -> list[str]:
+        """Return sorted list of applied migration names.
+
+        Plain query form: the engine rejects `?X <- rel(...)` projections
+        ("Query cannot contain a rule definition"), which previously made
+        this return [] forever and broke status/revert/idempotent apply.
+        """
+        result = self._execute(
+            f"?{MIGRATION_RELATION}(Name, At)", "reading applied migrations"
+        )
+        rows = getattr(result, "rows", []) or []
+        return sorted({str(row[0]) for row in rows})
+
+    def record_applied(self, name: str) -> None:
+        """Record that a migration has been applied."""
+        now = datetime.now(timezone.utc).isoformat()
+        safe = _escape(name)
+        self._execute(
+            f'+{MIGRATION_RELATION}[("{safe}", "{now}")]',
+            f"recording {name} as applied",
+        )
+
+    def record_reverted(self, name: str) -> None:
+        """Remove the record for a reverted migration."""
+        safe = _escape(name)
+        self._execute(
+            f'-{MIGRATION_RELATION}(Name, At) <- '
+            f'{MIGRATION_RELATION}(Name, At), Name = "{safe}"',
+            f"recording {name} as reverted",
+        )
+
+
+def _escape(value: str) -> str:
+    """Escape a string for interpolation into an IQL string literal."""
+    return value.replace("\\", "\\\\").replace('"', '\\"')

inputlayer_client_dev-0.1.0.dev919/src/inputlayer/migrations/writer.py

@@ -0,0 +1,51 @@
+"""Migration file writer - generates language-neutral JSON migration files.
+
+A migration is data, not code: typed operations (each with enough
+structure to derive both apply and revert IQL) plus a state snapshot.
+Any SDK language - and the il CLI natively - can read and apply them.
+JSON is the only migration format.
+"""
+
+from __future__ import annotations
+
+import json
+import re
+from typing import Any
+
+from inputlayer.migrations.errors import MigrationError
+from inputlayer.migrations.operations import Operation
+
+MIGRATION_FORMAT = 1
+
+
+def generate_migration(
+    number: int,
+    operations: list[Operation],
+    state: dict[str, Any],
+    dependencies: list[str],
+    *,
+    name_suffix: str | None = None,
+) -> tuple[str, str]:
+    """Generate a migration file.
+
+    Returns (filename, content).
+    """
+    if name_suffix is not None and not re.fullmatch(r"[a-z0-9_]+", name_suffix):
+        raise MigrationError(
+            f"invalid migration name {name_suffix!r}: use lowercase letters, "
+            "digits, and underscores only"
+        )
+    if number == 1 and name_suffix is None:
+        name_suffix = "initial"
+    elif name_suffix is None:
+        name_suffix = "auto"
+
+    filename = f"{number:04d}_{name_suffix}.json"
+
+    document = {
+        "format": MIGRATION_FORMAT,
+        "dependencies": dependencies,
+        "operations": [op.to_dict() for op in operations],
+        "state": state,
+    }
+    return filename, json.dumps(document, indent=2, sort_keys=False) + "\n"

{inputlayer_client_dev-0.1.0.dev915 → inputlayer_client_dev-0.1.0.dev919}/tests/test_cli.py

@@ -18,6 +18,21 @@ class TestParser:
     def test_no_command_returns_1(self):
         assert main([]) == 1
 
+    def test_connection_error_prints_cleanly(self, capsys, monkeypatch):
+        # SDK errors (unreachable server, auth failure) must exit 1 with a
+        # clean message, not a traceback.
+        from inputlayer.exceptions import InputLayerConnectionError
+        from inputlayer.migrations import cli
+
+        def boom(args):
+            raise InputLayerConnectionError("Failed to connect to ws://nope")
+
+        monkeypatch.setattr(cli, "_cmd_showmigrations", boom)
+        rc = main(["showmigrations", "--url", "ws://nope", "--kg", "k"])
+        assert rc == 1
+        out = capsys.readouterr().out
+        assert "error: Failed to connect" in out
+
     def test_makemigrations_requires_models(self):
         parser = build_parser()
         with pytest.raises(SystemExit):
@@ -33,6 +48,33 @@ class TestParser:
         with pytest.raises(SystemExit):
             parser.parse_args(["revert", "--url", "ws://x", "--kg", "test"])
 
+    def test_migrations_dir_before_subcommand(self):
+        parser = build_parser()
+        args = parser.parse_args(
+            ["--migrations-dir", "custom", "makemigrations", "--models", "m"]
+        )
+        assert args.migrations_dir == "custom"
+
+    def test_migrations_dir_after_subcommand(self):
+        # Delegating callers (the il CLI) always place flags after the
+        # subcommand; this position must work on every subcommand.
+        parser = build_parser()
+        args = parser.parse_args(
+            ["makemigrations", "--models", "m", "--migrations-dir", "custom"]
+        )
+        assert args.migrations_dir == "custom"
+        args = parser.parse_args(
+            ["showmigrations", "--url", "ws://x", "--kg", "k",
+             "--migrations-dir", "custom"]
+        )
+        assert args.migrations_dir == "custom"
+
+    def test_migrations_dir_default_survives_subparser(self):
+        # SUPPRESS on the subparser copy must not clobber the global default.
+        parser = build_parser()
+        args = parser.parse_args(["makemigrations", "--models", "m"])
+        assert args.migrations_dir == "migrations"
+
     def test_showmigrations_parses(self):
         parser = build_parser()
         args = parser.parse_args(["showmigrations", "--url", "ws://x", "--kg", "test"])
@@ -125,7 +167,7 @@ class TestMakemigrations:
                 "--models", "models_for_cli",
             ])
             assert result == 0
-            files = list(migrations_dir.glob("0001_*.py"))
+            files = list(migrations_dir.glob("0001_*.json"))
             assert len(files) == 1
             assert "initial" in files[0].name
         finally:
@@ -168,7 +210,7 @@ class TestMakemigrations:
             ])
             assert result == 0
             # Should not create a second migration
-            files = list(migrations_dir.glob("0002_*.py"))
+            files = list(migrations_dir.glob("0002_*.json"))
             assert len(files) == 0
         finally:
             sys.path.pop(0)
@@ -220,7 +262,7 @@ class TestMakemigrations:
             ])
             assert result == 0
 
-            files = sorted(migrations_dir.glob("*.py"))
+            files = sorted(migrations_dir.glob("*.json"))
             assert len(files) == 2
 
             # Load and check dependency