ClickHouse/tests/ci/finish_check.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

81 lines
2.5 KiB
Python
Raw Normal View History

2021-09-29 07:47:15 +00:00
#!/usr/bin/env python3
import logging
2024-04-25 11:28:06 +00:00
import sys
from github import Github
2021-11-26 14:00:09 +00:00
from ci_config import StatusNames
from commit_status_helper import (
get_commit,
get_commit_filtered_statuses,
post_commit_status,
2024-04-25 11:28:06 +00:00
set_mergeable_check,
trigger_mergeable_check,
update_upstream_sync_status,
)
2023-04-20 11:55:33 +00:00
from get_robot_token import get_best_robot_token
from pr_info import PRInfo
from report import PENDING, SUCCESS
from synchronizer_utils import SYNC_BRANCH_PREFIX
from env_helper import GITHUB_REPOSITORY, GITHUB_UPSTREAM_REPOSITORY
2021-09-29 07:47:15 +00:00
2021-11-26 14:00:09 +00:00
2023-04-20 11:55:33 +00:00
def main():
2021-09-29 07:47:15 +00:00
logging.basicConfig(level=logging.INFO)
2024-04-25 11:28:06 +00:00
has_failure = False
# FIXME: temporary hack to fail Mergeable Check in MQ if pipeline has any failed jobs
if len(sys.argv) > 1 and sys.argv[1] == "--pipeline-failure":
has_failure = True
2021-11-26 14:00:09 +00:00
pr_info = PRInfo(need_orgs=True)
gh = Github(get_best_robot_token(), per_page=100)
2021-09-29 07:47:15 +00:00
commit = get_commit(gh, pr_info.sha)
2024-04-25 11:28:06 +00:00
statuses = None
if pr_info.is_merge_queue:
# in MQ Mergeable check status must never be green if any failures in workflow
if has_failure:
set_mergeable_check(commit, "workflow failed", "failure")
else:
2024-04-25 15:05:02 +00:00
# This must be the only place where green MCheck is set in the MQ (in the end of CI) to avoid early merge
2024-04-25 11:28:06 +00:00
set_mergeable_check(commit, "workflow passed", "success")
else:
statuses = get_commit_filtered_statuses(commit)
state = trigger_mergeable_check(commit, statuses, set_if_green=True)
# Process upstream StatusNames.SYNC
if (
pr_info.head_ref.startswith(f"{SYNC_BRANCH_PREFIX}/pr/")
and GITHUB_REPOSITORY != GITHUB_UPSTREAM_REPOSITORY
):
upstream_pr_number = int(pr_info.head_ref.split("/pr/", maxsplit=1)[1])
update_upstream_sync_status(
upstream_pr_number,
pr_info.number,
gh,
state,
can_set_green_mergeable_status=True,
)
statuses = [s for s in statuses if s.context == StatusNames.CI]
2024-04-11 11:58:44 +00:00
if not statuses:
return
# Take the latest status
status = statuses[-1]
if status.state == PENDING:
post_commit_status(
commit,
state, # map Mergeable Check status to CI Running
2024-04-11 11:58:44 +00:00
status.target_url,
"All checks finished",
StatusNames.CI,
2024-04-11 11:58:44 +00:00
pr_info,
dump_to_file=True,
)
2023-04-20 11:55:33 +00:00
if __name__ == "__main__":
main()