From 6e670c8e88ac0a4e181746226c4fd2504fd0eaca Mon Sep 17 00:00:00 2001
From: Ian Foster <ian@vorsk.com>
Date: Thu, 8 Jan 2026 14:25:53 -0800
Subject: [PATCH] fix commit hook warnings

---
 examples/airtag.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/examples/airtag.py b/examples/airtag.py
index a4be599..bbc1dc5 100644
--- a/examples/airtag.py
+++ b/examples/airtag.py
@@ -12,6 +12,8 @@ from pathlib import Path
 from _login import get_account_sync
 
 from findmy import FindMyAccessory
+from findmy.accessory import RollingKeyPairSource
+from findmy.keys import HasHashedPublicKey
 
 # Default path where login session will be stored.
 # This is necessary to avoid generating a new session every time we log in.
@@ -39,19 +41,20 @@ def get_battery_level(status: int) -> str:
     return BATTERY_LEVEL.get(battery_id, "Unknown")
 
 
-def get_airtag_name(airtag: FindMyAccessory, path: Path) -> str:
+def get_airtag_name(airtag: HasHashedPublicKey | RollingKeyPairSource, path: Path) -> str:
     """Get a human-readable name for an airtag, with fallbacks."""
-    if airtag.name:
-        return airtag.name
-    if airtag.identifier:
-        return airtag.identifier
+    if isinstance(airtag, FindMyAccessory):
+        if airtag.name:
+            return airtag.name
+        if airtag.identifier:
+            return airtag.identifier
     return path.stem  # filename without extension
 
 
 def main(airtag_paths: list[Path], store_path: str) -> int:
     # Step 0: create accessory key generators for all paths
     airtags = [FindMyAccessory.from_json(path) for path in airtag_paths]
-    airtag_to_path = dict(zip(airtags, airtag_paths))
+    airtag_to_path: dict[HasHashedPublicKey | RollingKeyPairSource, Path] = dict(zip(airtags, airtag_paths))
 
     # Step 1: log into an Apple account
     acc = get_account_sync(store_path, ANISETTE_SERVER, ANISETTE_LIBS_PATH)
@@ -62,8 +65,9 @@ def main(airtag_paths: list[Path], store_path: str) -> int:
 
     # step 3: print 'em
     print("Last known locations:")
-    for airtag, location in locations.items():
-        name = get_airtag_name(airtag, airtag_to_path[airtag])
+    for airtag, path in airtag_to_path.items():
+        location = locations.get(airtag)  # type: ignore[union-attr]
+        name = get_airtag_name(airtag, path)
         if location:
             battery = get_battery_level(location.status)
             print(f" - {name}: {location} (Battery: {battery})")