From b22166d031e9e727b22d2e0c8fc652c453913986 Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Tue, 20 Jun 2023 17:39:11 +0300
Subject: [PATCH] Add company methods to get the list or to select a company.

---
 src/api/v2/company_api.py | 57 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 src/api/v2/company_api.py

diff --git a/src/api/v2/company_api.py b/src/api/v2/company_api.py
new file mode 100644
index 0000000..27b4d16
--- /dev/null
+++ b/src/api/v2/company_api.py
@@ -0,0 +1,57 @@
+import aiohttp
+import asyncio
+import ujson
+
+class CompanyAPI:
+    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/companies?matrixUsername={username}"
+
+        status, json_data = await self.__fetcher.fetch_json(endpoint)
+
+        if status == 200:
+            return {"ok": json_data}
+
+        return {"error": "Сталася помилка, спробуйте пізніше."}
+
+    async def select_obj(self, company_name: str, username: str):
+        endpoint = "/api/v2/select-company"
+
+        status, json_data = await self.__fetcher.send_json(endpoint, {
+            'companyName': company_name,
+            'matrixUsername': username,
+        })
+
+        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
+
+    async def select_object_message(self, company_name: str, username: str) -> str:
+        response = await self.select_obj(company_name, username)
+
+        match response:
+            case {"ok": json_data}:
+                return 'Компанія обрана'
+
+            case {"error": error_message}:
+                return error_message
+