Add method get_positions

alteralt · Sep 5, 2023 · cb8d8cb · cb8d8cb
1 parent 59bf614
commit cb8d8cb
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/tippisell_api/client.py b/tippisell_api/client.py
@@ -49,15 +49,19 @@ async def get_categories(
         result = await self._request(
             path="/v2/category/all",
             http_method="get",
-            params={
-                key: value
-                for key, value in {
-                    "offset": offset,
-                    "limit": limit,
-                    "shop_id": self.shop_id,
-                }.items()
-                if value is not None
-            },
+            params=self._clear_dict(
+                {"offset": offset, "limit": limit, "shop_id": self.shop_id}
+            ),
+        )
+        return result
+
+    async def get_positions(self, product_id: typing.Union[str, int]):
+        result = await self._request(
+            path="/v2/position/all",
+            http_method="get",
+            params=self._clear_dict(
+                {"shop_id": self.shop_id, "product_id": product_id}
+            ),
         )
         return result
 
@@ -152,3 +156,7 @@ def _check_response(cls, http_response: models.HttpResponse):
 
         if http_response.result["ok"] is False:
             raise exceptions.BaseTippisellException(http_response.result["message"])
+
+    @classmethod
+    def _clear_dict(cls, data: dict) -> dict:
+        return {key: value for key, value in data.items() if value is not None}