From 08b0096b7cdc03b7b363dd426c608b97afff65bd Mon Sep 17 00:00:00 2001 From: KKlochko Date: Tue, 20 Jun 2023 19:31:00 +0300 Subject: [PATCH] Add the method to show the items by categories. --- src/api/v2/category_api.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 src/api/v2/category_api.py diff --git a/src/api/v2/category_api.py b/src/api/v2/category_api.py new file mode 100644 index 0000000..58cef62 --- /dev/null +++ b/src/api/v2/category_api.py @@ -0,0 +1,34 @@ +import aiohttp +import asyncio +import ujson + +class CategoryAPI: + def __init__(self, fetcher, formatter): + """ + Sets needed classes: + + Fetcher (ApiFetcher) - class that will fetch data. + Format (AbstractFormat) - class that will format response data. + """ + self.__fetcher, self.__formatter = fetcher, formatter + + async def get_dict(self, username: str): + endpoint = f"/api/v2/categories?matrixUsername={username}" + + status, json_data = await self.__fetcher.fetch_json(endpoint) + + if status == 200: + return {"ok": json_data} + + return {"error": "Сталася помилка, спробуйте пізніше."} + + async def get_objects_message(self, username: str) -> str: + response = await self.get_dict(username) + + match response: + case {"ok": json_data}: + return self.__formatter.format_all(json_data) + + case {"error": error_message}: + return error_message +