Refactor linting configuration parser options to a function

django_migration_linter/management/commands/lintmigrations.py

@@ -5,6 +5,7 @@ from importlib import import_module
 
 from django.core.management.base import BaseCommand
 
+from ..utils import register_linting_configuration_options
 from ...constants import __version__
 
 from ...migration_linter import MigrationLinter, MessageType
@@ -94,14 +95,6 @@ class Command(BaseCommand):
             help="check only migrations that have already been applied to the database",
         )
 
-        parser.add_argument(
-            "--exclude-migration-tests",
-            type=str,
-            nargs="*",
-            help="Specify backward incompatible migration tests "
-            "to be ignored (e.g. ALTER_COLUMN)",
-        )
-
         parser.add_argument(
             "-q",
             "--quiet",
@@ -109,11 +102,7 @@ class Command(BaseCommand):
             choices=MessageType.values(),
             help="don't print linting messages to stdout",
         )
-        parser.add_argument(
+        register_linting_configuration_options(parser)
-            "--warnings-as-errors",
-            action="store_true",
-            help="handle warnings as errors",
-        )
 
     def handle(self, *args, **options):
         if options["project_root_path"]:

django_migration_linter/management/utils.py

@@ -0,0 +1,21 @@
+def register_linting_configuration_options(parser):
+    parser.add_argument(
+        "--database",
+        type=str,
+        nargs="?",
+        help=(
+            "specify the database for which to generate the SQL. " "Defaults to default"
+        ),
+    )
+
+    parser.add_argument(
+        "--exclude-migration-tests",
+        type=str,
+        nargs="*",
+        help="Specify backward incompatible migration tests "
+        "to be ignored (e.g. ALTER_COLUMN)",
+    )
+
+    parser.add_argument(
+        "--warnings-as-errors", action="store_true", help="handle warnings as errors",
+    )