Update pre-commit and run it

Author: Moustachauve

pyhilo/api.py

@@ -1,17 +1,17 @@
 from __future__ import annotations
 
 import asyncio
-from datetime import datetime, timedelta
 import json
 import random
 import string
 import sys
+from datetime import datetime, timedelta
 from typing import Any, Callable, Union, cast
 from urllib import parse
 
+import backoff
 from aiohttp import ClientSession
 from aiohttp.client_exceptions import ClientResponseError
-import backoff
 from homeassistant.helpers import config_entry_oauth2_flow
 
 from pyhilo.const import (
@@ -129,10 +129,8 @@ def headers(self) -> dict[str, Any]:
         }
         return {
             **headers,
-            **{
-                "Content-Type": "application/json; charset=utf-8",
-                "Ocp-Apim-Subscription-Key": SUBSCRIPTION_KEY,
-            },
+            "Content-Type": "application/json; charset=utf-8",
+            "Ocp-Apim-Subscription-Key": SUBSCRIPTION_KEY,
         }
 
     async def async_get_access_token(self) -> str:
@@ -516,7 +514,7 @@ async def _set_device_attribute(
         self,
         device: HiloDevice,
         key: DeviceAttribute,
-        value: Union[str, float, int, None],
+        value: Union[str, float, None],
     ) -> None:
         url = self._get_url(f"Devices/{device.id}/Attributes", device.location_id)
         LOG.debug(f"Device Attribute URL is {url}")
@@ -545,7 +543,8 @@ async def get_event_notifications(self, location_id: int) -> dict[str, Any]:
           "homePageNotificationBody": "Test manuel de l’alarme détecté.",
           "notificationDataJSON": "{\"NotificationType\":null,\"Title\":\"\",\"SubTitle\":null,\"Body\":\"Test manuel de l’alarme détecté.\",\"Badge\":0,\"Sound\":null,\"Data\":null,\"Tags\":null,\"Type\":\"TestDetected\",\"DeviceId\":324236,\"LocationId\":4051}",
           "viewed": false
-        }"""
+        }
+        """
         url = self._get_url(None, location_id, events=True)
         LOG.debug(f"Event Notifications URL is {url}")
         return cast(dict[str, Any], await self.async_request("get", url))
@@ -613,7 +612,6 @@ async def get_gd_events(
           }
         }
         """
-
         url = self._get_url("Events", location_id, True)
         if not event_id:
             url += "?active=true"

pyhilo/const.py

@@ -15,7 +15,7 @@
 ANDROID_PKG_NAME: Final = "com.hiloenergie.hilo"
 DOMAIN: Final = "hilo"
 # Auth constants
-AUTH_HOSTNAME: Final = "connexion.hiloenergie.com"
+AUTH_HOSTNAME: Final = "connexion.hiloenergie.com"  # codespell:ignore
 AUTH_ENDPOINT: Final = "/HiloDirectoryB2C.onmicrosoft.com/B2C_1A_SIGN_IN/oauth2/v2.0/"
 AUTH_AUTHORIZE: Final = f"https://{AUTH_HOSTNAME}{AUTH_ENDPOINT}authorize"
 AUTH_TOKEN: Final = f"https://{AUTH_HOSTNAME}{AUTH_ENDPOINT}token"

pyhilo/event.py

@@ -1,7 +1,8 @@
-"""Event object """
-from datetime import datetime, timedelta, timezone
+"""Event object"""
+
 import logging
 import re
+from datetime import datetime, timedelta, timezone
 from typing import Any, cast
 
 from pyhilo.util import camel_to_snake, from_utc_timestamp
@@ -72,7 +73,8 @@ def update_wh(self, used_wH: float) -> None:
 
     def should_check_for_allowed_wh(self) -> bool:
         """This function is used to authorize subscribing to a specific event in Hilo to receive the allowed_kWh
