Patrick Nisble
110 lines
2.9 KiB
Python
110 lines
2.9 KiB
Python
from collections.abc import Iterator
|
|
from dataclasses import dataclass, field
|
|
from json import loads
|
|
from pydantic import BaseModel
|
|
from requests import HTTPError, get
|
|
|
|
from .common import EnvVar, JSONDataMap, JSONData, JSONDataList, JSONDataScalar
|
|
|
|
|
|
class ClickupStatus(BaseModel):
|
|
status: str
|
|
|
|
@property
|
|
def status_symbol(self) -> str:
|
|
match self.status:
|
|
case "backlog":
|
|
return ""
|
|
case "selected for development":
|
|
return ""
|
|
case "in progress":
|
|
return ""
|
|
case "in review":
|
|
return ""
|
|
case "done":
|
|
return ""
|
|
case "closed":
|
|
return ""
|
|
case "on hold":
|
|
return ""
|
|
case _:
|
|
return self.status
|
|
|
|
|
|
class ClickupTask(BaseModel):
|
|
id: str
|
|
name: str
|
|
markdown_description: str
|
|
status: ClickupStatus
|
|
|
|
|
|
class ClickupList(BaseModel):
|
|
task_count: int
|
|
|
|
|
|
@dataclass
|
|
class ClickupSession:
|
|
auth_key: str = field(
|
|
default_factory=EnvVar(
|
|
"CLICKUP_AUTH",
|
|
"clickup auth token is required to be set",
|
|
)
|
|
)
|
|
workspace_id: str = field(
|
|
default_factory=EnvVar(
|
|
"CLICKUP_WORKSPACE_ID",
|
|
"clickup workspace id is required to be set",
|
|
)
|
|
)
|
|
user_id: str = field(
|
|
default_factory=EnvVar(
|
|
"CLICKUP_USER_ID",
|
|
"clickup user id is required to be set",
|
|
)
|
|
)
|
|
base_url: str = "https://api.clickup.com/api/v2"
|
|
|
|
def _get(self, endpoint: str, **query_params: str) -> JSONDataMap:
|
|
with get(
|
|
self.base_url + endpoint,
|
|
query_params,
|
|
headers={
|
|
"accept": "application/json",
|
|
"Authorization": self.auth_key,
|
|
},
|
|
) as resp:
|
|
return resp.json()
|
|
|
|
def get_tasks(self, **filters: str) -> Iterator[ClickupTask]:
|
|
pix = 0
|
|
while True:
|
|
partial_data = self._get(
|
|
f"/team/{self.workspace_id}/task",
|
|
**{
|
|
"subtasks": "true",
|
|
"include_markdown_description": "true",
|
|
"include_closed": "true",
|
|
"assignees[]": self.user_id,
|
|
"page": pix,
|
|
}
|
|
| filters,
|
|
).get("tasks", [])
|
|
pix += 1
|
|
|
|
for inst in partial_data:
|
|
yield ClickupTask.model_validate(inst)
|
|
|
|
if len(partial_data) < 100:
|
|
break
|
|
|
|
def get_task(self, task_id: str) -> ClickupTask:
|
|
return ClickupTask.model_validate(
|
|
self._get(
|
|
f"/task/{task_id}",
|
|
include_markdown_description="true",
|
|
),
|
|
)
|
|
|
|
def get_list(self, list_id: str) -> ClickupList:
|
|
return ClickupList.model_validate(self._get(f"/list/{list_id}"))
|