add ModuleNotFoundError exception for self version checks

ArieLevs · Mar 9, 2024 · ce5b91a · ce5b91a
1 parent d9a6ed4
commit ce5b91a
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/pybump.py b/src/pybump.py
@@ -4,6 +4,7 @@
 from sys import stderr
 
 from ruamel.yaml import YAML, YAMLError
+
 try:
  from .pybump_version import PybumpVersion
 except ImportError:
@@ -57,9 +58,12 @@ def get_self_version(dist_name):
  :param dist_name: string
  :return: version as string
  """
- # https://docs.python.org/3/library/importlib.metadata.html#distribution-versions
- from importlib.metadata import version
- return version(dist_name)
+ try:
+ # https://docs.python.org/3/library/importlib.metadata.html#distribution-versions
+ from importlib.metadata import version
+ return version(dist_name)
+ except ModuleNotFoundError:
+ return 'version not found'
 
 
 def write_version_to_file(file_path, file_content, version, app_version):