From: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Date: Wed, 14 Feb 2024 01:31:35 +0000 (+1300)
Subject: py:nt_time: add nt_time_from_string()
X-Git-Tag: tdb-1.4.11~1649
X-Git-Url: http://git.ipfire.org/gitweb/?a=commitdiff_plain;h=60022ed55f7a213440b70c6d4b4796a4a7b0fa9f;p=thirdparty%2Fsamba.git

py:nt_time: add nt_time_from_string()

This is for samba-tool, which could do with a common understanding of
time strings across various sub-tools.

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
---

diff --git a/python/samba/nt_time.py b/python/samba/nt_time.py
index 7a6417ecaaa..5feabb06ff9 100644
--- a/python/samba/nt_time.py
+++ b/python/samba/nt_time.py
@@ -19,6 +19,7 @@
 
 import datetime
 from typing import NewType
+import re
 
 
 NtTime = NewType("NtTime", int)
@@ -56,3 +57,46 @@ def nt_time_delta_from_datetime(dt: datetime.timedelta) -> NtTimeDelta:
 
 def timedelta_from_nt_time_delta(nt_time_delta: NtTimeDelta) -> datetime.timedelta:
     return datetime.timedelta(microseconds=nt_time_delta / NT_TICKS_PER_μSEC)
+
+
+def nt_time_from_string(s: str) -> NtTime:
+    """Convert a subset of ISO 8601 date/time strings, ldap timestamps,
+    and the string 'now' into NT time.
+
+    The ldap format is
+
+       YYYYmmddHHMMSS.0Z
+
+    which is 14 digits followed by the fixed string '.0Z'. This is
+    used in LDIF and internally by ldb.
+
+    The ISO format is
+
+    YYYY-mm-dd[*HH[:MM[:SS[.fff[fff]]]][+HH:MM[:SS[.ffffff]]]]
+
+    where the '*' can be any character, and the optional last
+    '[+HH:MM[:SS[.ffffff]]]' is a timezone offset (use '+00:00' for
+    UTC).
+    """
+    try:
+        if s == 'now':
+            dt = datetime.datetime.now(datetime.timezone.utc)
+        elif re.match(r'^\d{14}\.0Z$', s):
+            # "20230127223641.0Z"
+            dt = datetime.strptime(s, '%Y%m%d%H%M%S.0Z')
+        else:
+            dt = datetime.datetime.fromisoformat(s)
+    except ValueError:
+        raise ValueError("Expected a date in either "
+                         "ISO8601 'YYYY-MM-DD HH:MM:SS' format, "
+                         "LDAP timestamp 'YYYYmmddHHMMSS.0Z', "
+                         "or the literal string 'now'. "
+                         f" Got '{s}'.")
+
+    if dt.tzinfo is None:
+        # This is a cursed timestamp with no timezone info. We have to
+        # guess or nt_time_from_datetime() will fail. The best guess
+        # is the system timezone, which we can get this way:
+        dt = dt.astimezone()
+
+    return nt_time_from_datetime(dt)