100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
from chalicelib.core.integration_base_issue import BaseIntegrationIssue
|
|
from chalicelib.utils import github_client_v3
|
|
from chalicelib.utils.github_client_v3 import github_formatters as formatter
|
|
|
|
|
|
class GithubIntegrationIssue(BaseIntegrationIssue):
|
|
def __init__(self, integration_token):
|
|
self.__client = github_client_v3.githubV3Request(integration_token)
|
|
super(GithubIntegrationIssue, self).__init__("GITHUB", integration_token)
|
|
|
|
async def get_current_user(self):
|
|
return formatter.user(await self.__client.get("/user"))
|
|
|
|
async def get_meta(self, repoId):
|
|
current_user = await self.get_current_user()
|
|
try:
|
|
users = await self.__client.get(f"/repositories/{repoId}/collaborators")
|
|
except Exception as e:
|
|
users = []
|
|
users = [formatter.user(u) for u in users]
|
|
if current_user not in users:
|
|
users.insert(0, current_user)
|
|
meta = {
|
|
'users': users,
|
|
'issueTypes': [formatter.label(l) for l in
|
|
await self.__client.get(f"/repositories/{repoId}/labels")]
|
|
}
|
|
|
|
return meta
|
|
|
|
async def create_new_assignment(self, integration_project_id, title, description, assignee,
|
|
issue_type):
|
|
repoId = integration_project_id
|
|
assignees = [assignee]
|
|
labels = [str(issue_type)]
|
|
|
|
metas = await self.get_meta(repoId)
|
|
real_assignees = []
|
|
for a in assignees:
|
|
for u in metas["users"]:
|
|
if a == str(u["id"]):
|
|
real_assignees.append(u["name"])
|
|
break
|
|
real_labels = ["OpenReplay"]
|
|
for l in labels:
|
|
found = False
|
|
for ll in metas["issueTypes"]:
|
|
if l == str(ll["id"]):
|
|
real_labels.append(ll["name"])
|
|
found = True
|
|
break
|
|
if not found:
|
|
real_labels.append(l)
|
|
issue = await self.__client.post(f"/repositories/{repoId}/issues", body={"title": title, "body": description,
|
|
"assignees": real_assignees,
|
|
"labels": real_labels})
|
|
return formatter.issue(issue)
|
|
|
|
async def get_by_ids(self, saved_issues):
|
|
results = []
|
|
for i in saved_issues:
|
|
results.append(await self.get(integration_project_id=i["integrationProjectId"], assignment_id=i["id"]))
|
|
return {"issues": results}
|
|
|
|
async def get(self, integration_project_id, assignment_id):
|
|
repoId = integration_project_id
|
|
issueNumber = assignment_id
|
|
issue = await self.__client.get(f"/repositories/{repoId}/issues/{issueNumber}")
|
|
issue = formatter.issue(issue)
|
|
if issue["commentsCount"] > 0:
|
|
issue["comments"] = [formatter.comment(c) for c in
|
|
await self.__client.get(f"/repositories/{repoId}/issues/{issueNumber}/comments")]
|
|
return issue
|
|
|
|
async def comment(self, integration_project_id, assignment_id, comment):
|
|
repoId = integration_project_id
|
|
issueNumber = assignment_id
|
|
commentCreated = await self.__client.post(f"/repositories/{repoId}/issues/{issueNumber}/comments",
|
|
body={"body": comment})
|
|
return formatter.comment(commentCreated)
|
|
|
|
async def get_metas(self, integration_project_id):
|
|
current_user = await self.get_current_user()
|
|
try:
|
|
users = await self.__client.get(f"/repositories/{integration_project_id}/collaborators")
|
|
except Exception as e:
|
|
users = []
|
|
users = [formatter.user(u) for u in users]
|
|
if current_user not in users:
|
|
users.insert(0, current_user)
|
|
|
|
return {"provider": self.provider.lower(),
|
|
'users': users,
|
|
'issueTypes': [formatter.label(l) for l in
|
|
await self.__client.get(f"/repositories/{integration_project_id}/labels")]
|
|
}
|
|
|
|
async def get_projects(self):
|
|
repos = await self.__client.get("/user/repos")
|
|
return [formatter.repo(r) for r in repos]
|