поведенческое тестирование и поиск магазинов

Author: Miskler

chizhik_api/endpoints/geolocation.py

@@ -3,25 +3,50 @@
 from __future__ import annotations
 
 from typing import TYPE_CHECKING
+from dataclasses import dataclass
 
-from human_requests import ApiChild, autotest
+from human_requests import ApiChild, ApiParent, api_child_field, autotest
 from human_requests.abstraction import FetchResponse, HttpMethod
 
 if TYPE_CHECKING:
     from ..manager import ChizhikAPI  # noqa: F401
 
 
+@dataclass(init=False)
 class ClassGeolocation(ApiChild["ChizhikAPI"]):
     """Методы для работы с геолокацией и выбором магазинов.
 
     Включает получение информации о городах, адресах, поиск магазинов
     и управление настройками доставки.
     """
 
+    Shop: ShopService = api_child_field(lambda parent: ShopService(parent.parent))
+    """Сервис для работы с информацией о магазинах."""
+
+
+    def __init__(self, parent: "ChizhikAPI"):
+        super().__init__(parent)
+        ApiParent.__post_init__(self)
+
     @autotest
     async def cities_list(self, search_name: str, page: int = 1) -> FetchResponse:
         """Получить список городов по частичному совпадению имени."""
         return await self._parent._request(
             HttpMethod.GET,
             f"{self._parent.CATALOG_URL}/geo/cities/?name={search_name}&page={page}",
         )
+
+class ShopService(ApiChild["ChizhikAPI"]):
+    """Сервис для работы с информацией о магазинах."""
+
+    @autotest
+    async def all(self) -> FetchResponse:
+        """Получить список всех точек магазинов."""
+        url = f"{self._parent.CATALOG_URL}/shops"
+        return await self._parent._request(HttpMethod.GET, url)
+    
+    @autotest
+    async def search(self, query: str) -> FetchResponse:
+        """Получить список всех точек магазинов."""
+        url = f"{self._parent.CATALOG_URL}/shops?term={query}"
+        return await self._parent._request(HttpMethod.GET, url)

chizhik_api/manager.py

@@ -3,6 +3,7 @@
 from collections import defaultdict
 from dataclasses import dataclass, field
 from typing import Any
+import asyncio
 
 from camoufox.async_api import AsyncCamoufox
 from human_requests import (
@@ -44,6 +45,7 @@ class ChizhikAPI(ApiParent):
     browser_opts: dict[str, Any] = field(default_factory=dict)
     """Дополнительные опции для браузера (см. https://camoufox.com/python/installation/)"""
     CATALOG_URL: str = "https://app.chizhik.club/api/v1"
+    API_URL: str = "https://app.chizhik.club/api"
     """URL для работы с каталогом."""
     MAIN_SITE_URL: str = "https://chizhik.club/catalog/"
     """URL главной страницы сайта."""
@@ -101,9 +103,22 @@ async def _warmup(self) -> None:
             )
             await sniffer.start(self.ctx)
 
+            collected = {}
+            def on_request(request):
+                if request.url.startswith(self.API_URL):
+                    collected[request.url] = request.headers
+
+            self.ctx.on("request", on_request)
+
             await self.page.goto(self.MAIN_SITE_URL, wait_until="networkidle")
             await self.page.wait_for_selector("next-route-announcer", state="attached")
+            await asyncio.sleep(1)
+            await self.page.locator('main a[data-qa^="sidebar-sub-category-"][data-qa$="-link"]').first.click()
+            await self.page.locator('main div[itemtype="https://schema.org/Product"]').first.click()
+            await asyncio.sleep(1)
+            await self.page.wait_for_load_state("load")
 
+            await self.ctx.unroute("**/api/**", on_request)
             result_sniffer = await sniffer.complete()
 
             # Результат: {заголовок: [уникальные значения]}
@@ -116,7 +131,7 @@ async def _warmup(self) -> None:
 
             # Преобразуем set обратно в list
             self.unstandard_headers = {k: list(v)[0] for k, v in result.items()}
-            self.unstandard_urls = result_sniffer["request"]
+            self.unstandard_urls = collected
 
         await self.page.goto(self.CATALOG_URL, wait_until="networkidle")