diff options
Diffstat (limited to 'kombu/utils/text.py')
| -rw-r--r-- | kombu/utils/text.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/kombu/utils/text.py b/kombu/utils/text.py index 20444710..066b28af 100644 --- a/kombu/utils/text.py +++ b/kombu/utils/text.py @@ -3,6 +3,9 @@ from __future__ import absolute_import from difflib import SequenceMatcher +from kombu import version_info_t +from kombu.five import string_t + def fmatch_iter(needle, haystack, min_ratio=0.6): for key in haystack: @@ -18,3 +21,27 @@ def fmatch_best(needle, haystack, min_ratio=0.6): )[0][1] except IndexError: pass + + +def version_string_as_tuple(s): + v = _unpack_version(*s.split('.')) + # X.Y.3a1 -> (X, Y, 3, 'a1') + if isinstance(v.micro, string_t): + v = version_info_t(v.major, v.minor, *_splitmicro(*v[2:])) + # X.Y.3a1-40 -> (X, Y, 3, 'a1', '40') + if not v.serial and v.releaselevel and '-' in v.releaselevel: + v = version_info_t(*list(v[0:3]) + v.releaselevel.split('-')) + return v + + +def _unpack_version(major, minor=0, micro=0, releaselevel='', serial=''): + return version_info_t(int(major), int(minor), micro, releaselevel, serial) + + +def _splitmicro(micro, releaselevel='', serial=''): + for index, char in enumerate(micro): + if not char.isdigit(): + break + else: + return int(micro or 0), releaselevel, serial + return int(micro[:index]), micro[index:], serial |