a5d56e1705
Some scripts, such as declare.py, may provide dates and not datetimes. This update make the declare_version work for both.
126 lines
4.9 KiB
Python
126 lines
4.9 KiB
Python
import calendar
|
|
import datetime
|
|
|
|
|
|
def parse_date(text: str, formats: list[str] = frozenset([
|
|
"%B %d %Y", # January 1 2020
|
|
"%b %d %Y", # Jan 1 2020
|
|
"%d %B %Y", # 1 January 2020
|
|
"%d %b %Y", # 1 Jan 2020
|
|
"%d-%b-%Y", # 1-Jan-2020
|
|
"%Y-%b-%d", # 2020-Jan-01
|
|
"%d-%B-%Y", # 1-January-2020
|
|
"%Y-%m-%d", # 2020-01-01
|
|
"%m/%d/%Y", # 01/25/2020
|
|
"%Y/%m/%d", # 2020/01/25
|
|
"%A %d %B %Y", # Wednesday 1 January 2020
|
|
"%A %d %b %Y", # Wednesday 1 Jan 2020
|
|
])) -> datetime.datetime:
|
|
"""Parse a given text representing a date using a list of formats.
|
|
"""
|
|
return parse_datetime(text, formats, to_utc=False)
|
|
|
|
|
|
def parse_month_year_date(text: str, formats: list[str] = frozenset([
|
|
"%B %Y", # January 2020
|
|
"%b %Y", # Jan 2020
|
|
"%Y-%m", # 2020-01
|
|
"%Y/%m", # 2020/01
|
|
"%m-%Y", # 01-2020
|
|
"%m/%Y", # 01/2020
|
|
])) -> datetime.datetime:
|
|
"""Parse a given text representing a partial date using a list of formats,
|
|
adjusting it to the last day of the month.
|
|
"""
|
|
date = parse_datetime(text, formats, to_utc=False)
|
|
_, last_day = calendar.monthrange(date.year, date.month)
|
|
return date.replace(day=last_day)
|
|
|
|
|
|
def parse_date_or_month_year_date(text: str) -> datetime.datetime:
|
|
"""Parse a given text representing a date or a partial date using the default list of formats.
|
|
"""
|
|
try:
|
|
return parse_date(text)
|
|
except ValueError:
|
|
return parse_month_year_date(text)
|
|
|
|
|
|
def parse_datetime(text: str, formats: list[str] = frozenset([
|
|
"%Y-%m-%d %H:%M:%S", # 2023-05-01 08:32:34
|
|
"%Y-%m-%dT%H:%M:%S", # 2023-05-01T08:32:34
|
|
"%d-%b-%Y %H:%M", # 01-May-2023 08:32
|
|
"%Y-%m-%d %H:%M:%S %z", # 2023-05-01 08:32:34 +0900
|
|
"%Y-%m-%dT%H:%M:%S%z", # 2023-05-01T08:32:34+0900
|
|
"%Y-%m-%dT%H:%M:%S.%f%z", # 2023-05-01T08:32:34.123456Z
|
|
"%Y/%m/%d %H:%M:%S", # 2023/05/01 08:32:34
|
|
"%a %d %b %Y %H:%M:%S %Z", # Wed 01 Jan 2020 00:00:00 GMT
|
|
"%a %d %b %Y %H:%M:%S %z", # Wed 01 Jan 2020 00:00:00 -0400
|
|
"%a %b %d %H:%M:%S %Z %Y", # Wed Jan 01 00:00:00 UTC 2020
|
|
"%a %b %d %H:%M:%S %z %Y", # Wed Jan 01 00:00:00 -0400 2020
|
|
"%b %d %Y %I:%M %p", # Jan 1 2020 0:00 pm
|
|
"%Y%m%d%H%M%S", # 20230501083234
|
|
]), to_utc: bool = True) -> datetime.datetime:
|
|
"""Parse a given text representing a datetime using a list of formats,
|
|
optionally converting it to UTC.
|
|
"""
|
|
# so that we don't have to deal with some special cases in formats
|
|
text = (
|
|
text.strip()
|
|
.replace(", ", " ") # November 10, 2015 -> November 10 2015
|
|
.replace(" ,", " ") # November 10 ,2015 -> November 10 2015
|
|
.replace(". ", " ") # November 10. 2015 -> November 10 2015
|
|
.replace("(", "") # (November 10 2015) -> November 10 2015)
|
|
.replace(")", "") # (November 10 2015) -> (November 10 2015
|
|
.replace("Sept ", "Sep ") # 11 Sept 2025 -> 11 Sep 2025
|
|
.replace("Sept-", "Sep-") # 11-Sept-2025 -> 11-Sep-2025
|
|
.replace("sept ", "sep ") # 11 Sept 2025 -> 11 Sep 2025
|
|
.replace("sept-", "sep-") # 11-Sept-2025 -> 11-Sep-2025
|
|
)
|
|
for fmt in formats:
|
|
try:
|
|
dt = datetime.datetime.strptime(text, fmt) # NOQA: DTZ007, timezone is handled below
|
|
|
|
if to_utc:
|
|
dt = dt.astimezone(datetime.timezone.utc)
|
|
elif dt.tzinfo is None:
|
|
dt = dt.replace(tzinfo=datetime.timezone.utc)
|
|
|
|
return dt
|
|
except ValueError:
|
|
pass
|
|
|
|
msg = f"'{text}' could not be parsed as a date with any of the formats: {str(formats)}"
|
|
raise ValueError(msg)
|
|
|
|
|
|
def parse__datetime_or_date_or_month_year_date(text: str) -> datetime.datetime:
|
|
"""Parse a given text representing a datetime, date or a partial date using the default list of formats.
|
|
"""
|
|
try:
|
|
return parse_datetime(text)
|
|
except ValueError:
|
|
return parse_date_or_month_year_date(text)
|
|
|
|
|
|
def date(year: int, month: int, day: int) -> datetime.datetime:
|
|
"""Create a datetime object with the given year, month and day, at midnight."""
|
|
return datetime.datetime(year, month, day, tzinfo=datetime.timezone.utc)
|
|
|
|
|
|
def today_at_midnight() -> datetime.datetime:
|
|
"""Create a datetime object with today's date, at midnight."""
|
|
return datetime.datetime.now(tz=datetime.timezone.utc).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
|
|
def today_at_end_of_day() -> datetime.datetime:
|
|
"""Create a datetime object with today's date, at the end of the day."""
|
|
return datetime.datetime.now(tz=datetime.timezone.utc).replace(hour=23, minute=59, second=59, microsecond=999999)
|
|
|
|
def today() -> datetime.date:
|
|
"""Create a date object with today's date."""
|
|
return datetime.datetime.now(tz=datetime.timezone.utc).date()
|
|
|
|
def to_datetime(dt: datetime.date) -> datetime.datetime:
|
|
"""Convert a date object to a datetime object at midnight."""
|
|
return datetime.datetime(dt.year, dt.month, dt.day, tzinfo=datetime.timezone.utc)
|