Simplify _estimate_duration in snapshot.py

src/ert/gui/model/snapshot.py

@@ -1,10 +1,9 @@
-import datetime
 import logging
 from collections import defaultdict
 from contextlib import ExitStack
+from datetime import datetime, timedelta
 from typing import Any, Dict, Final, List, Optional, Sequence, Union, overload
 
-from dateutil import tz
 from qtpy.QtCore import QAbstractItemModel, QModelIndex, QObject, QSize, Qt, QVariant
 from qtpy.QtGui import QColor, QFont
 from typing_extensions import override
@@ -68,13 +67,10 @@
 
 
 def _estimate_duration(
-    start_time: datetime.datetime, end_time: Optional[datetime.datetime] = None
-) -> datetime.timedelta:
-    timezone = None
-    if start_time.tzname() is not None:
-        timezone = tz.gettz(start_time.tzname())
+    start_time: datetime, end_time: Optional[datetime] = None
+) -> timedelta:
     if not end_time or end_time < start_time:
-        end_time = datetime.datetime.now(timezone)
+        end_time = datetime.now(start_time.tzinfo)
     return end_time - start_time
 
 
@@ -441,7 +437,7 @@ def _job_data(
                     start_time, end_time=node.data.get(ids.END_TIME)
                 )
                 # There is no method for truncating microseconds, so we remove them
-                delta -= datetime.timedelta(microseconds=delta.microseconds)
+                delta -= timedelta(microseconds=delta.microseconds)
                 return str(delta)
 
             return node.data.get(data_name)

tests/unit_tests/gui/model/test_job_list.py

@@ -1,4 +1,4 @@
-import datetime
+from datetime import datetime
 from unittest.mock import patch
 
 import pytest
@@ -60,8 +60,8 @@ def test_changes(full_snapshot):
     assert model.index(0, _id_to_col(ids.STATUS)).data() == FORWARD_MODEL_STATE_START
 
     partial = PartialSnapshot(full_snapshot)
-    start_time = datetime.datetime(year=2020, month=10, day=27, hour=12)
-    end_time = datetime.datetime(year=2020, month=10, day=28, hour=13)
+    start_time = datetime(year=2020, month=10, day=27, hour=12)
+    end_time = datetime(year=2020, month=10, day=28, hour=13)
     partial.update_forward_model(
         "0",
         "0",
@@ -102,11 +102,9 @@ def test_duration(mock_datetime, timezone, full_snapshot):
     )
 
     partial = PartialSnapshot(full_snapshot)
-    start_time = datetime.datetime(
-        year=2020, month=10, day=27, hour=12, tzinfo=timezone
-    )
-    # mock only datetime.datetime.now()
-    mock_datetime.datetime.now.return_value = datetime.datetime(
+    start_time = datetime(year=2020, month=10, day=27, hour=12, tzinfo=timezone)
+    # mock only datetime.now()
+    mock_datetime.now.return_value = datetime(
         year=2020,
         month=10,
         day=28,
@@ -128,7 +126,7 @@ def test_duration(mock_datetime, timezone, full_snapshot):
     assert (
         model.index(2, _id_to_col(DURATION), QModelIndex()).data() == "1 day, 1:12:11"
     )
-    mock_datetime.datetime.now.assert_called_once_with(timezone)
+    mock_datetime.now.assert_called_once_with(timezone)
 
 
 @pytest.mark.requires_window_manager