Prototype ordering

up42/tasking.py

@@ -2,6 +2,7 @@
 Tasking functionality
 """
 
+import dataclasses
 import datetime
 from typing import Any, Dict, List, Literal, Optional, Union
 
@@ -198,3 +199,85 @@ def choose_feasibility(self, feasibility_id: str, accepted_option_id: str) -> di
         """
         url = host.endpoint(f"/v2/tasking/feasibility/{feasibility_id}")
         return self.session.patch(url=url, json={"acceptedOptionId": accepted_option_id}).json()
+
+
+@dataclasses.dataclass
+class FeasibilityOption:
+    id: str
+    feasibility_id: str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    description: str
+
+
+class FeasibilityStudySorting:
+    created_at = utils.SortingField("created_at")
+
+
+@dataclasses.dataclass
+class FeasibilityStudy:
+    id: str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    decided_at: datetime.datetime
+    account_id: str
+    workspace_id: str
+    order_id: str  # Do we need to link order here?
+    decision_status: FeasibilityStatus
+    options: list[FeasibilityOption]
+    decision_type: Optional[Literal["USER", "ADMIN", "API", "SYSTEM"]]
+    type: Literal["AUTO", "MANUAL"]
+    decision_option: Optional[FeasibilityOption]
+
+    def save(self):
+        # save if decision_option is set
+        ...
+
+    @classmethod
+    def all(
+        cls,
+        id: Optional[str] = None,
+        workspace_id: Optional[str] = None,
+        order_id: Optional[str] = None,
+        decision_status: Optional[List[FeasibilityStatus]] = None,
+        sort_by: utils.SortingField = FeasibilityStudySorting.created_at.desc,
+    ):
+        pass
+
+
+class QuotationSorting:
+    created_at = utils.SortingField("createdAt")
+
+
+@dataclasses.dataclass
+class Quotation:
+    id: str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    decided_at: datetime.datetime
+    decision_type: Optional[Literal["USER", "ADMIN", "API", "SYSTEM"]]
+    account_id: str
+    workspace_id: str
+    order_id: str  # Do we need to link order here?
+    credits_price: int
+    decision_status: QuotationStatus
+
+    def accept(self):
+        self.decision_status = "ACCEPTED"
+
+    def reject(self):
+        self.decision_status = "REJECTED"
+
+    def save(self):
+        ...
+
+    @classmethod
+    def all(
+        cls,
+        id: Optional[str] = None,
+        workspace_id: Optional[str] = None,
+        order_id: Optional[str] = None,
+        decision_status: Optional[List[QuotationStatus]] = None,
+        sort_by: utils.SortingField = QuotationSorting.created_at.desc,
+    ):
+        ...