Merge main into snapshot-bot/update-snapshots

Author: Miskler

.github/workflows/publish.yml

@@ -11,7 +11,9 @@ on:
         options: [docs, package, all]
 
 permissions:
-  contents: read
+  contents: write
+  issues: write
+  pull-requests: write
   pages: write
   id-token: write   # нужно для PyPI Trusted Publishing (OIDC)
 

Makefile

@@ -7,7 +7,7 @@ install-dev:
 	pip install -e .[dev]
 
 test:
-	pytest --cov=chizhik_api --cov-report=xml --cov-report=html --cov-report=term-missing
+	pytest --cov=fixprice_api --cov-report=xml --cov-report=html --cov-report=term-missing
 
 test-quick:
 	pytest --tb=short
@@ -19,8 +19,8 @@ type-check:
 	python -m mypy fixprice_api
 
 format:
-	black chizhik_api/ tests/
-	isort chizhik_api/ tests/
+	black fixprice_api/ tests/
+	isort fixprice_api/ tests/
 
 clean:
 	rm -rf build/ dist/ *.egg-info/

README.md

@@ -55,6 +55,7 @@ async def main():
             )
         ).json()
         first_product_id = products[0]["id"]
+        first_product_url = products[0]["url"]
         print(f"Первый товар: {products[0]['title']!s:.60s} ({first_product_id})")
 
         # 3. Геолокация (влияет на каталог и баланс)
@@ -66,7 +67,11 @@ async def main():
         balance = (await api.Catalog.Product.balance(product_id=first_product_id)).json()
         print(f"Проверено магазинов: {len(balance)}")
 
-        # 5. Загрузка изображения
+        # 5. Подробное инфо о товаре
+        info = (await api.Catalog.Product.info(url=first_product_url)).json()
+        print(f"Подробно о товаре: {list(info.keys())}")
+
+        # 6. Загрузка изображения
         image_url = products[0]["images"][0]["src"]
         image_stream = await api.General.download_image(image_url)
         with Image.open(image_stream) as img:

docs/source/conf.py

@@ -38,7 +38,7 @@ def _get_version() -> str:
         pass
     # 3) fallback: import from package (may fail on RTD without deps)
     try:
-        from chizhik_api import __version__  # type: ignore
+        from fixprice_api import __version__  # type: ignore
         return __version__
     except Exception:
         return "0.0.0"

docs/source/quick_start.rst

@@ -8,41 +8,53 @@ Quick Start
 
 .. code-block:: python
     
-    from chizhik_api import ChizhikAPI
+    import asyncio
+    from fixprice_api import FixPriceAPI, CatalogSort
+    from PIL import Image
+
 
     async def main():
-        # RUS: Использование проксирования опционально. Вы можете создать несколько агентов с разными прокси для ускорения парса.
-        # ENG: Proxy usage is optional. You can create multiple agents with different proxies for faster parsing.
-        async with ChizhikAPI(proxy="user:password@host:port", headless=False) as API:
-            # RUS: Выводит активные предложения магазина
-            # ENG: Outputs active offers of the store
-            print(f"Active offers output: {(await API.Advertising.active_inout()).json()!s:.100s}...\n")
-            
-            # RUS: Выводит список городов соответствующих поисковому запросу (только на русском языке)
-            # ENG: Outputs a list of cities corresponding to the search query (only in Russian language)
-            city_list = (await API.Geolocation.cities_list(search_name='ар', page=1)).json()
-            print(f"Cities list output: {city_list!s:.100s}...\n")
-            # Счет страниц с единицы / index starts from 1
-
-            # RUS: Выводит список всех категорий на сайте
-            # ENG: Outputs a list of all categories on the site
-            catalog = (await API.Catalog.tree()).json()
-            print(f"Categories list output: {catalog!s:.100s}...\n")
-
-            # RUS: Выводит список всех товаров выбранной категории (ограничение 100 элементов, если превышает - запрашивайте через дополнительные страницы)
-            # ENG: Outputs a list of all items in the selected category (limiting to 100 elements, if exceeds - request through additional pages)
-            items = (await API.Catalog.products_list(category_id=catalog[0]['id'], page=1)).json()
-            print(f"Items list output: {items!s:.100s}...\n")
-            # Счет страниц с единицы / index starts from 1
-
-            # RUS: Сохраняем изображение с сервера (в принципе, сервер отдал бы их и без обертки моего объекта, но лучше максимально претворяться обычным пользователем)
-            # ENG: Saving an image from the server (in fact, the server gave them and without wrapping my object, but better to be as a regular user)
-            image = await API.General.download_image(items['items'][0]['images'][0]['image'])
-            with open(image.name, 'wb') as f:
-                f.write(image.read())
+        async with FixPriceAPI() as api:
+            # 1. Получаем дерево категорий
+            tree_data = (await api.Catalog.tree()).json()
+            first_alias = tree_data[next(iter(tree_data))]["alias"]
+            print(f"Первая категория: {first_alias}")
 
