force utc timezone for datetime.fromisoformat

laixintao · Apr 11, 2024 · e98349f · e98349f
1 parent 79b4be1
commit e98349f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/iredis/completers.py b/iredis/completers.py
@@ -1,6 +1,6 @@
 import logging
 from typing import Iterable
-from datetime import datetime
+from datetime import datetime, timezone
 
 from dateutil.relativedelta import relativedelta
 from prompt_toolkit.completion import (
@@ -123,7 +123,7 @@ def _completion_humanize_time(self, document: Document) -> Iterable[Completion]:
     def _completion_formatted_time(self, document: Document) -> Iterable[Completion]:
         text = document.text
         try:
-            dt = datetime.fromisoformat(text)
+            dt = datetime.fromisoformat(text).replace(tzinfo=timezone.utc)
         except Exception:
             return
         yield Completion(

diff --git a/tests/unittests/test_completers.py b/tests/unittests/test_completers.py
@@ -347,7 +347,7 @@ def test_timestamp_completer_datetime_format_time_completion():
             text="1581033600000",
             start_position=-10,
             display=FormattedText([("", "1581033600000")]),
-            display_meta="2020-02-07T00:00:00+00:00",
+            display_meta="2020-02-07 00:00:00+00:00",
         )
     ]