diff options
Diffstat (limited to 'src/pip/_vendor/tomli/_re.py')
-rw-r--r-- | src/pip/_vendor/tomli/_re.py | 94 |
1 files changed, 59 insertions, 35 deletions
diff --git a/src/pip/_vendor/tomli/_re.py b/src/pip/_vendor/tomli/_re.py index 3883fdd9c..994bb7493 100644 --- a/src/pip/_vendor/tomli/_re.py +++ b/src/pip/_vendor/tomli/_re.py @@ -1,37 +1,55 @@ +# SPDX-License-Identifier: MIT +# SPDX-FileCopyrightText: 2021 Taneli Hukkinen +# Licensed to PSF under a Contributor Agreement. + +from __future__ import annotations + from datetime import date, datetime, time, timedelta, timezone, tzinfo +from functools import lru_cache import re -from typing import TYPE_CHECKING, Any, Optional, Union +from typing import Any -if TYPE_CHECKING: - from re import Match - - from pip._vendor.tomli._parser import ParseFloat +from ._types import ParseFloat # E.g. # - 00:32:00.999999 # - 00:32:00 -_TIME_RE_STR = r"([01][0-9]|2[0-3]):([0-5][0-9]):([0-5][0-9])(\.[0-9]+)?" +_TIME_RE_STR = r"([01][0-9]|2[0-3]):([0-5][0-9]):([0-5][0-9])(?:\.([0-9]{1,6})[0-9]*)?" -RE_HEX = re.compile(r"[0-9A-Fa-f](?:_?[0-9A-Fa-f])*") -RE_BIN = re.compile(r"[01](?:_?[01])*") -RE_OCT = re.compile(r"[0-7](?:_?[0-7])*") RE_NUMBER = re.compile( - r"[+-]?(?:0|[1-9](?:_?[0-9])*)" # integer - + r"(?:\.[0-9](?:_?[0-9])*)?" # optional fractional part - + r"(?:[eE][+-]?[0-9](?:_?[0-9])*)?" # optional exponent part + r""" +0 +(?: + x[0-9A-Fa-f](?:_?[0-9A-Fa-f])* # hex + | + b[01](?:_?[01])* # bin + | + o[0-7](?:_?[0-7])* # oct +) +| +[+-]?(?:0|[1-9](?:_?[0-9])*) # dec, integer part +(?P<floatpart> + (?:\.[0-9](?:_?[0-9])*)? # optional fractional part + (?:[eE][+-]?[0-9](?:_?[0-9])*)? # optional exponent part +) +""", + flags=re.VERBOSE, ) RE_LOCALTIME = re.compile(_TIME_RE_STR) RE_DATETIME = re.compile( - r"([0-9]{4})-(0[1-9]|1[0-2])-(0[1-9]|1[0-9]|2[0-9]|3[01])" # date, e.g. 1988-10-27 - + r"(?:" - + r"[T ]" - + _TIME_RE_STR - + r"(?:(Z)|([+-])([01][0-9]|2[0-3]):([0-5][0-9]))?" # time offset - + r")?" + rf""" +([0-9]{{4}})-(0[1-9]|1[0-2])-(0[1-9]|[12][0-9]|3[01]) # date, e.g. 1988-10-27 +(?: + [Tt ] + {_TIME_RE_STR} + (?:([Zz])|([+-])([01][0-9]|2[0-3]):([0-5][0-9]))? # optional time offset +)? +""", + flags=re.VERBOSE, ) -def match_to_datetime(match: "Match") -> Union[datetime, date]: +def match_to_datetime(match: re.Match) -> datetime | date: """Convert a `RE_DATETIME` match to `datetime.datetime` or `datetime.date`. Raises ValueError if the match does not correspond to a valid date @@ -46,7 +64,7 @@ def match_to_datetime(match: "Match") -> Union[datetime, date]: sec_str, micros_str, zulu_time, - offset_dir_str, + offset_sign_str, offset_hour_str, offset_minute_str, ) = match.groups() @@ -54,14 +72,10 @@ def match_to_datetime(match: "Match") -> Union[datetime, date]: if hour_str is None: return date(year, month, day) hour, minute, sec = int(hour_str), int(minute_str), int(sec_str) - micros = int(micros_str[1:].ljust(6, "0")[:6]) if micros_str else 0 - if offset_dir_str: - offset_dir = 1 if offset_dir_str == "+" else -1 - tz: Optional[tzinfo] = timezone( - timedelta( - hours=offset_dir * int(offset_hour_str), - minutes=offset_dir * int(offset_minute_str), - ) + micros = int(micros_str.ljust(6, "0")) if micros_str else 0 + if offset_sign_str: + tz: tzinfo | None = cached_tz( + offset_hour_str, offset_minute_str, offset_sign_str ) elif zulu_time: tz = timezone.utc @@ -70,14 +84,24 @@ def match_to_datetime(match: "Match") -> Union[datetime, date]: return datetime(year, month, day, hour, minute, sec, micros, tzinfo=tz) -def match_to_localtime(match: "Match") -> time: +@lru_cache(maxsize=None) +def cached_tz(hour_str: str, minute_str: str, sign_str: str) -> timezone: + sign = 1 if sign_str == "+" else -1 + return timezone( + timedelta( + hours=sign * int(hour_str), + minutes=sign * int(minute_str), + ) + ) + + +def match_to_localtime(match: re.Match) -> time: hour_str, minute_str, sec_str, micros_str = match.groups() - micros = int(micros_str[1:].ljust(6, "0")[:6]) if micros_str else 0 + micros = int(micros_str.ljust(6, "0")) if micros_str else 0 return time(int(hour_str), int(minute_str), int(sec_str), micros) -def match_to_number(match: "Match", parse_float: "ParseFloat") -> Any: - match_str = match.group() - if "." in match_str or "e" in match_str or "E" in match_str: - return parse_float(match_str) - return int(match_str) +def match_to_number(match: re.Match, parse_float: ParseFloat) -> Any: + if match.group("floatpart"): + return parse_float(match.group()) + return int(match.group(), 0) |