Python pkg_resources.get_distribution alternative

Question

我有一些代码非常慢，我可以追溯到使用pkg_resources.get_distribution，我正在这样做：

# ...
pkg_resources.get_distribution(module_name).version

是否有我可以使用的替代库/方法可以做同样的事情？ 

Answer 1

这取决于

你正在使用的模块，大多数python库都实现了__version__属性，试试：

我的时间也表明它是最好的。我尝试了三种不同的方法：

import time
import pip
import pkg_resources
from pip.operations import freeze

# __version__
t0 = time.time()
version_number = pip.__version__
t1 = time.time()
total = t1-t0
print("__version__ =", version_number, "in", total)
# pip freeze
t0 = time.time()
x = freeze.freeze()
for p in x:
    if p[:p.find('==')] == "pip":
        version_number = p[p.find('==')+2:]
        break
t1 = time.time()
total = t1-t0
print("pip freeze =", version_number, "in", total)
# pkg_resources
t0 = time.time()
version_number = pkg_resources.get_distribution("pip").version
t1 = time.time()
total = t1-t0
print("pkg_resources =", version_number, "in", total)

请记住，pip freeze方法在时间上将高度可变，具体取决于系统中有多少模块。

结果：

__version__ = 9.0.1 in 2.1457672119140625e-06
pip freeze = 9.0.1 in 0.04388904571533203
pkg_resources = 9.0.1 in 0.0016148090362548828

在我的系统上，__version__比pkg_resources快 7.5 倍