add argparse with --version option

Signed-off-by: Doug Hellmann <doug@doughellmann.com>
This commit is contained in:
Doug Hellmann 2018-04-26 15:25:45 -04:00
parent ba465598ef
commit eb61124102
1 changed files with 15 additions and 1 deletions

View File

@ -14,9 +14,23 @@
from __future__ import print_function 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(): def main():
pass parser = argparse.ArgumentParser()
parser.add_argument(
'--version',
action='version',
version=get_version(),
)
args = parser.parse_args()
if __name__ == '__main__': if __name__ == '__main__':