mirror of
https://github.com/malmeloo/FindMy.py.git
synced 2026-04-17 21:53:57 +02:00
110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
from findmy import (
|
|
AppleAccount,
|
|
AsyncAppleAccount,
|
|
LocalAnisetteProvider,
|
|
LoginState,
|
|
RemoteAnisetteProvider,
|
|
SmsSecondFactorMethod,
|
|
TrustedDeviceSecondFactorMethod,
|
|
)
|
|
|
|
|
|
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(
|
|
store_path: str,
|
|
anisette_url: str | None,
|
|
libs_path: str | None,
|
|
) -> AppleAccount:
|
|
"""Tries to restore a saved Apple account, or prompts the user for login otherwise. (sync)"""
|
|
try:
|
|
acc = AppleAccount.from_json(store_path, anisette_libs_path=libs_path)
|
|
except FileNotFoundError:
|
|
ani = (
|
|
LocalAnisetteProvider(libs_path=libs_path)
|
|
if anisette_url is None
|
|
else RemoteAnisetteProvider(anisette_url)
|
|
)
|
|
acc = AppleAccount(ani)
|
|
_login_sync(acc)
|
|
|
|
acc.to_json(store_path)
|
|
|
|
return acc
|
|
|
|
|
|
async def get_account_async(
|
|
store_path: str,
|
|
anisette_url: str | None,
|
|
libs_path: str | None,
|
|
) -> AsyncAppleAccount:
|
|
"""Tries to restore a saved Apple account, or prompts the user for login otherwise. (async)"""
|
|
try:
|
|
acc = AsyncAppleAccount.from_json(store_path, anisette_libs_path=libs_path)
|
|
except FileNotFoundError:
|
|
ani = (
|
|
LocalAnisetteProvider(libs_path=libs_path)
|
|
if anisette_url is None
|
|
else RemoteAnisetteProvider(anisette_url)
|
|
)
|
|
acc = AsyncAppleAccount(ani)
|
|
await _login_async(acc)
|
|
|
|
acc.to_json(store_path)
|
|
|
|
return acc
|