Skip to content

Commit 9cf0303

Browse files
authored
Merge pull request #161 from dknowles2/mypy
Fix typing errors and deserialization errors
2 parents 7e50ac0 + 5e9b6df commit 9cf0303

File tree

5 files changed

+115
-77
lines changed

5 files changed

+115
-77
lines changed

pydrawise/legacy.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
from datetime import datetime, timedelta
77
import time
8+
from typing import Any
89

910
import aiohttp
1011
import requests
@@ -237,7 +238,7 @@ def _zone_from_json(zone_json: dict) -> Zone:
237238
remaining_time=timedelta(seconds=zone_json["run"]),
238239
)
239240
elif zone_json["time"] == 1576800000:
240-
suspended_until = ZoneSuspension(end_time=datetime.max)
241+
suspended_until = datetime.max
241242
else:
242243
now = datetime.now().replace(microsecond=0)
243244
start_time = now + timedelta(seconds=zone_json["time"])
@@ -270,8 +271,8 @@ class LegacyHydrawise:
270271

271272
def __init__(self, user_token: str, load_on_init: bool = True) -> None:
272273
self._api_key = user_token
273-
self.controller_info = {}
274-
self.controller_status = {}
274+
self.controller_info: dict[str, Any] = {}
275+
self.controller_status: dict[str, Any] = {}
275276
if load_on_init:
276277
self.update_controller_info()
277278

@@ -350,7 +351,7 @@ def _get_controller_status(self) -> dict:
350351
return self._get("statusschedule.php")
351352

352353
def suspend_zone(self, days: int, zone: int | None = None) -> dict:
353-
params = {}
354+
params: dict[str, Any] = {}
354355

355356
if days > 0:
356357
params["custom"] = int(time.time() + (days * 24 * 60 * 60))
@@ -370,7 +371,7 @@ def suspend_zone(self, days: int, zone: int | None = None) -> dict:
370371
return self._get("setzone.php", **params)
371372

372373
def run_zone(self, minutes: int, zone: int | None = None) -> dict:
373-
params = {}
374+
params: dict[str, Any] = {}
374375

375376
if zone is not None:
376377
if not self.relays:

0 commit comments

Comments
 (0)