mirror of
https://github.com/malmeloo/FindMy.py.git
synced 2026-04-18 16:54:01 +02:00
105 lines
2.9 KiB
Python
105 lines
2.9 KiB
Python
"""Module for Anisette header providers."""
|
|
from __future__ import annotations
|
|
|
|
import base64
|
|
import locale
|
|
import logging
|
|
from abc import ABC, abstractmethod
|
|
from datetime import datetime, timezone
|
|
|
|
from typing_extensions import override
|
|
|
|
from findmy.util.closable import Closable
|
|
from findmy.util.http import HttpSession
|
|
|
|
|
|
def _gen_meta_headers(
|
|
user_id: str,
|
|
device_id: str,
|
|
serial: str = "0",
|
|
) -> dict[str, str]:
|
|
now = datetime.now(tz=timezone.utc)
|
|
locale_str = locale.getdefaultlocale()[0] or "en_US"
|
|
|
|
return {
|
|
"X-Apple-I-Client-Time": now.replace(microsecond=0).isoformat() + "Z",
|
|
"X-Apple-I-TimeZone": str(now.astimezone().tzinfo),
|
|
"loc": locale_str,
|
|
"X-Apple-Locale": locale_str,
|
|
"X-Apple-I-MD-RINFO": "17106176",
|
|
"X-Apple-I-MD-LU": base64.b64encode(str(user_id).upper().encode()).decode(),
|
|
"X-Mme-Device-Id": str(device_id).upper(),
|
|
"X-Apple-I-SRL-NO": serial,
|
|
}
|
|
|
|
|
|
class BaseAnisetteProvider(Closable, ABC):
|
|
"""Abstract base class for Anisette providers."""
|
|
|
|
@abstractmethod
|
|
async def _get_base_headers(self) -> dict[str, str]:
|
|
raise NotImplementedError
|
|
|
|
async def get_headers(
|
|
self,
|
|
user_id: str,
|
|
device_id: str,
|
|
serial: str = "0",
|
|
) -> dict[str, str]:
|
|
"""
|
|
Retrieve a complete dictionary of Anisette headers.
|
|
|
|
Consider using `BaseAppleAccount.get_anisette_headers` instead.
|
|
"""
|
|
base_headers = await self._get_base_headers()
|
|
base_headers.update(_gen_meta_headers(user_id, device_id, serial))
|
|
|
|
return base_headers
|
|
|
|
|
|
class RemoteAnisetteProvider(BaseAnisetteProvider):
|
|
"""Anisette provider. Fetches headers from a remote Anisette server."""
|
|
|
|
def __init__(self, server_url: str) -> None:
|
|
"""Initialize the provider with URL to te remote server."""
|
|
super().__init__()
|
|
|
|
self._server_url = server_url
|
|
|
|
self._http = HttpSession()
|
|
|
|
logging.info("Using remote anisette server: %s", self._server_url)
|
|
|
|
@override
|
|
async def _get_base_headers(self) -> dict[str, str]:
|
|
r = await self._http.get(self._server_url)
|
|
headers = r.json()
|
|
|
|
return {
|
|
"X-Apple-I-MD": headers["X-Apple-I-MD"],
|
|
"X-Apple-I-MD-M": headers["X-Apple-I-MD-M"],
|
|
}
|
|
|
|
@override
|
|
async def close(self) -> None:
|
|
"""See `AnisetteProvider.close`."""
|
|
await self._http.close()
|
|
|
|
|
|
# TODO(malmeloo): implement using pyprovision
|
|
# https://github.com/malmeloo/FindMy.py/issues/2
|
|
class LocalAnisetteProvider(BaseAnisetteProvider):
|
|
"""Anisette provider. Generates headers without a remote server using pyprovision."""
|
|
|
|
def __init__(self) -> None:
|
|
"""Initialize the provider."""
|
|
super().__init__()
|
|
|
|
@override
|
|
async def _get_base_headers(self) -> dict[str, str]:
|
|
return NotImplemented
|
|
|
|
@override
|
|
async def close(self) -> None:
|
|
"""See `AnisetteProvider.close`."""
|