Added alembic command, added Dockerfile

Author: Alexey Nikitenko

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.11-slim
+
+WORKDIR /app/
+
+RUN apt-get update -y \
+    && apt-get install -y git curl \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN pip install migration-lint
+
+ENTRYPOINT ["migration-lint"]

migration_lint/extractor/alembic.py

@@ -14,7 +14,7 @@ class AlembicExtractor(BaseExtractor):
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
-        self.command = "make sqlmigrate"
+        self.command = kwargs.get("alembic_command") or "make sqlmigrate"
 
     def is_migration(self, path: str) -> bool:
         """Check if the specified file is a migration."""

migration_lint/main.py

@@ -63,6 +63,12 @@
     help="squawk config path",
     default=os.getenv("MIGRATION_LINTER_SQUAWK_CONFIG_PATH"),
 )
+@click.option(
+    "--alembic-command",
+    "alembic_command",
+    help="command to get Alembic migrations sql",
+    default=os.getenv("MIGRATION_LINTER_ALEMBIC_COMMAND"),
+)
 def main(
     loader_type: str, extractor_type: str, squawk_config_path: str, **kwargs
 ) -> None:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "migration-lint"
-version = "0.2.4"
+version = "0.2.5"
 description = "Tool for lint operations in DB migrations SQL"
 authors = ["Alexey Nikitenko <alexey.nikitenko@pandadoc.com>"]
 readme = "README.md"

tests/test_alembic_extractor.py

@@ -1,5 +1,7 @@
 from unittest import mock
 
+import pytest
+
 from migration_lint.extractor.alembic import AlembicExtractor
 from migration_lint.source_loader.model import SourceDiff
 
@@ -30,3 +32,24 @@ def test_alembic_extractor__ok():
     assert metadata.changed_files[0].allowed_with_backward_incompatible is True
     assert metadata.changed_files[1].allowed_with_backward_incompatible is True
     assert metadata.changed_files[2].allowed_with_backward_incompatible is False
+
+
+@pytest.mark.parametrize(
+    "command,expected_command",
+    [
+        (None, "make sqlmigrate"),
+        ("alembic upgrade head --sql", "alembic upgrade head --sql"),
+    ],
+)
+def test_alembic_extractor_command__ok(command, expected_command):
+    extractor = AlembicExtractor(alembic_command=command)
+    changed_files = [
+        SourceDiff(path="src/db/migrations/versions/202202151945_fbea801d4464_auto.py"),
+    ]
+
+    with mock.patch(
+        "migration_lint.extractor.alembic.subprocess.check_output"
+    ) as subprocess_mock:
+        subprocess_mock.return_value = "".encode("utf-8")
+        extractor.create_metadata(changed_files)
+        subprocess_mock.assert_called_once_with(expected_command.split())