diff --git a/django_migration_linter/management/commands/lintmigrations.py b/django_migration_linter/management/commands/lintmigrations.py index b0da4e6..d3e47e0 100644 --- a/django_migration_linter/management/commands/lintmigrations.py +++ b/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( - "--warnings-as-errors", - action="store_true", - help="handle warnings as errors", - ) + register_linting_configuration_options(parser) def handle(self, *args, **options): if options["project_root_path"]: diff --git a/django_migration_linter/management/utils.py b/django_migration_linter/management/utils.py new file mode 100644 index 0000000..40f251e --- /dev/null +++ b/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", + )