ClickHouse/tests/ci/finish_check.py

45 lines
1.2 KiB
Python
Raw Normal View History

2021-09-29 07:47:15 +00:00
#!/usr/bin/env python3
import logging
from github import Github
2021-11-26 14:00:09 +00:00
from env_helper import GITHUB_RUN_URL
2021-11-26 14:00:09 +00:00
from pr_info import PRInfo
from get_robot_token import get_best_robot_token
2021-11-12 12:36:25 +00:00
from commit_status_helper import get_commit
2021-09-29 07:47:15 +00:00
NAME = "Run Check"
2021-09-29 07:47:15 +00:00
2021-11-26 14:00:09 +00:00
2021-09-29 07:47:15 +00:00
def filter_statuses(statuses):
"""
Squash statuses to latest state
1. context="first", state="success", update_time=1
2. context="second", state="success", update_time=2
3. context="first", stat="failure", update_time=3
=========>
1. context="second", state="success"
2. context="first", stat="failure"
"""
filt = {}
for status in sorted(statuses, key=lambda x: x.updated_at):
filt[status.context] = status
return filt
2021-11-26 14:00:09 +00:00
2021-09-29 07:47:15 +00:00
if __name__ == "__main__":
logging.basicConfig(level=logging.INFO)
2021-11-26 14:00:09 +00:00
pr_info = PRInfo(need_orgs=True)
gh = Github(get_best_robot_token())
2021-09-29 07:47:15 +00:00
commit = get_commit(gh, pr_info.sha)
url = GITHUB_RUN_URL
2021-09-29 07:47:15 +00:00
statuses = filter_statuses(list(commit.get_statuses()))
if NAME in statuses and statuses[NAME].state == "pending":
commit.create_status(
context=NAME,
description="All checks finished",
state="success",
target_url=url,
)