feat: add components support

danfimov · Sep 28, 2023 · d6173ac · d6173ac
1 parent da4fd88
commit d6173ac
Show file tree

Hide file tree

Showing 5 changed files with 36 additions and 3 deletions.
diff --git a/ya_tracker_client/domain/entities/checklist.py b/ya_tracker_client/domain/entities/checklist.py
@@ -1,4 +1,3 @@
-
 from ya_tracker_client.domain.entities.base import AbstractEntity
 from ya_tracker_client.domain.entities.deadline import Deadline
 from ya_tracker_client.domain.entities.user import UserShort

diff --git a/ya_tracker_client/domain/entities/component.py b/ya_tracker_client/domain/entities/component.py
@@ -0,0 +1,14 @@
+from ya_tracker_client.domain.entities.base import AbstractEntity
+from ya_tracker_client.domain.entities.queue import QueueShort
+from ya_tracker_client.domain.entities.user import UserShort
+
+
+class Component(AbstractEntity):
+    url: str
+    id: int
+    version: int
+    name: str
+    queue: QueueShort
+    description: str
+    lead: UserShort | None = None
+    assign_auto: bool
diff --git a/ya_tracker_client/domain/repositories/__init__.py b/ya_tracker_client/domain/repositories/__init__.py
@@ -1,4 +1,5 @@
 from ya_tracker_client.domain.repositories.checklist import ChecklistRepository
+from ya_tracker_client.domain.repositories.component import ComponentRepository
 from ya_tracker_client.domain.repositories.issue import IssueRepository
 from ya_tracker_client.domain.repositories.issue_relationship import IssueRelationshipRepository
 from ya_tracker_client.domain.repositories.queue import QueueRepository
@@ -7,8 +8,9 @@
 
 __all__ = [
     "ChecklistRepository",
-    "IssueRepository",
+    "ComponentRepository",
     "IssueRelationshipRepository",
+    "IssueRepository",
     "QueueRepository",
     "UserRepository",
 ]
diff --git a/ya_tracker_client/domain/repositories/component.py b/ya_tracker_client/domain/repositories/component.py
@@ -0,0 +1,16 @@
+from json import loads
+
+from ya_tracker_client.domain.entities.component import Component
+from ya_tracker_client.domain.repositories.base import EntityRepository
+
+
+class ComponentRepository(EntityRepository):
+    async def get_components(self) -> list[Component]:
+        """
+        YT docs: https://cloud.yandex.com/en/docs/tracker/get-components
+        """
+        raw_response = await self._client.request(
+            method="GET",
+            uri="/components",
+        )
+        return [Component(**raw_component) for raw_component in loads(raw_response)]
diff --git a/ya_tracker_client/service/api.py b/ya_tracker_client/service/api.py
@@ -1,5 +1,6 @@
 from ya_tracker_client.domain.repositories import (
     ChecklistRepository,
+    ComponentRepository,
     IssueRelationshipRepository,
     IssueRepository,
     QueueRepository,
@@ -9,11 +10,12 @@
 
 
 class YaTrackerClient(
+    ChecklistRepository,
+    ComponentRepository,
     IssueRelationshipRepository,
     IssueRepository,
     QueueRepository,
     UserRepository,
-    ChecklistRepository,
 ):
     def __init__(
         self,