Merge pull request #45 from fosslight/develop

Add function for print package version
fosslight · Jan 4, 2022 · f4812a3 · f4812a3
2 parents 8e30826 + 9e9c225
commit f4812a3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/fosslight_util/help.py b/src/fosslight_util/help.py
@@ -3,6 +3,7 @@
 # Copyright (c) 2021 LG Electronics Inc.
 # SPDX-License-Identifier: Apache-2.0
 import sys
+import pkg_resources
 
 _HELP_MESSAGE_COMMON = """
          _______  _______  _______  _______  ___      ___   _______  __   __  _______
@@ -27,3 +28,13 @@ def print_help_msg(self, exitopt):
 
         if exitopt:
             sys.exit()
+
+
+def print_package_version(pkg_name, msg="", exitopt=True):
+    if msg == "":
+        msg = f"{pkg_name} Version :"
+    cur_version = pkg_resources.get_distribution(pkg_name).version
+    print(f'{msg} {cur_version}')
+
+    if exitopt:
+        sys.exit(0)
diff --git a/tests/test_help.py b/tests/test_help.py
@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 # Copyright (c) 2021 LG Electronics Inc.
 # SPDX-License-Identifier: Apache-2.0
-from fosslight_util.help import PrintHelpMsg
+from fosslight_util.help import PrintHelpMsg, print_package_version
 
 
 def main():
@@ -12,7 +12,9 @@ def main():
         Test Test Test
     """
     helpMsg = PrintHelpMsg(_HELP_MESSAGE_TEST)
-    helpMsg.print_help_msg(True)
+    helpMsg.print_help_msg(False)
+
+    print_package_version("fosslight_util", "FOSSLight Util Version")
 
 
 if __name__ == '__main__':