81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
from typing import Annotated, Any
|
|
|
|
from arcade.sdk import ToolContext, tool
|
|
from arcade.sdk.auth import Asana
|
|
|
|
from arcade_asana.constants import PROJECT_OPT_FIELDS
|
|
from arcade_asana.models import AsanaClient
|
|
from arcade_asana.utils import (
|
|
get_next_page,
|
|
get_unique_workspace_id_or_raise_error,
|
|
remove_none_values,
|
|
)
|
|
|
|
|
|
@tool(requires_auth=Asana(scopes=["default"]))
|
|
async def get_project_by_id(
|
|
context: ToolContext,
|
|
project_id: Annotated[str, "The ID of the project."],
|
|
) -> Annotated[
|
|
dict[str, Any],
|
|
"Get a project by its ID",
|
|
]:
|
|
"""Get an Asana project by its ID"""
|
|
client = AsanaClient(context.get_auth_token_or_empty())
|
|
response = await client.get(
|
|
f"/projects/{project_id}",
|
|
params={"opt_fields": PROJECT_OPT_FIELDS},
|
|
)
|
|
return {"project": response["data"]}
|
|
|
|
|
|
@tool(requires_auth=Asana(scopes=["default"]))
|
|
async def list_projects(
|
|
context: ToolContext,
|
|
team_id: Annotated[
|
|
str | None,
|
|
"The team ID to get projects from. Defaults to None (does not filter by team).",
|
|
] = None,
|
|
workspace_id: Annotated[
|
|
str | None,
|
|
"The workspace ID to get projects from. Defaults to None. If not provided and the user "
|
|
"has only one workspace, it will use that workspace. If not provided and the user has "
|
|
"multiple workspaces, it will raise an error listing the available workspaces.",
|
|
] = None,
|
|
limit: Annotated[
|
|
int, "The maximum number of projects to return. Min is 1, max is 100. Defaults to 100."
|
|
] = 100,
|
|
next_page_token: Annotated[
|
|
str | None,
|
|
"The token to retrieve the next page of projects. Defaults to None (start from the first "
|
|
"page of projects).",
|
|
] = None,
|
|
) -> Annotated[
|
|
dict[str, Any],
|
|
"List projects in Asana associated to teams the current user is a member of",
|
|
]:
|
|
"""List projects in Asana"""
|
|
# Note: Asana recommends filtering by team to avoid timeout in large domains.
|
|
# Ref: https://developers.asana.com/reference/getprojects
|
|
limit = max(1, min(100, limit))
|
|
|
|
workspace_id = workspace_id or await get_unique_workspace_id_or_raise_error(context)
|
|
|
|
client = AsanaClient(context.get_auth_token_or_empty())
|
|
|
|
response = await client.get(
|
|
"/projects",
|
|
params=remove_none_values({
|
|
"limit": limit,
|
|
"offset": next_page_token,
|
|
"team": team_id,
|
|
"workspace": workspace_id,
|
|
"opt_fields": PROJECT_OPT_FIELDS,
|
|
}),
|
|
)
|
|
|
|
return {
|
|
"projects": response["data"],
|
|
"count": len(response["data"]),
|
|
"next_page": get_next_page(response),
|
|
}
|