Mercurial > libervia-backend
changeset 3879:46930301f0c1
tools: renamed module `sat.tools.datetime` to `date.tools.xmpp_datetime` to avoid conflict with Python's standard lib
author | Goffi <goffi@goffi.org> |
---|---|
date | Wed, 31 Aug 2022 13:18:56 +0200 |
parents | 32087d7c25d4 |
children | 03761f8ba8bb |
files | sat/plugins/plugin_xep_0082.py sat/tools/datetime.py sat/tools/utils.py sat/tools/xmpp_datetime.py |
diffstat | 4 files changed, 200 insertions(+), 197 deletions(-) [+] |
line wrap: on
line diff
--- a/sat/plugins/plugin_xep_0082.py Wed Aug 31 13:11:26 2022 +0200 +++ b/sat/plugins/plugin_xep_0082.py Wed Aug 31 13:18:56 2022 +0200 @@ -23,7 +23,7 @@ from sat.core.constants import Const as C from sat.core.i18n import D_ from sat.core.sat_main import SAT -from sat.tools import datetime +from sat.tools import xmpp_datetime __all__ = [ # pylint: disable=unused-variable @@ -49,7 +49,7 @@ """ Implementation of the date and time profiles specified in XEP-0082 using Python's datetime module. The legacy format described in XEP-0082 section "4. Migration" is not - supported. Reexports of the functions in :mod:`sat.tools.datetime`. + supported. Reexports of the functions in :mod:`sat.tools.xmpp_datetime`. This is a passive plugin, i.e. it doesn't hook into any triggers to process stanzas actively, but offers API for other plugins to use. @@ -60,9 +60,9 @@ @param sat: The SAT instance. """ - format_date = staticmethod(datetime.format_date) - parse_date = staticmethod(datetime.parse_date) - format_datetime = staticmethod(datetime.format_datetime) - parse_datetime = staticmethod(datetime.parse_datetime) - format_time = staticmethod(datetime.format_time) - parse_time = staticmethod(datetime.parse_time) + format_date = staticmethod(xmpp_datetime.format_date) + parse_date = staticmethod(xmpp_datetime.parse_date) + format_datetime = staticmethod(xmpp_datetime.format_datetime) + parse_datetime = staticmethod(xmpp_datetime.parse_datetime) + format_time = staticmethod(xmpp_datetime.format_time) + parse_time = staticmethod(xmpp_datetime.parse_time)
--- a/sat/tools/datetime.py Wed Aug 31 13:11:26 2022 +0200 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,187 +0,0 @@ -#!/usr/bin/env python3 - -# Libervia: XMPP Date and Time profiles as per XEP-0082 -# Copyright (C) 2022-2022 Tim Henkes (me@syndace.dev) - -# This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU Affero General Public License as published by -# the Free Software Foundation, either version 3 of the License, or -# (at your option) any later version. - -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU Affero General Public License for more details. - -# You should have received a copy of the GNU Affero General Public License -# along with this program. If not, see <http://www.gnu.org/licenses/>. - -# Type-check with `mypy --strict` -# Lint with `pylint` - -from sat.core import exceptions -from datetime import date, datetime, time, timezone -import re -from typing import Optional, Tuple - - -__all__ = [ # pylint: disable=unused-variable - "format_date", - "parse_date", - "format_datetime", - "parse_datetime", - "format_time", - "parse_time" -] - - -def __parse_fraction_of_a_second(value: str) -> Tuple[str, Optional[int]]: - """ - datetime's strptime only supports up to six digits of the fraction of a seconds, while - the XEP-0082 specification allows for any number of digits. This function parses and - removes the optional fraction of a second from the input string. - - @param value: The input string, containing a section of the format [.sss]. - @return: The input string with the fraction of a second removed, and the fraction of a - second parsed with microsecond resolution. Returns the unaltered input string and - ``None`` if no fraction of a second was found in the input string. - """ - - # The following regex matches the optional fraction of a seconds for manual - # processing. - match = re.search(r"\.(\d*)", value) - microsecond: Optional[int] = None - if match is not None: - # Remove the fraction of a second from the input string - value = value[:match.start()] + value[match.end():] - - # datetime supports microsecond resolution for the fraction of a second, thus - # limit/pad the parsed fraction of a second to six digits - microsecond = int(match.group(1)[:6].ljust(6, '0')) - - return value, microsecond - - -def format_date(value: Optional[date] = None) -> str: - """ - @param value: The date for format. Defaults to the current date in the UTC timezone. - @return: The date formatted according to the Date profile specified in XEP-0082. - - @warning: Formatting of the current date in the local timezone may leak geographical - information of the sender. Thus, it is advised to only format the current date in - UTC. - """ - # CCYY-MM-DD - - # The Date profile of XEP-0082 is equal to the ISO 8601 format. - return (datetime.now(timezone.utc).date() if value is None else value).isoformat() - - -def parse_date(value: str) -> date: - """ - @param value: A string containing date information formatted according to the Date - profile specified in XEP-0082. - @return: The date parsed from the input string. - @raise ValueError: if the input string is not correctly formatted. - """ - # CCYY-MM-DD - - # The Date profile of XEP-0082 is equal to the ISO 8601 format. - return date.fromisoformat(value) - - -def format_datetime( - value: Optional[datetime] = None, - include_microsecond: bool = False -) -> str: - """ - @param value: The datetime to format. Defaults to the current datetime. - must be an aware datetime object (timezone must be specified) - @param include_microsecond: Include the microsecond of the datetime in the output. - @return: The datetime formatted according to the DateTime profile specified in - XEP-0082. The datetime is always converted to UTC before formatting to avoid - leaking geographical information of the sender. - """ - # CCYY-MM-DDThh:mm:ss[.sss]TZD - - # We format the time in UTC, since the %z formatter of strftime doesn't include colons - # to separate hours and minutes which is required by XEP-0082. UTC allows us to put a - # simple letter 'Z' as the time zone definition. - if value is not None: - if value.tzinfo is None: - raise exceptions.InternalError( - "an aware datetime object must be used, but a naive one has been provided" - ) - value = value.astimezone(timezone.utc) # pylint: disable=no-member - else: - value = datetime.now(timezone.utc) - - if include_microsecond: - return value.strftime("%Y-%m-%dT%H:%M:%S.%fZ") - - return value.strftime("%Y-%m-%dT%H:%M:%SZ") - - -def parse_datetime(value: str) -> datetime: - """ - @param value: A string containing datetime information formatted according to the - DateTime profile specified in XEP-0082. - @return: The datetime parsed from the input string. - @raise ValueError: if the input string is not correctly formatted. - """ - # CCYY-MM-DDThh:mm:ss[.sss]TZD - - value, microsecond = __parse_fraction_of_a_second(value) - - result = datetime.strptime(value, "%Y-%m-%dT%H:%M:%S%z") - - if microsecond is not None: - result = result.replace(microsecond=microsecond) - - return result - - -def format_time(value: Optional[time] = None, include_microsecond: bool = False) -> str: - """ - @param value: The time to format. Defaults to the current time in the UTC timezone. - @param include_microsecond: Include the microsecond of the time in the output. - @return: The time formatted according to the Time profile specified in XEP-0082. - - @warning: Since accurate timezone conversion requires the date to be known, this - function cannot convert input times to UTC before formatting. This means that - geographical information of the sender may be leaked if a time in local timezone - is formatted. Thus, when passing a time to format, it is advised to pass the time - in UTC if possible. - """ - # hh:mm:ss[.sss][TZD] - - if value is None: - # There is no time.now() method as one might expect, but the current time can be - # extracted from a datetime object including time zone information. - value = datetime.now(timezone.utc).timetz() - - # The format created by time.isoformat complies with the XEP-0082 Time profile. - return value.isoformat("auto" if include_microsecond else "seconds") - - -def parse_time(value: str) -> time: - """ - @param value: A string containing time information formatted according to the Time - profile specified in XEP-0082. - @return: The time parsed from the input string. - @raise ValueError: if the input string is not correctly formatted. - """ - # hh:mm:ss[.sss][TZD] - - value, microsecond = __parse_fraction_of_a_second(value) - - # The format parsed by time.fromisoformat mostly complies with the XEP-0082 Time - # profile, except that it doesn't handle the letter Z as time zone information for - # UTC. This can be fixed with a simple string replacement of 'Z' with "+00:00", which - # is another way to represent UTC. - result = time.fromisoformat(value.replace('Z', "+00:00")) - - if microsecond is not None: - result = result.replace(microsecond=microsecond) - - return result
--- a/sat/tools/utils.py Wed Aug 31 13:11:26 2022 +0200 +++ b/sat/tools/utils.py Wed Aug 31 13:18:56 2022 +0200 @@ -34,7 +34,7 @@ from twisted.internet import defer from sat.core.constants import Const as C from sat.core.log import getLogger -from sat.tools.datetime import format_date, format_datetime +from sat.tools import xmpp_datetime log = getLogger(__name__) @@ -157,7 +157,10 @@ datetime.timezone.utc ) - return format_datetime(dtime) if with_time else format_date(dtime.date()) + return ( + xmpp_datetime.format_datetime(dtime) if with_time + else xmpp_datetime.format_date(dtime.date()) + ) def generatePassword(vocabulary=None, size=20):
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/sat/tools/xmpp_datetime.py Wed Aug 31 13:18:56 2022 +0200 @@ -0,0 +1,187 @@ +#!/usr/bin/env python3 + +# Libervia: XMPP Date and Time profiles as per XEP-0082 +# Copyright (C) 2022-2022 Tim Henkes (me@syndace.dev) + +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. + +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. + +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +# Type-check with `mypy --strict` +# Lint with `pylint` + +from sat.core import exceptions +from datetime import date, datetime, time, timezone +import re +from typing import Optional, Tuple + + +__all__ = [ # pylint: disable=unused-variable + "format_date", + "parse_date", + "format_datetime", + "parse_datetime", + "format_time", + "parse_time" +] + + +def __parse_fraction_of_a_second(value: str) -> Tuple[str, Optional[int]]: + """ + datetime's strptime only supports up to six digits of the fraction of a seconds, while + the XEP-0082 specification allows for any number of digits. This function parses and + removes the optional fraction of a second from the input string. + + @param value: The input string, containing a section of the format [.sss]. + @return: The input string with the fraction of a second removed, and the fraction of a + second parsed with microsecond resolution. Returns the unaltered input string and + ``None`` if no fraction of a second was found in the input string. + """ + + # The following regex matches the optional fraction of a seconds for manual + # processing. + match = re.search(r"\.(\d*)", value) + microsecond: Optional[int] = None + if match is not None: + # Remove the fraction of a second from the input string + value = value[:match.start()] + value[match.end():] + + # datetime supports microsecond resolution for the fraction of a second, thus + # limit/pad the parsed fraction of a second to six digits + microsecond = int(match.group(1)[:6].ljust(6, '0')) + + return value, microsecond + + +def format_date(value: Optional[date] = None) -> str: + """ + @param value: The date for format. Defaults to the current date in the UTC timezone. + @return: The date formatted according to the Date profile specified in XEP-0082. + + @warning: Formatting of the current date in the local timezone may leak geographical + information of the sender. Thus, it is advised to only format the current date in + UTC. + """ + # CCYY-MM-DD + + # The Date profile of XEP-0082 is equal to the ISO 8601 format. + return (datetime.now(timezone.utc).date() if value is None else value).isoformat() + + +def parse_date(value: str) -> date: + """ + @param value: A string containing date information formatted according to the Date + profile specified in XEP-0082. + @return: The date parsed from the input string. + @raise ValueError: if the input string is not correctly formatted. + """ + # CCYY-MM-DD + + # The Date profile of XEP-0082 is equal to the ISO 8601 format. + return date.fromisoformat(value) + + +def format_datetime( + value: Optional[datetime] = None, + include_microsecond: bool = False +) -> str: + """ + @param value: The datetime to format. Defaults to the current datetime. + must be an aware datetime object (timezone must be specified) + @param include_microsecond: Include the microsecond of the datetime in the output. + @return: The datetime formatted according to the DateTime profile specified in + XEP-0082. The datetime is always converted to UTC before formatting to avoid + leaking geographical information of the sender. + """ + # CCYY-MM-DDThh:mm:ss[.sss]TZD + + # We format the time in UTC, since the %z formatter of strftime doesn't include colons + # to separate hours and minutes which is required by XEP-0082. UTC allows us to put a + # simple letter 'Z' as the time zone definition. + if value is not None: + if value.tzinfo is None: + raise exceptions.InternalError( + "an aware datetime object must be used, but a naive one has been provided" + ) + value = value.astimezone(timezone.utc) # pylint: disable=no-member + else: + value = datetime.now(timezone.utc) + + if include_microsecond: + return value.strftime("%Y-%m-%dT%H:%M:%S.%fZ") + + return value.strftime("%Y-%m-%dT%H:%M:%SZ") + + +def parse_datetime(value: str) -> datetime: + """ + @param value: A string containing datetime information formatted according to the + DateTime profile specified in XEP-0082. + @return: The datetime parsed from the input string. + @raise ValueError: if the input string is not correctly formatted. + """ + # CCYY-MM-DDThh:mm:ss[.sss]TZD + + value, microsecond = __parse_fraction_of_a_second(value) + + result = datetime.strptime(value, "%Y-%m-%dT%H:%M:%S%z") + + if microsecond is not None: + result = result.replace(microsecond=microsecond) + + return result + + +def format_time(value: Optional[time] = None, include_microsecond: bool = False) -> str: + """ + @param value: The time to format. Defaults to the current time in the UTC timezone. + @param include_microsecond: Include the microsecond of the time in the output. + @return: The time formatted according to the Time profile specified in XEP-0082. + + @warning: Since accurate timezone conversion requires the date to be known, this + function cannot convert input times to UTC before formatting. This means that + geographical information of the sender may be leaked if a time in local timezone + is formatted. Thus, when passing a time to format, it is advised to pass the time + in UTC if possible. + """ + # hh:mm:ss[.sss][TZD] + + if value is None: + # There is no time.now() method as one might expect, but the current time can be + # extracted from a datetime object including time zone information. + value = datetime.now(timezone.utc).timetz() + + # The format created by time.isoformat complies with the XEP-0082 Time profile. + return value.isoformat("auto" if include_microsecond else "seconds") + + +def parse_time(value: str) -> time: + """ + @param value: A string containing time information formatted according to the Time + profile specified in XEP-0082. + @return: The time parsed from the input string. + @raise ValueError: if the input string is not correctly formatted. + """ + # hh:mm:ss[.sss][TZD] + + value, microsecond = __parse_fraction_of_a_second(value) + + # The format parsed by time.fromisoformat mostly complies with the XEP-0082 Time + # profile, except that it doesn't handle the letter Z as time zone information for + # UTC. This can be fixed with a simple string replacement of 'Z' with "+00:00", which + # is another way to represent UTC. + result = time.fromisoformat(value.replace('Z', "+00:00")) + + if microsecond is not None: + result = result.replace(microsecond=microsecond) + + return result