74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
import logging
|
|
|
|
from github import Issue
|
|
|
|
from actioner.clients import get_todoist_client, github
|
|
from actioner.utils import get_todoist_project_from_repo
|
|
from actioner.utils.github import get_existing_task, get_issue_link, get_relevant_issues
|
|
from actioner.utils.todoist import get_existing_tasks, is_task_completed
|
|
|
|
LABEL_TO_STATUS = {"must have": 4, "critical": 4, "should have": 2}
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_status_for_issue(issue: Issue) -> int:
|
|
priorities = {LABEL_TO_STATUS.get(label.name.lower(), 1) for label in issue.labels}
|
|
return max(priorities, default=1)
|
|
|
|
|
|
def issue_to_task_name(issue: Issue) -> str:
|
|
return get_issue_link(issue) + ": " + issue.title
|
|
|
|
|
|
def todoist_assigned_issues():
|
|
todoist = get_todoist_client()
|
|
me = github.get_user()
|
|
todoist.projects.sync()
|
|
todoist.items.sync()
|
|
for issue in get_relevant_issues():
|
|
project_id = get_todoist_project_from_repo(issue.repository.full_name)
|
|
if not project_id:
|
|
continue
|
|
existing_tasks = get_existing_tasks(project_id, todoist)
|
|
me_assigned = me.login in {assignee.login for assignee in issue.assignees}
|
|
existing_task_id = get_existing_task(existing_tasks, issue)
|
|
|
|
if existing_task_id and not me_assigned:
|
|
logger.info("Deleting task for '{}'".format(issue.title))
|
|
todoist.items.delete(existing_task_id)
|
|
|
|
if not me_assigned:
|
|
continue
|
|
|
|
elif (
|
|
issue.state == "closed"
|
|
and existing_task_id is not None
|
|
and not is_task_completed(todoist.items.get_by_id(existing_task_id))
|
|
):
|
|
logger.info("Completing task for '{}'".format(issue.title))
|
|
todoist.items.complete([existing_task_id])
|
|
|
|
elif issue.state == "open":
|
|
if existing_task_id is None:
|
|
logger.info("Creating task for '{}'".format(issue.title))
|
|
existing_task_id = todoist.items.add(
|
|
issue_to_task_name(issue), project_id=project_id
|
|
)["id"]
|
|
existing_task = todoist.items.get_by_id(existing_task_id)
|
|
|
|
if is_task_completed(existing_task):
|
|
logger.info("Re-opening task '{}'".format(issue.title))
|
|
todoist.items.unarchive(existing_task_id)
|
|
todoist.items.uncomplete(existing_task_id)
|
|
|
|
existing_task.update(
|
|
content=issue_to_task_name(issue), priority=get_status_for_issue(issue)
|
|
)
|
|
if issue.milestone and issue.milestone.due_on:
|
|
existing_task.update(
|
|
date_string=issue.milestone.due_on.strftime("%d/%m/%Y")
|
|
)
|
|
|
|
todoist.commit()
|