mirror of
https://github.com/malmeloo/FindMy.py.git
synced 2026-04-18 02:54:01 +02:00
102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
from findmy.reports import (
|
|
AppleAccount,
|
|
AsyncAppleAccount,
|
|
BaseAnisetteProvider,
|
|
LoginState,
|
|
SmsSecondFactorMethod,
|
|
TrustedDeviceSecondFactorMethod,
|
|
)
|
|
|
|
ACCOUNT_STORE = "account.json"
|
|
|
|
|
|
def _login_sync(account: AppleAccount) -> None:
|
|
email = input("email? > ")
|
|
password = input("passwd? > ")
|
|
|
|
state = account.login(email, password)
|
|
|
|
if state == LoginState.REQUIRE_2FA: # Account requires 2FA
|
|
# This only supports SMS methods for now
|
|
methods = account.get_2fa_methods()
|
|
|
|
# Print the (masked) phone numbers
|
|
for i, method in enumerate(methods):
|
|
if isinstance(method, TrustedDeviceSecondFactorMethod):
|
|
print(f"{i} - Trusted Device")
|
|
elif isinstance(method, SmsSecondFactorMethod):
|
|
print(f"{i} - SMS ({method.phone_number})")
|
|
|
|
ind = int(input("Method? > "))
|
|
|
|
method = methods[ind]
|
|
method.request()
|
|
code = input("Code? > ")
|
|
|
|
# This automatically finishes the post-2FA login flow
|
|
method.submit(code)
|
|
|
|
|
|
async def _login_async(account: AsyncAppleAccount) -> None:
|
|
email = input("email? > ")
|
|
password = input("passwd? > ")
|
|
|
|
state = await account.login(email, password)
|
|
|
|
if state == LoginState.REQUIRE_2FA: # Account requires 2FA
|
|
# This only supports SMS methods for now
|
|
methods = await account.get_2fa_methods()
|
|
|
|
# Print the (masked) phone numbers
|
|
for i, method in enumerate(methods):
|
|
if isinstance(method, TrustedDeviceSecondFactorMethod):
|
|
print(f"{i} - Trusted Device")
|
|
elif isinstance(method, SmsSecondFactorMethod):
|
|
print(f"{i} - SMS ({method.phone_number})")
|
|
|
|
ind = int(input("Method? > "))
|
|
|
|
method = methods[ind]
|
|
await method.request()
|
|
code = input("Code? > ")
|
|
|
|
# This automatically finishes the post-2FA login flow
|
|
await method.submit(code)
|
|
|
|
|
|
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")
|
|
try:
|
|
with acc_store.open() as f:
|
|
acc.restore(json.load(f))
|
|
except FileNotFoundError:
|
|
_login_sync(acc)
|
|
with acc_store.open("w+") as f:
|
|
json.dump(acc.export(), f)
|
|
|
|
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")
|
|
try:
|
|
with acc_store.open() as f:
|
|
acc.restore(json.load(f))
|
|
except FileNotFoundError:
|
|
await _login_async(acc)
|
|
with acc_store.open("w+") as f:
|
|
json.dump(acc.export(), f)
|
|
|
|
return acc
|