Comparação de números de versão em Python

Eu quero escrever umcmpFunção semelhante a uma que compara dois números de versão e retorna-1, 0ou1 com base em seus vales comparados.

Retorna-1 se a versão A for mais antiga que a versão BRetorna0 se a versão A e B são equivalentesRetorna1 se a versão A for mais recente que a versão B

Cada subseção deve ser interpretada como um número, portanto 1.10> 1.1.

Saídas de função desejadas são

mycmp('1.0', '1') == 0
mycmp('1.0.0', '1') == 0
mycmp('1', '1.0.0.1') == -1
mycmp('12.10', '11.0.0.0.0') == 1
...

E aqui está minha implementação, aberta para melhorias:

def mycmp(version1, version2):
    parts1 = [int(x) for x in version1.split('.')]
    parts2 = [int(x) for x in version2.split('.')]

    # fill up the shorter version with zeros ...
    lendiff = len(parts1) - len(parts2)
    if lendiff > 0:
        parts2.extend([0] * lendiff)
    elif lendiff < 0:
        parts1.extend([0] * (-lendiff))

    for i, p in enumerate(parts1):
        ret = cmp(p, parts2[i])
        if ret: return ret
    return 0

Estou usando o Python 2.4.5 btw. (instalado no meu local de trabalho ...).

Aqui está uma pequena 'suíte de testes' que você pode usar

assert mycmp('1', '2') == -1
assert mycmp('2', '1') == 1
assert mycmp('1', '1') == 0
assert mycmp('1.0', '1') == 0
assert mycmp('1', '1.000') == 0
assert mycmp('12.01', '12.1') == 0
assert mycmp('13.0.1', '13.00.02') == -1
assert mycmp('1.1.1.1', '1.1.1.1') == 0
assert mycmp('1.1.1.2', '1.1.1.1') == 1
assert mycmp('1.1.3', '1.1.3.000') == 0
assert mycmp('3.1.1.0', '3.1.2.10') == -1
assert mycmp('1.1', '1.10') == -1

questionAnswers(16)

yourAnswerToTheQuestion