Alembic path customization

Author: Alexey Nikitenko

migration_lint/extractor/alembic.py

@@ -1,3 +1,4 @@
+import os
 import re
 import subprocess
 from io import StringIO
@@ -15,12 +16,15 @@ class AlembicExtractor(BaseExtractor):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self.command = kwargs.get("alembic_command") or "make sqlmigrate"
+        self.migration_path = os.environ.get(
+            "MIGRATION_LINT_ALEMBIC_MIGRATIONS_PATH", "/migrations/versions/"
+        )
 
     def is_migration(self, path: str) -> bool:
         """Check if the specified file is a migration."""
 
         return (
-            "/migrations/versions/" in path
+            self.migration_path in path
             and path.endswith(".py")
             and "__init__.py" not in path
         )
@@ -34,7 +38,7 @@ def is_allowed_with_backward_incompatible_migration(self, path: str) -> bool:
             r".*/tables\.py",
             r".*/constants\.py",
             r".*/enums\.py",
-            r".*/migrations/versions/.*\.py",
+            rf".*{self.migration_path}.*\.py",
             r"^(?!.*\.py).*$",
         ]
         for pattern in allowed_patterns:

tests/test_alembic_extractor.py

@@ -2,7 +2,7 @@
 
 import pytest
 
-from migration_lint.extractor.alembic import AlembicExtractor
+from migration_lint.extractor.alembic import AlembicExtractor, os
 from migration_lint.source_loader.model import SourceDiff
 
 
@@ -53,3 +53,23 @@ def test_alembic_extractor_command__ok(command, expected_command):
         subprocess_mock.return_value = "".encode("utf-8")
         extractor.create_metadata(changed_files)
         subprocess_mock.assert_called_once_with(expected_command.split())
+
+
+def test_alembic_extractor_path__ok():
+    changed_files = [
+        SourceDiff(path="src/db/random/path/202202151945_fbea801d4464_auto.py"),
+    ]
+
+    with mock.patch(
+        "migration_lint.extractor.alembic.subprocess.check_output"
+    ) as subprocess_mock, mock.patch.dict(
+        os.environ,
+        {"MIGRATION_LINT_ALEMBIC_MIGRATIONS_PATH": "/random/path"},
+        clear=True,
+    ):
+        extractor = AlembicExtractor()
+        subprocess_mock.return_value = "".encode("utf-8")
+        metadata = extractor.create_metadata(changed_files)
+
+    assert len(metadata.migrations) == 1
+    assert metadata.changed_files[0].allowed_with_backward_incompatible is True