Use metadata to detect version of package (#4782)
Signed-off-by: Kebe <mail@kebe7jun.com>
This commit is contained in:
@@ -1,6 +1,6 @@
|
|||||||
"""Check environment configurations and dependency versions."""
|
"""Check environment configurations and dependency versions."""
|
||||||
|
|
||||||
import importlib
|
import importlib.metadata
|
||||||
import os
|
import os
|
||||||
import resource
|
import resource
|
||||||
import subprocess
|
import subprocess
|
||||||
@@ -59,9 +59,8 @@ def get_package_versions(packages):
|
|||||||
for package in packages:
|
for package in packages:
|
||||||
package_name = package.split("==")[0].split(">=")[0].split("<=")[0]
|
package_name = package.split("==")[0].split(">=")[0].split("<=")[0]
|
||||||
try:
|
try:
|
||||||
module = importlib.import_module(package_name)
|
version = importlib.metadata.version(package_name)
|
||||||
if hasattr(module, "__version__"):
|
versions[package_name] = version
|
||||||
versions[package_name] = module.__version__
|
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
versions[package_name] = "Module Not Found"
|
versions[package_name] = "Module Not Found"
|
||||||
return versions
|
return versions
|
||||||
|
|||||||
Reference in New Issue
Block a user