version.py 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461
  1. """Extract reference documentation from the pypa/packaging source tree.
  2. In the process of copying, some unused methods / classes were removed.
  3. These include:
  4. - parse()
  5. - anything involving LegacyVersion
  6. This software is made available under the terms of *either* of the licenses
  7. found in LICENSE.APACHE or LICENSE.BSD. Contributions to this software is made
  8. under the terms of *both* these licenses.
  9. Vendored from:
  10. - https://github.com/pypa/packaging/
  11. - commit ba07d8287b4554754ac7178d177033ea3f75d489 (09/09/2021)
  12. """
  13. # This file is dual licensed under the terms of the Apache License, Version
  14. # 2.0, and the BSD License. See the LICENSE file in the root of this repository
  15. # for complete details.
  16. import collections
  17. import itertools
  18. import re
  19. from typing import Callable, Optional, SupportsInt, Tuple, Union
  20. __all__ = ["Version", "InvalidVersion", "VERSION_PATTERN"]
  21. # Vendored from https://github.com/pypa/packaging/blob/main/packaging/_structures.py
  22. class InfinityType:
  23. def __repr__(self) -> str:
  24. return "Infinity"
  25. def __hash__(self) -> int:
  26. return hash(repr(self))
  27. def __lt__(self, other: object) -> bool:
  28. return False
  29. def __le__(self, other: object) -> bool:
  30. return False
  31. def __eq__(self, other: object) -> bool:
  32. return isinstance(other, self.__class__)
  33. def __ne__(self, other: object) -> bool:
  34. return not isinstance(other, self.__class__)
  35. def __gt__(self, other: object) -> bool:
  36. return True
  37. def __ge__(self, other: object) -> bool:
  38. return True
  39. def __neg__(self: object) -> "NegativeInfinityType":
  40. return NegativeInfinity
  41. Infinity = InfinityType()
  42. class NegativeInfinityType:
  43. def __repr__(self) -> str:
  44. return "-Infinity"
  45. def __hash__(self) -> int:
  46. return hash(repr(self))
  47. def __lt__(self, other: object) -> bool:
  48. return True
  49. def __le__(self, other: object) -> bool:
  50. return True
  51. def __eq__(self, other: object) -> bool:
  52. return isinstance(other, self.__class__)
  53. def __ne__(self, other: object) -> bool:
  54. return not isinstance(other, self.__class__)
  55. def __gt__(self, other: object) -> bool:
  56. return False
  57. def __ge__(self, other: object) -> bool:
  58. return False
  59. def __neg__(self: object) -> InfinityType:
  60. return Infinity
  61. NegativeInfinity = NegativeInfinityType()
  62. # Vendored from https://github.com/pypa/packaging/blob/main/packaging/version.py
  63. InfiniteTypes = Union[InfinityType, NegativeInfinityType]
  64. PrePostDevType = Union[InfiniteTypes, Tuple[str, int]]
  65. SubLocalType = Union[InfiniteTypes, int, str]
  66. LocalType = Union[
  67. NegativeInfinityType,
  68. Tuple[
  69. Union[
  70. SubLocalType,
  71. Tuple[SubLocalType, str],
  72. Tuple[NegativeInfinityType, SubLocalType],
  73. ],
  74. ...,
  75. ],
  76. ]
  77. CmpKey = Tuple[
  78. int, Tuple[int, ...], PrePostDevType, PrePostDevType, PrePostDevType, LocalType
  79. ]
  80. LegacyCmpKey = Tuple[int, Tuple[str, ...]]
  81. VersionComparisonMethod = Callable[
  82. [Union[CmpKey, LegacyCmpKey], Union[CmpKey, LegacyCmpKey]], bool
  83. ]
  84. _Version = collections.namedtuple(
  85. "_Version", ["epoch", "release", "dev", "pre", "post", "local"]
  86. )
  87. class InvalidVersion(ValueError):
  88. """
  89. An invalid version was found, users should refer to PEP 440.
  90. """
  91. class _BaseVersion:
  92. _key: Union[CmpKey, LegacyCmpKey]
  93. def __hash__(self) -> int:
  94. return hash(self._key)
  95. # Please keep the duplicated `isinstance` check
  96. # in the six comparisons hereunder
  97. # unless you find a way to avoid adding overhead function calls.
  98. def __lt__(self, other: "_BaseVersion") -> bool:
  99. if not isinstance(other, _BaseVersion):
  100. return NotImplemented
  101. return self._key < other._key
  102. def __le__(self, other: "_BaseVersion") -> bool:
  103. if not isinstance(other, _BaseVersion):
  104. return NotImplemented
  105. return self._key <= other._key
  106. def __eq__(self, other: object) -> bool:
  107. if not isinstance(other, _BaseVersion):
  108. return NotImplemented
  109. return self._key == other._key
  110. def __ge__(self, other: "_BaseVersion") -> bool:
  111. if not isinstance(other, _BaseVersion):
  112. return NotImplemented
  113. return self._key >= other._key
  114. def __gt__(self, other: "_BaseVersion") -> bool:
  115. if not isinstance(other, _BaseVersion):
  116. return NotImplemented
  117. return self._key > other._key
  118. def __ne__(self, other: object) -> bool:
  119. if not isinstance(other, _BaseVersion):
  120. return NotImplemented
  121. return self._key != other._key
  122. # Deliberately not anchored to the start and end of the string, to make it
  123. # easier for 3rd party code to reuse
  124. VERSION_PATTERN = r"""
  125. v?
  126. (?:
  127. (?:(?P<epoch>[0-9]+)!)? # epoch
  128. (?P<release>[0-9]+(?:\.[0-9]+)*) # release segment
  129. (?P<pre> # pre-release
  130. [-_\.]?
  131. (?P<pre_l>(a|b|c|rc|alpha|beta|pre|preview))
  132. [-_\.]?
  133. (?P<pre_n>[0-9]+)?
  134. )?
  135. (?P<post> # post release
  136. (?:-(?P<post_n1>[0-9]+))
  137. |
  138. (?:
  139. [-_\.]?
  140. (?P<post_l>post|rev|r)
  141. [-_\.]?
  142. (?P<post_n2>[0-9]+)?
  143. )
  144. )?
  145. (?P<dev> # dev release
  146. [-_\.]?
  147. (?P<dev_l>dev)
  148. [-_\.]?
  149. (?P<dev_n>[0-9]+)?
  150. )?
  151. )
  152. (?:\+(?P<local>[a-z0-9]+(?:[-_\.][a-z0-9]+)*))? # local version
  153. """
  154. class Version(_BaseVersion):
  155. _regex = re.compile(r"^\s*" + VERSION_PATTERN + r"\s*$", re.VERBOSE | re.IGNORECASE)
  156. def __init__(self, version: str) -> None:
  157. # Validate the version and parse it into pieces
  158. match = self._regex.search(version)
  159. if not match:
  160. raise InvalidVersion(f"Invalid version: '{version}'")
  161. # Store the parsed out pieces of the version
  162. self._version = _Version(
  163. epoch=int(match.group("epoch")) if match.group("epoch") else 0,
  164. release=tuple(int(i) for i in match.group("release").split(".")),
  165. pre=_parse_letter_version(match.group("pre_l"), match.group("pre_n")),
  166. post=_parse_letter_version(
  167. match.group("post_l"), match.group("post_n1") or match.group("post_n2")
  168. ),
  169. dev=_parse_letter_version(match.group("dev_l"), match.group("dev_n")),
  170. local=_parse_local_version(match.group("local")),
  171. )
  172. # Generate a key which will be used for sorting
  173. self._key = _cmpkey(
  174. self._version.epoch,
  175. self._version.release,
  176. self._version.pre,
  177. self._version.post,
  178. self._version.dev,
  179. self._version.local,
  180. )
  181. def __repr__(self) -> str:
  182. return f"<Version('{self}')>"
  183. def __str__(self) -> str:
  184. parts = []
  185. # Epoch
  186. if self.epoch != 0:
  187. parts.append(f"{self.epoch}!")
  188. # Release segment
  189. parts.append(".".join(str(x) for x in self.release))
  190. # Pre-release
  191. if self.pre is not None:
  192. parts.append("".join(str(x) for x in self.pre))
  193. # Post-release
  194. if self.post is not None:
  195. parts.append(f".post{self.post}")
  196. # Development release
  197. if self.dev is not None:
  198. parts.append(f".dev{self.dev}")
  199. # Local version segment
  200. if self.local is not None:
  201. parts.append(f"+{self.local}")
  202. return "".join(parts)
  203. @property
  204. def epoch(self) -> int:
  205. _epoch: int = self._version.epoch
  206. return _epoch
  207. @property
  208. def release(self) -> Tuple[int, ...]:
  209. _release: Tuple[int, ...] = self._version.release
  210. return _release
  211. @property
  212. def pre(self) -> Optional[Tuple[str, int]]:
  213. _pre: Optional[Tuple[str, int]] = self._version.pre
  214. return _pre
  215. @property
  216. def post(self) -> Optional[int]:
  217. return self._version.post[1] if self._version.post else None
  218. @property
  219. def dev(self) -> Optional[int]:
  220. return self._version.dev[1] if self._version.dev else None
  221. @property
  222. def local(self) -> Optional[str]:
  223. if self._version.local:
  224. return ".".join(str(x) for x in self._version.local)
  225. else:
  226. return None
  227. @property
  228. def public(self) -> str:
  229. return str(self).split("+", 1)[0]
  230. @property
  231. def base_version(self) -> str:
  232. parts = []
  233. # Epoch
  234. if self.epoch != 0:
  235. parts.append(f"{self.epoch}!")
  236. # Release segment
  237. parts.append(".".join(str(x) for x in self.release))
  238. return "".join(parts)
  239. @property
  240. def is_prerelease(self) -> bool:
  241. return self.dev is not None or self.pre is not None
  242. @property
  243. def is_postrelease(self) -> bool:
  244. return self.post is not None
  245. @property
  246. def is_devrelease(self) -> bool:
  247. return self.dev is not None
  248. @property
  249. def major(self) -> int:
  250. return self.release[0] if len(self.release) >= 1 else 0
  251. @property
  252. def minor(self) -> int:
  253. return self.release[1] if len(self.release) >= 2 else 0
  254. @property
  255. def micro(self) -> int:
  256. return self.release[2] if len(self.release) >= 3 else 0
  257. def _parse_letter_version(
  258. letter: str, number: Union[str, bytes, SupportsInt]
  259. ) -> Optional[Tuple[str, int]]:
  260. if letter:
  261. # We consider there to be an implicit 0 in a pre-release if there is
  262. # not a numeral associated with it.
  263. if number is None:
  264. number = 0
  265. # We normalize any letters to their lower case form
  266. letter = letter.lower()
  267. # We consider some words to be alternate spellings of other words and
  268. # in those cases we want to normalize the spellings to our preferred
  269. # spelling.
  270. if letter == "alpha":
  271. letter = "a"
  272. elif letter == "beta":
  273. letter = "b"
  274. elif letter in ["c", "pre", "preview"]:
  275. letter = "rc"
  276. elif letter in ["rev", "r"]:
  277. letter = "post"
  278. return letter, int(number)
  279. if not letter and number:
  280. # We assume if we are given a number, but we are not given a letter
  281. # then this is using the implicit post release syntax (e.g. 1.0-1)
  282. letter = "post"
  283. return letter, int(number)
  284. return None
  285. _local_version_separators = re.compile(r"[\._-]")
  286. def _parse_local_version(local: str) -> Optional[LocalType]:
  287. """
  288. Takes a string like abc.1.twelve and turns it into ("abc", 1, "twelve").
  289. """
  290. if local is not None:
  291. return tuple(
  292. part.lower() if not part.isdigit() else int(part)
  293. for part in _local_version_separators.split(local)
  294. )
  295. return None
  296. def _cmpkey(
  297. epoch: int,
  298. release: Tuple[int, ...],
  299. pre: Optional[Tuple[str, int]],
  300. post: Optional[Tuple[str, int]],
  301. dev: Optional[Tuple[str, int]],
  302. local: Optional[Tuple[SubLocalType]],
  303. ) -> CmpKey:
  304. # When we compare a release version, we want to compare it with all of the
  305. # trailing zeros removed. So we'll use a reverse the list, drop all the now
  306. # leading zeros until we come to something non zero, then take the rest
  307. # re-reverse it back into the correct order and make it a tuple and use
  308. # that for our sorting key.
  309. _release = tuple(
  310. reversed(list(itertools.dropwhile(lambda x: x == 0, reversed(release))))
  311. )
  312. # We need to "trick" the sorting algorithm to put 1.0.dev0 before 1.0a0.
  313. # We'll do this by abusing the pre segment, but we _only_ want to do this
  314. # if there is not a pre or a post segment. If we have one of those then
  315. # the normal sorting rules will handle this case correctly.
  316. if pre is None and post is None and dev is not None:
  317. _pre: PrePostDevType = NegativeInfinity
  318. # Versions without a pre-release (except as noted above) should sort after
  319. # those with one.
  320. elif pre is None:
  321. _pre = Infinity
  322. else:
  323. _pre = pre
  324. # Versions without a post segment should sort before those with one.
  325. if post is None:
  326. _post: PrePostDevType = NegativeInfinity
  327. else:
  328. _post = post
  329. # Versions without a development segment should sort after those with one.
  330. if dev is None:
  331. _dev: PrePostDevType = Infinity
  332. else:
  333. _dev = dev
  334. if local is None:
  335. # Versions without a local segment should sort before those with one.
  336. _local: LocalType = NegativeInfinity
  337. else:
  338. # Versions with a local segment need that segment parsed to implement
  339. # the sorting rules in PEP440.
  340. # - Alpha numeric segments sort before numeric segments
  341. # - Alpha numeric segments sort lexicographically
  342. # - Numeric segments sort numerically
  343. # - Shorter versions sort before longer versions when the prefixes
  344. # match exactly
  345. _local = tuple(
  346. (i, "") if isinstance(i, int) else (NegativeInfinity, i) for i in local
  347. )
  348. return epoch, _release, _pre, _post, _dev, _local