|
- # This file is dual licensed under the terms of the Apache License, Version
- # 2.0, and the BSD License. See the LICENSE file in the root of this repository
- # for complete details.
- from __future__ import absolute_import, division, print_function
-
- import re
-
- from ._typing import TYPE_CHECKING, cast
- from .version import InvalidVersion, Version
-
- if TYPE_CHECKING: # pragma: no cover
- from typing import NewType, Union
-
- NormalizedName = NewType("NormalizedName", str)
-
- _canonicalize_regex = re.compile(r"[-_.]+")
-
-
- def canonicalize_name(name):
- # type: (str) -> NormalizedName
- # This is taken from PEP 503.
- value = _canonicalize_regex.sub("-", name).lower()
- return cast("NormalizedName", value)
-
-
- def canonicalize_version(_version):
- # type: (str) -> Union[Version, str]
- """
- This is very similar to Version.__str__, but has one subtle difference
- with the way it handles the release segment.
- """
-
- try:
- version = Version(_version)
- except InvalidVersion:
- # Legacy versions cannot be normalized
- return _version
-
- parts = []
-
- # Epoch
- if version.epoch != 0:
- parts.append("{0}!".format(version.epoch))
-
- # Release segment
- # NB: This strips trailing '.0's to normalize
- parts.append(re.sub(r"(\.0)+$", "", ".".join(str(x) for x in version.release)))
-
- # Pre-release
- if version.pre is not None:
- parts.append("".join(str(x) for x in version.pre))
-
- # Post-release
- if version.post is not None:
- parts.append(".post{0}".format(version.post))
-
- # Development release
- if version.dev is not None:
- parts.append(".dev{0}".format(version.dev))
-
- # Local version segment
- if version.local is not None:
- parts.append("+{0}".format(version.local))
-
- return "".join(parts)
|
