Update manager.py

Author: Miskler

pyaterochka_api/manager.py

@@ -8,6 +8,7 @@
 from camoufox.async_api import AsyncCamoufox
 from human_requests import HumanBrowser, HumanContext, HumanPage
 from human_requests.abstraction import FetchResponse, HttpMethod, Proxy
+from human_requests.network_analyzer.anomaly_sniffer import HeaderAnomalySniffer, WaitSource, WaitHeader
 
 from .endpoints.advertising import ClassAdvertising
 from .endpoints.catalog import ClassCatalog
@@ -48,6 +49,9 @@ class PyaterochkaAPI:
     """Внутренний контекст сессии браузера"""
     page: HumanPage = field(init=False, repr=False)
     """Внутренний страница сессии браузера"""
+    
+    unstandard_headers: dict[str, str] = {}
+    """Список нестандартных заголовков пойманных при инициализации"""
 
     Geolocation: ClassGeolocation = field(init=False)
     """API для работы с геолокацией."""
@@ -83,10 +87,38 @@ async def _warmup(self) -> None:
         self.ctx = await self.session.new_context()
         self.page = await self.ctx.new_page()
 
+        sniffer = HeaderAnomalySniffer(
+            # доп. вайтлист, если нужно
+            extra_request_allow=["x-forwarded-for", "x-real-ip"],
+            extra_response_allow=[],
+            # нормализуем URL: без фрагмента, но с query
+            #url_normalizer=lambda u: u.split("#", 1)[0],
+            include_subresources=True,   # или False, если интересны только документы
+            url_filter=lambda u: u.startswith("https://5d.5ka.ru/")
+        )
+        await sniffer.start(self.ctx)
+
         await self.page.goto("https://5ka.ru", wait_until="load")
         await self.page.wait_for_selector(selector="next-route-announcer", state="attached")
         await self.page.wait_for_load_state('networkidle')
 
+        await sniffer.wait(
+            tasks=[
+                WaitHeader(
+                    source=WaitSource.REQUEST,
+                    headers=[
+                        "x-app-version",
+                        "x-device-id",
+                        "x-platform"
+                    ]
+                )
+            ],
+            timeout_ms=self.timeout_ms
+        )
+
+        result = await sniffer.complete()
+        self.unstandard_headers = result.get("https://5d.5ka.ru")
+
     async def __aexit__(self, *exc):
         """Выход из контекстного менеджера с закрытием сессии."""
         await self.close()
@@ -99,12 +131,18 @@ async def delivery_panel_store(self) -> dict:
         """Текущий адрес доставке (при инициализации проставляется автоматически)"""
         return json.loads((await self.page.local_storage()).get("DeliveryPanelStore"))
 
+    async def device_id(self) -> str:
+        """Анонимный (так как в библиотеке нет возможности авторизации) индефекатор пользователя,
+        который отправляется на сервер почти с каждым запросом (изменить нельзя)."""
+        return str((await self.page.local_storage()).get("deviceId"))
+
     async def _request(
         self,
         method: HttpMethod,
         url: str,
         *,
         json_body: Any | None = None,
+        add_unstandard_headers: bool = True
     ) -> FetchResponse:
         """Выполнить HTTP-запрос через внутреннюю сессию.
 
@@ -116,6 +154,7 @@ async def _request(
             json_body: Тело запроса в формате JSON (опционально)
         """
         # Единая точка входа в чужую библиотеку для удобства
+        # TODO: пройтись по библиотеке и проверить где add_unstandard_headers должен быть false
         print(url)
         resp: FetchResponse = await self.page.fetch(
             url=url,
@@ -124,7 +163,7 @@ async def _request(
             mode="cors",
             timeout_ms=self.timeout_ms,
             referrer=self.MAIN_SITE_URL,
-            headers={"Accept": "application/json, text/plain, */*"},
+            headers={"Accept": "application/json, text/plain, */*"}.update(self.unstandard_headers if add_unstandard_headers else {}),
         )
 
         return resp