chore: fix CI issues, linting, and type safety

- Fix syntax errors and truncated code in src/nwp500/models/schedule.py
- Add 'from __future__ import annotations' to all source files to fix runtime NameErrors
- Resolve all ruff linting and formatting issues (E501, F401, etc.)
- Fix pyright type-checking errors in AWS credential handling and model methods
- Update tests/test_reservations.py to match high-level MQTT client API
- Restore AuthTokens.to_dict() for compatibility

Author: eman

examples/advanced/recirculation_control.py

@@ -119,9 +119,7 @@ def on_button_only_set(status):
                 button_only_set = True
 
             await mqtt_client.subscribe_device_status(device, on_button_only_set)
-            await mqtt_client.set_recirculation_mode(
-                device, 2
-            )  # 2 = Button Only
+            await mqtt_client.set_recirculation_mode(device, 2)  # 2 = Button Only
 
             # Wait for confirmation
             for i in range(10):  # Wait up to 10 seconds

src/nwp500/__init__.py

@@ -4,6 +4,8 @@
 communication for NWP500 heat pump water heaters.
 """
 
+from __future__ import annotations
+
 from importlib.metadata import (
     PackageNotFoundError,
     version,

src/nwp500/auth.py

@@ -15,7 +15,6 @@
 
 import json
 import logging
-import warnings
 from datetime import UTC, datetime, timedelta
 from typing import Any, Self, cast
 
@@ -182,6 +181,21 @@ def bearer_token(self) -> str:
         """Get the formatted Bearer token for Authorization header."""
         return f"Bearer {self.access_token}"
 
+    def to_dict(self) -> dict[str, Any]:
+        """Convert tokens to a dictionary for serialization.
+
+        This includes the calculated issued_at timestamp, which is needed
+        to maintain the correct expiration time when restoring tokens.
+        """
+        data = self.model_dump()
+        # Ensure issued_at is serialized in a format that model_validate can
+        # parse
+        if isinstance(data.get("issued_at"), datetime):
+            data["issued_at"] = (
+                data["issued_at"].strftime("%Y-%m-%dT%H:%M:%S.%f") + "Z"
+            )
+        return data
+
 
 class AuthenticationResponse(NavienBaseModel):
     """Complete authentication response including user info and tokens."""
@@ -428,7 +442,9 @@ async def sign_in(
                     )
 
                 # Parse successful response
-                auth_response = AuthenticationResponse.model_validate(response_data)
+                auth_response = AuthenticationResponse.model_validate(
+                    response_data
+                )
                 self._auth_response = auth_response
                 self._user_email = user_id  # Store the email for later use
 

src/nwp500/cli/__init__.py

@@ -1,5 +1,7 @@
 """CLI package for nwp500-python."""
 
+from __future__ import annotations
+
 from .__main__ import run
 from .handlers import (
     handle_device_info_request,

src/nwp500/cli/__main__.py

@@ -1,5 +1,7 @@
 """Navien Water Heater Control CLI - Main Entry Point."""
 
+from __future__ import annotations
+
 import asyncio
 import functools
 import logging

src/nwp500/cli/commands.py

@@ -1,5 +1,7 @@
 """Command registry for NWP500 CLI."""
 
+from __future__ import annotations
+
 from collections.abc import Callable
 from dataclasses import dataclass
 from typing import Any

src/nwp500/cli/handlers.py

@@ -1,5 +1,7 @@
 """Command handlers for CLI operations."""
 
+from __future__ import annotations
+
 import asyncio
 import json
 import logging
@@ -328,9 +330,7 @@ def raw_callback(topic: str, message: dict[str, Any]) -> None:
         device_type, mqtt.client_id, "rsv/rd"
     )
     await mqtt.subscribe(response_topic, raw_callback)
-    await mqtt.update_reservations(
-        device, reservations, enabled=enabled
-    )
+    await mqtt.update_reservations(device, reservations, enabled=enabled)
     try:
         await asyncio.wait_for(future, timeout=10)
     except TimeoutError:

src/nwp500/cli/monitoring.py

@@ -1,5 +1,7 @@
 """Monitoring and periodic status polling."""
 
+from __future__ import annotations
+
 import asyncio
 import logging
 
@@ -41,9 +43,7 @@ def on_status_update(status: DeviceStatus) -> None:
 
     await mqtt.subscribe_device_status(device, on_status_update)
     await mqtt.start_periodic_requests(device, period_seconds=30)
-    await mqtt.request_device_status(
-        device
-    )  # Get an initial status right away
+    await mqtt.request_device_status(device)  # Get an initial status right away
 
     # Keep the script running indefinitely
     await asyncio.Event().wait()

src/nwp500/cli/output_formatters.py

@@ -1,5 +1,7 @@
 """Output formatting utilities for CLI (CSV, JSON)."""
 
+from __future__ import annotations
+
 import csv
 import json
 import logging

src/nwp500/cli/rich_output.py

@@ -1,5 +1,7 @@
 """Rich-enhanced output formatting with graceful fallback."""
 
+from __future__ import annotations
+
 import json
 import logging
 import os