diff --git a/django_migration_linter/migration_linter.py b/django_migration_linter/migration_linter.py index 9e755c6..84c8746 100644 --- a/django_migration_linter/migration_linter.py +++ b/django_migration_linter/migration_linter.py @@ -238,7 +238,7 @@ def split_path(path): return (split_path(a) if len(a) > 0 else []) + [b] -if __name__ == '__main__': +def _main(): parser = argparse.ArgumentParser( description='Detect backward incompatible django migrations.') parser.add_argument( @@ -310,3 +310,7 @@ if __name__ == '__main__': sys.exit(1) else: sys.exit(1) + + +if __name__ == '__main__': + _main() diff --git a/setup.py b/setup.py index 22017db..ae711ef 100644 --- a/setup.py +++ b/setup.py @@ -46,6 +46,11 @@ setup( extras_require={ 'test': test_requirements }, + entry_points={ + 'console_scripts': [ + 'django-migration-linter = django_migration_linter.migration_linter:_main' + ] + }, keywords='django migration lint linter database backward compatibility', classifiers=[