-        that is made available in the pre_heat phase"""
+        that is made available in the pre_heat phase
+        """
         now = datetime.now(self.preheat_start.tzinfo)
         time_since_preheat_start = (self.preheat_start - now).total_seconds()
         already_has_allowed_wh = self.allowed_kWh > 0
@@ -150,25 +152,24 @@ def state(self) -> str:
         now = datetime.now(self.preheat_start.tzinfo)
         if self.pre_cold_start and self.pre_cold_start <= now < self.pre_cold_end:
             return "pre_cold"
-        elif (
+        if (
             self.appreciation_start
             and self.appreciation_start <= now < self.appreciation_end
         ):
             return "appreciation"
-        elif self.preheat_start > now:
+        if self.preheat_start > now:
             return "scheduled"
-        elif self.preheat_start <= now < self.preheat_end:
+        if self.preheat_start <= now < self.preheat_end:
             return "pre_heat"
-        elif self.reduction_start <= now < self.reduction_end:
+        if self.reduction_start <= now < self.reduction_end:
             return "reduction"
-        elif self.recovery_start <= now < self.recovery_end:
+        if self.recovery_start <= now < self.recovery_end:
             return "recovery"
-        elif now >= self.recovery_end + timedelta(minutes=5):
+        if now >= self.recovery_end + timedelta(minutes=5):
             return "off"
-        elif now >= self.recovery_end:
+        if now >= self.recovery_end:
             return "completed"
-        elif self.progress:
+        if self.progress:
             return self.progress
 
-        else:
-            return "unknown"
+        return "unknown"

pyhilo/util/__init__.py

@@ -1,7 +1,8 @@
 """Define utility modules."""
+
 import asyncio
-from datetime import datetime, timedelta
 import re
+from datetime import datetime, timedelta
 from typing import Any, Callable
 
 from dateutil import tz

pyhilo/websocket.py

@@ -1,11 +1,12 @@
 """Define a connection to the Hilo websocket."""
+
 from __future__ import annotations
 
 import asyncio
+import json
 from dataclasses import dataclass, field
 from datetime import datetime, timedelta
 from enum import IntEnum
-import json
 from os import environ
 from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple
 from urllib import parse
@@ -338,10 +339,8 @@ async def async_listen(self) -> None:
         except ConnectionClosedError as err:
             LOG.error(f"Websocket: Closed while listening: {err}")
             LOG.exception(err)
-            pass
         except InvalidMessageError as err:
             LOG.warning(f"Websocket: Received invalid json : {err}")
-            pass
         finally:
             LOG.info("Websocket: Listen completed; cleaning up")
             self._watchdog.cancel()
@@ -420,6 +419,7 @@ def __init__(
             async_request: The async request method from the API class
             state_yaml: Path to the state file
             set_state_callback: Callback to save state
+
         """
         self.session = session
         self.async_request = async_request
@@ -445,8 +445,10 @@ async def refresh_token(
         self, config: WebsocketConfig, get_new_token: bool = True
     ) -> None:
         """Refresh token for a specific websocket configuration.
+
         Args:
             config: The websocket configuration to refresh
+
         """
         if get_new_token:
             config.url, self._shared_token = await self._negotiate(config)
@@ -459,10 +461,12 @@ async def refresh_token(
 
     async def _negotiate(self, config: WebsocketConfig) -> Tuple[str, str]:
         """Negotiate websocket connection and get URL and token.
+
         Args:
             config: The websocket configuration to negotiate
         Returns:
             Tuple containing the websocket URL and access token
+
         """
         LOG.debug(f"Getting websocket url for {config.endpoint}")
         url = f"{config.endpoint}/negotiate"
@@ -494,6 +498,7 @@ async def _get_websocket_params(self, config: WebsocketConfig) -> None:
 
         Args:
             config: The websocket configuration to get parameters for
+
         """
         uri = parse.urlparse(config.url)
         LOG.debug(f"Getting websocket params for {config.endpoint}")