Add DTZ005 mypy rule

music_assistant/providers/ard_audiothek/__init__.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from collections.abc import AsyncGenerator, Sequence
-from datetime import datetime, timedelta
+from datetime import datetime
 from typing import TYPE_CHECKING, Any
 
 from gql import Client
@@ -36,6 +36,7 @@
 
 from music_assistant.constants import CONF_PASSWORD
 from music_assistant.controllers.cache import use_cache
+from music_assistant.helpers.datetime import from_utc_timestamp, future_timestamp, utc
 from music_assistant.models.music_provider import MusicProvider
 from music_assistant.providers.ard_audiothek.database_queries import (
     get_history_query,
@@ -248,26 +249,22 @@ async def get_client(self) -> Client:
         _password = self.config.get_value(CONF_PASSWORD)
         self.token = self.config.get_value(CONF_TOKEN_BEARER)
         self.user_id = self.config.get_value(CONF_USERID)
-        self.token_expire = datetime.fromtimestamp(
-            float(str(self.config.get_value(CONF_EXPIRY_TIME)))
-        )
+        self.token_expire = from_utc_timestamp(float(str(self.config.get_value(CONF_EXPIRY_TIME))))
 
         self.max_bitrate = int(float(str(self.config.get_value(CONF_MAX_BITRATE))))
 
         if (
             _email is not None
             and _password is not None
-            and (self.token is None or self.user_id is None or self.token_expire < datetime.now())
+            and (self.token is None or self.user_id is None or self.token_expire < utc())
         ):
             self.token, self.user_id, _display_name = await _login(
                 self.mass.http_session, str(_email), str(_password)
             )
             self._update_config_value(CONF_TOKEN_BEARER, self.token, encrypted=True)
             self._update_config_value(CONF_USERID, self.user_id, encrypted=True)
             self._update_config_value(CONF_DISPLAY_NAME, _display_name)
-            self._update_config_value(
-                CONF_EXPIRY_TIME, str((datetime.now() + timedelta(hours=1)).timestamp())
-            )
+            self._update_config_value(CONF_EXPIRY_TIME, str(future_timestamp(hours=1)))
             self._client_initialized = False
 
         if not self._client_initialized:

music_assistant/providers/deezer/gw_client.py

@@ -4,7 +4,6 @@
 cookie based on the api_token.
 """
 
-import datetime
 import json
 from collections.abc import Mapping
 from http.cookies import BaseCookie, Morsel
@@ -15,7 +14,7 @@
 from music_assistant_models.streamdetails import StreamDetails
 from yarl import URL
 
-from music_assistant.helpers.datetime import utc_timestamp
+from music_assistant.helpers.datetime import future_timestamp, utc_timestamp
 
 USER_AGENT_HEADER = (
     "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) "
@@ -89,10 +88,7 @@ async def setup(self) -> None:
         await self._update_user_data()
 
     async def _get_license(self) -> str | None:
-        if (
-            self._license_expiration_timestamp
-            < (datetime.datetime.now() + datetime.timedelta(days=1)).timestamp()
-        ):
+        if self._license_expiration_timestamp < future_timestamp(days=1):
             await self._update_user_data()
         return self._license
 

music_assistant/providers/phishin/provider.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 from collections.abc import AsyncGenerator
-from datetime import datetime
 from typing import TYPE_CHECKING, Any
 
 from music_assistant_models.enums import (
@@ -30,6 +29,7 @@
 from music_assistant_models.unique_list import UniqueList
 
 from music_assistant.controllers.cache import use_cache
+from music_assistant.helpers.datetime import now
 from music_assistant.models.music_provider import MusicProvider
 
 from .constants import (
@@ -642,7 +642,7 @@ async def _browse_random(self) -> list[Album]:
     async def _browse_today(self) -> list[Album]:
         """Get shows that happened on this day in history."""
         try:
-            today = datetime.now()
+            today = now()
             target_date = today.strftime("%Y-%m-%d")
 
             shows_data = await api_request(

music_assistant/providers/spotify_connect/events.py

@@ -5,12 +5,12 @@
 import json
 import os
 import urllib.request
-from datetime import datetime
+from datetime import UTC, datetime
 
 player_event = os.getenv("PLAYER_EVENT")
 
 json_dict = {
-    "event_time": str(datetime.now()),
+    "event_time": str(datetime.now(UTC)),
     "event": player_event,
 }
 

pyproject.toml

@@ -236,7 +236,6 @@ ignore = [
   "PYI034",
   "G004",
   "PGH003",
-  "DTZ005",
   "S104",
   "S105",
   "S106",