complite reverse api

Author: Miskler

pyaterochka_api/endpoints/advertising.py

@@ -1,6 +1,6 @@
 """Реклама"""
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Literal
 
 from human_requests.abstraction import FetchResponse, HttpMethod
 
@@ -17,18 +17,18 @@ class ClassAdvertising:
     def __init__(self, parent: "PyaterochkaAPI"):
         self._parent: "PyaterochkaAPI" = parent
 
-    async def get_news(self, limit: int | None = None) -> FetchResponse:
-        """
-        Asynchronously retrieves news from the Pyaterochka API.
-
-        Args:
-            limit (int, optional): The maximum number of news items to retrieve. Defaults to None.
-        
-        Returns:
-            dict: A dictionary representing the news if the request is successful, error otherwise.
-        """
-        url = f"{self._parent.MAIN_SITE_URL}/api/public/v1/news/"
-        if limit and limit > 0:
-            url += f"?limit={limit}"
-
-        return await self._parent._request(method=HttpMethod.GET, url=url)
+    async def news(self,
+                   limit: int = 12,
+                   offset: int = 0):
+        request_url = f"{self._parent.CATALOG_URL}/public/v1/news/?limit={limit}&offset={offset}"
+        return await self._parent._request(method=HttpMethod.GET, url=request_url)
+
+    async def promo_offers(self,
+                       limit: int = 20,
+                       web_version: bool = True,
+                       type_offers: Literal["mainpage_promotion",
+                                            "zooclub_promotion",
+                                            "childrenclub_promotion",
+                                            "barclub_promotion"] = "mainpage_promotion") -> FetchResponse:
+        request_url = f"{self._parent.CATALOG_URL}/public/v1/promo-offers/?limit={limit}&web_version={str(web_version).lower()}&type={type_offers}"
+        return await self._parent._request(method=HttpMethod.GET, url=request_url)

pyaterochka_api/endpoints/geolocation.py

@@ -1,6 +1,7 @@
 """Геолокация"""
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Optional
+import urllib
 
 from human_requests.abstraction import FetchResponse, HttpMethod
 
@@ -32,3 +33,22 @@ async def find_store(self, longitude: float, latitude: float) -> FetchResponse:
 
         request_url = f"{self._parent.CATALOG_URL}/orders/v1/orders/stores/?lon={longitude}&lat={latitude}"
         return await self._parent._request(method=HttpMethod.GET, url=request_url)
+
+    async def suggest(self, query: str) -> FetchResponse:
+        """Начинайте вводить адрес, он предложит точные варианты"""
+        request_url = f"{self._parent.CATALOG_URL}/maps/suggest/?text={urllib.parse.quote(query)}"
+        return await self._parent._request(method=HttpMethod.GET, url=request_url)
+
+    async def geocode(self,
+                      country: str = "Россия",
+                      city: str = "Москва",
+                      street: str = "проспект Мира",
+                      house: Optional[str] = None) -> FetchResponse:
+        """Возвращает геокод (геопозицию) на четкий запрос (используй suggest)"""
+        tup = [country, city, street]
+        if house:
+            tup.append(house)
+        string = urllib.parse.quote(", ".join(tup))
+        
+        request_url = f"{self._parent.MAIN_SITE_URL}/api/maps/geocode/?geocode={string}"
+        return await self._parent._request(method=HttpMethod.GET, url=request_url)

pyaterochka_api/manager.py

@@ -3,6 +3,7 @@
 import os
 from dataclasses import dataclass, field
 from typing import Any
+import json
 
 from camoufox.async_api import AsyncCamoufox
 from human_requests import HumanBrowser, HumanContext, HumanPage
@@ -94,6 +95,10 @@ async def close(self):
         """Закрыть HTTP-сессию и освободить ресурсы."""
         await self.session.close()
 
+    async def delivery_panel_store(self) -> dict:
+        """Текущий адрес доставке (при инициализации проставляется автоматически)"""
+        return json.loads((await self.page.local_storage()).get("DeliveryPanelStore"))
+
     async def _request(
         self,
         method: HttpMethod,
@@ -104,7 +109,6 @@ async def _request(
         """Выполнить HTTP-запрос через внутреннюю сессию.
 
         Единая точка входа для всех HTTP-запросов библиотеки.
-        Добавляет к ответу объект Request для совместимости.
 
         Args:
             method: HTTP метод (GET, POST, PUT, DELETE и т.д.)