diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index a3e5c6de75814e6060046c9e80eee46d1c1a27ab..7e45650dcbfd276146e7df69c11df531260349b1 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -11,6 +11,7 @@ install-wheel: extends: .install-wheel script: - template --help + - template --version test: stage: test diff --git a/template/__init__.py b/template/__init__.py index 59ce2f81f20c1c770b78c63f7b82632cc2a327d6..ccf326def783d9f991fa76d963c3235b6c3d2d08 100644 --- a/template/__init__.py +++ b/template/__init__.py @@ -53,7 +53,15 @@ def main(): help="Output to filename", type=argparse.FileType("w"), ) + parser.add_argument( + "-V", + "--version", + help="Template version", + action="store_true", + ) args = parser.parse_args() + if args.version: + print("Template version {}.".format(__version__)) infd = args.filename if args.filename else sys.stdin outfd = args.output if args.output else sys.stdout print(render(infd.read()), file=outfd)