codex
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
#! python3 # noqa: E265
|
|
|
|
|
|
"""
|
|
Manage timezones for pages date(time)s using zoneinfo module, added in Python 3.9.
|
|
|
|
"""
|
|
|
|
# ############################################################################
|
|
# ########## Libraries #############
|
|
# ##################################
|
|
|
|
# standard library
|
|
from datetime import datetime, timezone
|
|
from zoneinfo import ZoneInfo
|
|
|
|
# 3rd party
|
|
from mkdocs.plugins import get_plugin_logger
|
|
|
|
# package
|
|
from mkdocs_rss_plugin.constants import MKDOCS_LOGGER_NAME
|
|
|
|
# ############################################################################
|
|
# ########## Globals #############
|
|
# ################################
|
|
|
|
|
|
logger = get_plugin_logger(MKDOCS_LOGGER_NAME)
|
|
|
|
|
|
# ############################################################################
|
|
# ########## Functions ###########
|
|
# ################################
|
|
|
|
|
|
def set_datetime_zoneinfo(
|
|
input_datetime: datetime, config_timezone: str = "UTC"
|
|
) -> datetime:
|
|
"""Apply timezone to a naive datetime.
|
|
|
|
Args:
|
|
input_datetime (datetime): offset-naive datetime
|
|
config_timezone (str, optional): name of timezone as registered in IANA
|
|
database. Defaults to "UTC". Example : Europe/Paris.
|
|
|
|
Returns:
|
|
datetime: offset-aware datetime
|
|
"""
|
|
if input_datetime.tzinfo:
|
|
return input_datetime
|
|
elif not config_timezone:
|
|
return input_datetime.replace(tzinfo=timezone.utc)
|
|
else:
|
|
config_tz = ZoneInfo(config_timezone)
|
|
return input_datetime.replace(tzinfo=config_tz)
|