taskoteka: add StrEnum types for method parameters

altrepo/taskoteka/methods.py

@@ -4,6 +4,7 @@ from typing import Literal, overload
 
 from ..api.methods import BaseAPI
 from . import models
+from . import types
 from .events import EventStream
 
 
@@ -32,10 +33,10 @@ class ALTRepoTaskoteka:
     @overload
     async def tasks(
         self, *,
-        state: str | None = ...,
-        repo: str | None = ...,
+        state: types.TaskState | str | None = ...,
+        repo: types.Repo | str | None = ...,
         user: str | None = ...,
-        needs_approval: Literal["maint", "tester"] | None = ...,
+        needs_approval: types.ApprovalGroup | None = ...,
         q: str | None = ...,
         brief: Literal[True],
     ) -> list[int]: ...
@@ -43,10 +44,10 @@ class ALTRepoTaskoteka:
     @overload
     async def tasks(
         self, *,
-        state: str | None = ...,
-        repo: str | None = ...,
+        state: types.TaskState | str | None = ...,
+        repo: types.Repo | str | None = ...,
         user: str | None = ...,
-        needs_approval: Literal["maint", "tester"] | None = ...,
+        needs_approval: types.ApprovalGroup | None = ...,
         q: str | None = ...,
         brief: Literal[False] = ...,
     ) -> list[models.TaskSummaryModel]: ...
@@ -54,10 +55,10 @@ class ALTRepoTaskoteka:
     async def tasks(
         self,
         task_id: int | None = None,
-        state: str | None = None,
-        repo: str | None = None,
+        state: types.TaskState | str | None = None,
+        repo: types.Repo | str | None = None,
         user: str | None = None,
-        needs_approval: Literal["maint", "tester"] | None = None,
+        needs_approval: types.ApprovalGroup | None = None,
         q: str | None = None,
         brief: bool = False,
     ) -> list[models.TaskSummaryModel] | list[int] | models.TaskModel | int:

altrepo/taskoteka/types.py

+from enum import StrEnum
+
+
+class TaskState(StrEnum):
+    NEW = "NEW"
+    AWAITING = "AWAITING"
+    PENDING = "PENDING"
+    BUILDING = "BUILDING"
+    COMMITTING = "COMMITTING"
+    TESTED = "TESTED"
+    FAILING = "FAILING"
+    FAILED = "FAILED"
+    DONE = "DONE"
+    SWEPT = "SWEPT"
+    EPERM = "EPERM"
+    POSTPONED = "POSTPONED"
+
+
+class Repo(StrEnum):
+    sisyphus = "sisyphus"
+    p11 = "p11"
+    p10 = "p10"
+    p9 = "p9"
+    p8 = "p8"
+    p7 = "p7"
+    p6 = "p6"
+    c10f2 = "c10f2"
+    c10f1 = "c10f1"
+    c9f2 = "c9f2"
+    c9f1 = "c9f1"
+    c9m2 = "c9m2"
+    c9m1 = "c9m1"
+    c8_1 = "c8.1"
+    c8 = "c8"
+    c7_1 = "c7.1"
+    c7 = "c7"
+    t7 = "t7"
+    icarus = "icarus"
+
+
+class ApprovalGroup(StrEnum):
+    maint = "maint"
+    tester = "tester"