Skip to content

Commit

Permalink
Fixes to support check runs
Browse files Browse the repository at this point in the history
  • Loading branch information
tcarmet committed Apr 25, 2024
1 parent 24a94a2 commit d837d0f
Show file tree
Hide file tree
Showing 6 changed files with 27 additions and 21 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -10,3 +10,4 @@ settings.yml
.coverage
.pytest_cache/
build/
.env
9 changes: 5 additions & 4 deletions bert_e/exceptions.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,13 +22,13 @@

class BertE_Exception(Exception):
code = -1
status: Literal[None, "in_progress", "queued", "success", "failure"] = None


# base exceptions
class TemplateException(BertE_Exception):
code = -2
template = None
status: Literal[None, "in_progress", "success", "failure"] = None
# whether to re-publish if the message is already in the history
dont_repeat_if_in_history = -1

Expand Down Expand Up @@ -148,7 +148,7 @@ class Conflict(TemplateException):
class ApprovalRequired(TemplateException):
code = 115
template = 'need_approval.md'
status = "in_progress"
status = "queued"


class BuildFailed(TemplateException):
Expand All @@ -160,7 +160,7 @@ class BuildFailed(TemplateException):
class AfterPullRequest(TemplateException):
code = 120
template = 'after_pull_request.md'
status = "in_progress"
status = "queued"


class IntegrationDataCreated(InformationException):
Expand Down Expand Up @@ -265,7 +265,7 @@ class RequestIntegrationBranches(TemplateException):
code = 135
template = "request_integration_branches.md"
# TODO: review if it should be failure.
status = "in_progress"
status = "queued"


class QueueBuildFailedMessage(TemplateException):
Expand Down Expand Up @@ -568,6 +568,7 @@ class NothingToDo(SilentException):

class BuildInProgress(SilentException):
code = 303
status = "in_progress"


class BuildNotStarted(SilentException):
Expand Down
30 changes: 17 additions & 13 deletions bert_e/git_host/github/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,8 @@ def __init__(self, login: str, password: str, email: str,
self.password = password
self.app_id = app_id
self.installation_id = installation_id
self.private_key = jwk_from_pem(private_key.encode('utf-8')) if private_key else None
self.private_key = jwk_from_pem(private_key.encode('utf-8')) \
if private_key else None
self.email = email
self.org = org
self.base_url = base_url.rstrip('/')
Expand Down Expand Up @@ -832,11 +833,11 @@ def set_bot_status(self, status: str | None, title: str, summary: str):
if status == "success":
conclusion = "success"
status = "completed"
elif status == "in_progress":
conclusion = None
elif status == "failure":
conclusion = "failure"
status = "completed"
else:
conclusion = None

self._add_checkrun(
name='bert-e', status=status, conclusion=conclusion,
Expand All @@ -846,18 +847,21 @@ def set_bot_status(self, status: str | None, title: str, summary: str):
def _add_checkrun(
self, name: str, status: str, conclusion: str | None,
title: str, summary: str):
data = {
'name': name,
'head_sha': self.src_commit,
'status': status,
'output': {
'title': title,
'summary': summary,
},
}
if conclusion is not None:
data['conclusion'] = conclusion
LOG.debug(data)
return CheckRun.create(
client=self.client,
data={
'name': name,
'head_sha': self.src_commit,
'status': status,
'conclusion': conclusion,
'output': {
'title': title,
'summary': summary,
},
},
data=data,
owner=self.repo.owner, repo=self.repo.slug
)

Expand Down
2 changes: 1 addition & 1 deletion bert_e/git_host/github/schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ class AggregateCheckSuites(GitHubSchema):
class Output(GitHubSchema):
title = fields.Str()
summary = fields.Str()
text = fields.Str()
text = fields.Str(allow_none=True)


class CheckRun(GitHubSchema):
Expand Down
4 changes: 2 additions & 2 deletions bert_e/workflow/pr_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,8 +87,8 @@ def _send_comment(settings, pull_request: AbstractPullRequest, msg: str,
def _send_bot_status(settings, pull_request: AbstractPullRequest,
comment: exceptions.TemplateException):
"""Post the bot status in a pull request."""
if settings.send_bot_status is False:
LOG.debug("Not sending bot status (send_bot_status==False)")
if settings.send_bot_status is False or comment.status is None:
LOG.debug("No need to send bot status")
return
LOG.info(f"Setting bot status to {comment.status} as {comment.title}")
pull_request.set_bot_status(
Expand Down
2 changes: 1 addition & 1 deletion tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ setenv =
WEBHOOK_LOGIN = {env:WEBHOOK_LOGIN:'webhook'}
WEBHOOK_PWD = {env:WEBHOOK_PWD:'webhook'}

commands = bert-e-serve -v -f {env:BERT_E_SETTINGS:'settings.yml'}
commands = bert-e-serve -v -f {env:BERT_E_SETTINGS:'settings.yml'} -p 8000

[testenv:tests-githost]
passenv = BERT_E_*
Expand Down

0 comments on commit d837d0f

Please sign in to comment.