Consider the water temperature to be out of date 6 hours after measur…

…ement (#40)

Co-authored-by: Maciej Bieniek <[email protected]>

imgw_pib/__init__.py

@@ -1,7 +1,7 @@
 """Python wrapper for IMGW-PIB API."""
 
 import logging
-from datetime import UTC, datetime
+from datetime import UTC, datetime, timedelta
 from http import HTTPStatus
 from typing import Any, Self
 
@@ -254,10 +254,9 @@ def _parse_hydrological_data(self: Self, data: dict[str, Any]) -> HydrologicalDa
             data[ApiNames.WATER_TEMPERATURE_MEASUREMENT_DATE],
             "%Y-%m-%d %H:%M:%S",
         )
-        if (
-            water_temperature_measurement_date is not None
-            and water_temperature_measurement_date.date() == datetime.now(tz=UTC).date()
-        ):
+        if water_temperature_measurement_date is not None and datetime.now(
+            tz=UTC
+        ) - water_temperature_measurement_date < timedelta(hours=6):
             water_temperature = data[ApiNames.WATER_TEMPERATURE]
         else:
             water_temperature_measurement_date = None

setup.py

@@ -6,7 +6,7 @@
 
 PROJECT_DIR = Path(__file__).parent.resolve()
 README_FILE = PROJECT_DIR / "README.md"
-VERSION = "1.0.3"
+VERSION = "1.0.4"
 
 setup(
     name="imgw_pib",

tests/test_init.py

@@ -384,7 +384,8 @@ async def test_water_temperature_not_current(
     """Test water_temperature is not current."""
     session = aiohttp.ClientSession()
 
-    hydrological_station["temperatura_wody_data_pomiaru"] = "2002-01-01 12:00:01"
+    # The measurement was performed more than 6 hours before the test time
+    hydrological_station["temperatura_wody_data_pomiaru"] = "2024-04-22 05:00:00"
 
     with aioresponses() as session_mock, freeze_time(TEST_TIME):
         session_mock.get(API_HYDROLOGICAL_ENDPOINT, payload=hydrological_stations)