Merge pull request #219 from lanrat/airtag_example

update airtag example

Author: malmeloo

examples/airtag.py

@@ -8,14 +8,19 @@
 import logging
 import sys
 from pathlib import Path
+from typing import TYPE_CHECKING
 
 from _login import get_account_sync
 
 from findmy import FindMyAccessory
 
-# Path where login session will be stored.
+if TYPE_CHECKING:
+    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.
-STORE_PATH = "account.json"
+DEFAULT_STORE_PATH = "account.json"
 
 # URL to LOCAL anisette server. Set to None to use built-in Anisette generator instead (recommended)
 # IF YOU USE A PUBLIC SERVER, DO NOT COMPLAIN THAT YOU KEEP RUNNING INTO AUTHENTICATION ERRORS!
@@ -30,32 +35,67 @@
 
 logging.basicConfig(level=logging.INFO)
 
+BATTERY_LEVEL = {0b00: "Full", 0b01: "Medium", 0b10: "Low", 0b11: "Very Low"}
+
+
+def get_battery_level(status: int) -> str:
+    """Extract battery level from status byte."""
+    battery_id = (status >> 6) & 0b11
+    return BATTERY_LEVEL.get(battery_id, "Unknown")
+
+
+def get_airtag_name(airtag: HasHashedPublicKey | RollingKeyPairSource, path: Path) -> str:
+    """Get a human-readable name for an airtag, with fallbacks."""
+    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_path: Path) -> int:
-    # Step 0: create an accessory key generator
-    airtag = FindMyAccessory.from_json(airtag_path)
+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[HasHashedPublicKey | RollingKeyPairSource, Path] = dict(
+        zip(airtags, airtag_paths, strict=False)
+    )
 
     # Step 1: log into an Apple account
-    acc = get_account_sync(STORE_PATH, ANISETTE_SERVER, ANISETTE_LIBS_PATH)
+    acc = get_account_sync(store_path, ANISETTE_SERVER, ANISETTE_LIBS_PATH)
     print(f"Logged in as: {acc.account_name} ({acc.first_name} {acc.last_name})")
 
     # step 2: fetch reports!
-    location = acc.fetch_location(airtag)
+    locations = acc.fetch_location(airtags)
 
     # step 3: print 'em
-    print("Last known location:")
-    print(f" - {location}")
+    print("Last known locations:")
+    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})")
+        else:
+            print(f" - {name}: No location found")
 
     # step 4: save current account state to disk
-    acc.to_json(STORE_PATH)
-    airtag.to_json(airtag_path)
+    acc.to_json(store_path)
+    for airtag, path in zip(airtags, airtag_paths, strict=False):
+        airtag.to_json(path)
 
     return 0
 
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("airtag_path", type=Path)
+    parser.add_argument("airtag_paths", type=Path, nargs="+")
+    parser.add_argument(
+        "--store-path",
+        type=str,
+        default=DEFAULT_STORE_PATH,
+        help=f"Path to account session file (default: {DEFAULT_STORE_PATH})",
+    )
     args = parser.parse_args()
 
-    sys.exit(main(args.airtag_path))
+    sys.exit(main(args.airtag_paths, args.store_path))