Python 使用动态比较说明符比较版本号

Question

给定两个版本号和一个比较说明符：

def check_versions(ver1, specifier, ver2):
    # TODO
check_versions("1.2.3", ">=", "2.0.0") # Should return True
check_versions("1.2.3", "==", "2.0.0") # Should return False

我想将比较应用于两个版本，以便从上面获得预期的输出。

我

发现我可以使用LooseVersion distutils.version比较版本，如下所示：

LooseVersion("1.2.3") >= LooseVersion("2.0.0")

但是我怎样才能使比较本身取决于传入的specifier？

我可以从这里使用解决方案：

def check_versions(ver1, specifier, ver2):
    return {
        '==': LooseVersion(ver1) == LooseVersion(ver2),
        '>=': LooseVersion(ver1) >= LooseVersion(ver2),
        '<=': LooseVersion(ver1) <= LooseVersion(ver2),
        '>': LooseVersion(ver1) > LooseVersion(ver2),
        '<': LooseVersion(ver1) < LooseVersion(ver2),
    }.get(x, False)

但这对我来说真的很笨拙。难道没有更优雅的解决方案吗？

Answer 1

无需创建包含所有可能版本比较的字典，您只需为使用的运算符创建查找，例如：

import operator as op
from distutils.version import LooseVersion
lookup = {'<': op.lt, '<=': op.le, '==': op.eq, '>=': op.ge, '>': op.gt}
def check_versions(ver1, specifier, ver2):
    try:
        return lookup[specifier](LooseVersion(ver1), LooseVersion(ver2))
    except KeyError:
        # unknown specifier
        return False