From eb6112410218ddcbaa39e44f6328d5f92850d767 Mon Sep 17 00:00:00 2001 From: Doug Hellmann Date: Thu, 26 Apr 2018 15:25:45 -0400 Subject: [PATCH] add argparse with --version option Signed-off-by: Doug Hellmann --- git_nit/cmd.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/git_nit/cmd.py b/git_nit/cmd.py index 44a2d8f..6fe8f57 100644 --- a/git_nit/cmd.py +++ b/git_nit/cmd.py @@ -14,9 +14,23 @@ from __future__ import print_function +import argparse +import pkg_resources +def get_version(): + requirement = pkg_resources.Requirement.parse('git-nit') + provider = pkg_resources.get_provider(requirement) + return provider.version + + def main(): - pass + parser = argparse.ArgumentParser() + parser.add_argument( + '--version', + action='version', + version=get_version(), + ) + args = parser.parse_args() if __name__ == '__main__':