add Product.info

Author: Miskler

fixprice_api/endpoints/catalog.py

@@ -2,8 +2,12 @@
 
 from __future__ import annotations
 
+import asyncio
+import json
+
+from playwright.async_api import Response as PWResponse
 from dataclasses import dataclass
-from typing import Optional, TYPE_CHECKING
+from typing import Optional, TYPE_CHECKING, overload
 from .. import abstraction
 from human_requests import autotest
 from human_requests.abstraction import FetchResponse, HttpMethod
@@ -101,3 +105,69 @@ async def balance(
             url += "&inStock=true"
 
         return await self._parent._request(HttpMethod.GET, url)
+    
+    @overload
+    async def info(self, *, url: str): ...
+
+    @overload
+    async def info(self, *, category: str, product_id: int, slug: str): ...
+
+    @autotest
+    async def info(
+        self,
+        *,
+        url: str | None = None,
+        category: str | None = None,
+        product_id: int | None = None,
+        slug: str | None = None,
+    ) -> dict:
+        """
+        Информация СПАРСИВАЕТСЯ (в отличии от других методов).
+        Инфо о товаре со страницы типа
+        https://fix-price.com/catalog/produkty-i-napitki/p-1902248-shokoladnye-konfety-inis-nickers-135-g
+        
+        Либо предоставляете url напрямую, например `products[0]["url"]`
+
+        Данные карточки лежат в obj["data"][0]["categoryData"]["product"]
+        """
+
+        real_url = "https://fix-price.com/catalog/"
+        if url is None:
+            if category is None or product_id is None or slug is None:
+                raise TypeError("Either url or (category, product_id, slug) must be provided")
+
+            real_url += f"{category}/p-{product_id}-{slug}"
+        else:
+            real_url += url
+
+        page = await self._parent.ctx.new_page()
+        await page.goto(real_url, wait_until="domcontentloaded")
+
+        raw_json = await page.evaluate("""
+        () => {
+            const marker = "window.__NUXT__=";
+
+            for (const s of document.scripts) {
+                const txt = s.textContent || "";
+                const idx = txt.indexOf(marker);
+
+                if (idx !== -1) {
+                    let expr = txt.slice(idx + marker.length).trim();
+
+                    if (expr.endsWith(";")) {
+                        expr = expr.slice(0, -1);
+                    }
+
+                    const obj = Function('"use strict"; return (' + expr + ')')();
+                    return JSON.stringify(obj);
+                }
+            }
+
+            return null;
+        }
+        """)
+        nuxt_data = json.loads(raw_json)["data"][0]["categoryData"]["product"] if raw_json else None
+
+        await page.close()
+        
+        return nuxt_data

tests/api_test.py

@@ -77,11 +77,11 @@ def _capture_product_id(
     if not isinstance(data, list) or not data:
         pytest.fail("Catalog.products_list returned empty data.")
 
-    product_id = data[0].get("id")
-    if not isinstance(product_id, int):
-        pytest.fail("Catalog.products_list did not return a valid product id.")
-
-    ctx.state["autotest_product_id"] = product_id
+    for key in ["id", "url"]:
+        value = data[0].get(key)
+        if not isinstance(value[key], int):
+            pytest.fail(f"Catalog.products_list did not return a valid {key}.")
+        ctx.state[f"autotest_product_{key}"] = value[key]
 
 
 @autotest_depends_on(ClassCatalog.products_list)
@@ -93,6 +93,15 @@ def _product_balance_params(ctx: AutotestCallContext) -> dict[str, int]:
     pytest.fail("ProductService.balance depends on Catalog.products_list.")
 
 
+@autotest_depends_on(ClassCatalog.products_list)
+@autotest_params(target=ProductService.info)
+def _product_info_params(ctx: AutotestCallContext) -> dict[str, int]:
+    cached_id = ctx.state.get("autotest_product_url")
+    if isinstance(cached_id, int):
+        return {"url": cached_id}
+    pytest.fail("ProductService.info depends on Catalog.products_list.")
+
+
 @autotest_data(name="unstandard_headers")
 def _unstandard_headers_data(ctx: AutotestDataContext) -> dict[str, Any]:
     return ctx.api.unstandard_headers