Browse code

added suggested changes

Ching-Hwa Yu authored on 12/04/2021 17:57:12
Showing 1 changed files
... ...
@@ -203,7 +203,17 @@ def install(args):
203 203
 
204 204
 def ver(args):
205 205
     package_dir = args.package_directory
206
-    version_location = glob.glob(f"{package_dir}/**/_version.py")[0]
206
+    try:
207
+        version_location, = glob.glob(f"{package_dir}/**/_version.py", recursive=True)
208
+    except ValueError as err:
209
+        if "not enough" in str(err):
210
+            print(f"'_version.py' not found in '{package_dir}'", file=sys.stderr)
211
+            sys.exit(1)
212
+        elif "too many" in str(err):
213
+            print(f"More than 1 '_version.py' found in '{package_dir}'",  file=sys.stderr)
214
+            sys.exit(1)
215
+        else:
216
+            raise
207 217
     version_spec = spec_from_file_location("version", version_location)
208 218
     version = module_from_spec(version_spec)
209 219
     version_spec.loader.exec_module(version)