starred/FindMy.py
1
0
Fork 0
mirror of https://github.com/malmeloo/FindMy.py.git synced 2026-04-17 21:53:57 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: adjust the ser/deser logic of AppleAccount

Browse Source
This commit is contained in:
Nick Crews
2025-05-25 17:51:24 -06:00
parent 7f6ee0ec51
commit 5142350241
2 changed files with 35 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
examples/_login.py
View File

@@ -1,8 +1,5 @@
# ruff: noqa: ASYNC230
import json
from pathlib import Path
from findmy.reports import (
AppleAccount,
AsyncAppleAccount,
@@ -71,33 +68,25 @@ async def _login_async(account: AsyncAppleAccount) -> None:
def get_account_sync(anisette: BaseAnisetteProvider) -> AppleAccount:
"""Tries to restore a saved Apple account, or prompts the user for login otherwise. (sync)"""
acc = AppleAccount(anisette)
# Save / restore account logic
acc_store = Path("account.json")
acc = AppleAccount(anisette=anisette)
acc_store = "account.json"
try:
with acc_store.open() as f:
acc.restore(json.load(f))
acc.from_json(acc_store)
except FileNotFoundError:
_login_sync(acc)
with acc_store.open("w+") as f:
json.dump(acc.export(), f)
acc.to_json(acc_store)
return acc
async def get_account_async(anisette: BaseAnisetteProvider) -> AsyncAppleAccount:
"""Tries to restore a saved Apple account, or prompts the user for login otherwise. (async)"""
acc = AsyncAppleAccount(anisette)
# Save / restore account logic
acc_store = Path("account.json")
acc = AsyncAppleAccount(anisette=anisette)
acc_store = "account.json"
try:
with acc_store.open() as f:
acc.restore(json.load(f))
acc.from_json(acc_store)
except FileNotFoundError:
await _login_async(acc)
with acc_store.open("w+") as f:
json.dump(acc.export(), f)
acc.to_json(acc_store)
return acc