Merge "versionutil: Remove trailing alpha/beta/rc suffix"

This commit is contained in:
Zuul 2023-11-07 20:28:13 +00:00 committed by Gerrit Code Review
commit 505d80e557
2 changed files with 12 additions and 0 deletions

View File

@ -83,3 +83,13 @@ class IsCompatibleTestCase(test_base.BaseTestCase):
def test_convert_version_to_tuple(self):
self.assertEqual((6, 7, 0),
versionutils.convert_version_to_tuple('6.7.0'))
self.assertEqual((6, 7, 0),
versionutils.convert_version_to_tuple('6.7.0a1'))
self.assertEqual((6, 7, 0),
versionutils.convert_version_to_tuple('6.7.0alpha1'))
self.assertEqual((6, 7, 0),
versionutils.convert_version_to_tuple('6.7.0b1'))
self.assertEqual((6, 7, 0),
versionutils.convert_version_to_tuple('6.7.0beta1'))
self.assertEqual((6, 7, 0),
versionutils.convert_version_to_tuple('6.7.0rc1'))

View File

@ -20,6 +20,7 @@ Helpers for comparing version strings.
"""
import functools
import re
import packaging.version
@ -87,4 +88,5 @@ def convert_version_to_tuple(version_str):
.. versionadded:: 2.0
"""
version_str = re.sub(r'(\d+)(a|alpha|b|beta|rc)\d+$', '\\1', version_str)
return tuple(int(part) for part in version_str.split('.'))