-    import asyncio
-    asyncio.run(main())
+            # 2. Список товаров в категории
+            products = (
+                await api.Catalog.products_list(
+                    category_alias=first_alias,
+                    page=1,
+                    limit=24,
+                    sort=CatalogSort.POPULARITY,
+                )
+            ).json()
+            first_product_id = products[0]["id"]
+            first_product_url = products[0]["url"]
+            print(f"Первый товар: {products[0]['title']!s:.60s} ({first_product_id})")
+
+            # 3. Геолокация (влияет на каталог и баланс)
+            cities = (await api.Geolocation.cities_list(country_id=2)).json()  # Россия
+            api.city_id = cities[0]["id"]
+            print(f"Текущий city_id: {api.city_id}")
+
+            # 4. Проверка наличия товара по магазинам
+            balance = (await api.Catalog.Product.balance(product_id=first_product_id)).json()
+            print(f"Проверено магазинов: {len(balance)}")
+
+            # 5. Подробное инфо о товаре
+            info = (await api.Catalog.Product.info(url=first_product_url)).json()
+            print(f"Подробно о товаре: {list(info.keys())}")
+
+            # 6. Загрузка изображения
+            image_url = products[0]["images"][0]["src"]
+            image_stream = await api.General.download_image(image_url)
+            with Image.open(image_stream) as img:
+                print(f"Image format: {img.format}, size: {img.size}")
+
+
+    if __name__ == "__main__":
+        asyncio.run(main())
 
 .. code-block:: console
 

fixprice_api/__init__.py

@@ -2,4 +2,4 @@
 from .abstraction import CatalogSort
 
 __all__ = ["FixPriceAPI", "CatalogSort"]
-__version__ = "0.2.2"
+__version__ = "0.2.4"

fixprice_api/abstraction.py

@@ -1,4 +1,3 @@
-
 class CatalogSort:
     POPULARITY = "sold"
     """Сначало самые популярные"""

fixprice_api/endpoints/catalog.py

@@ -2,8 +2,12 @@
 
 from __future__ import annotations
 
+import json
+from types import MethodType
+
+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,84 @@ 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,
+    ) -> PWResponse:
+        """
+        Информация СПАРСИВАЕТСЯ (в отличии от других методов).
+        Инфо о товаре со страницы типа
+        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()
+        try:
+            resp = await page.goto(real_url, wait_until="domcontentloaded")
+            if resp is None:
+                raise RuntimeError("page.goto() returned None")
+
+            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
+            )
+
+            def _json(self):
+                return nuxt_data
+
+            resp.json = MethodType(_json, resp)
+
+            return resp
+        finally:
+            await page.close()

fixprice_api/endpoints/general.py

@@ -26,7 +26,11 @@ async def download_image(
             attempts=retry_attempts, start_timeout=3.0, max_timeout=timeout
         )
 
-        px = self._parent.proxy if isinstance(self._parent.proxy, Proxy) else Proxy(self._parent.proxy)
+        px = (
+            self._parent.proxy
+            if isinstance(self._parent.proxy, Proxy)
+            else Proxy(self._parent.proxy)
+        )
         async with RetryClient(retry_options=retry_options) as retry_client:
             async with retry_client.get(
                 url, raise_for_status=True, proxy=px.as_str()

fixprice_api/endpoints/geolocation.py

@@ -44,7 +44,7 @@ async def countries_list(self, alias: str = None) -> FetchResponse:
     async def regions_list(self, country_id: int = None) -> FetchResponse:
         """Возвращает список всех регионов, их id и название. Если фильтр не применен - выдача всех регионов независимо от страны."""
         url = f"{self._parent.CATALOG_URL}/v1/location/region"
-        if not country_id:
+        if country_id:
             url += f"?countryId={country_id}"
 
         return await self._parent._request(HttpMethod.GET, url=url)