Add --version option

3YOURMIND · Feb 3, 2019 · fea6236 · fea6236
1 parent fbf0f44
commit fea6236
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/django_migration_linter/migration_linter.py b/django_migration_linter/migration_linter.py
@@ -20,7 +20,7 @@
 import sys
 
 from .cache import Cache
-from .constants import DEFAULT_CACHE_PATH, MIGRATION_FOLDER_NAME
+from .constants import DEFAULT_CACHE_PATH, MIGRATION_FOLDER_NAME, __version__
 from .migration import Migration
 from .utils import is_directory, is_django_project, clean_bytes_to_str
 from .sql_analyser import analyse_sql_statements
@@ -287,6 +287,9 @@ def _main():
         action="store_true",
         help="print more information during execution",
     )
+    parser.add_argument(
+        "--version", "-V", action="version", version="%(prog)s {}".format(__version__)
+    )
     parser.add_argument(
         "--database",
         type=str,