mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-27 18:12:02 +00:00
125 lines
3.4 KiB
Python
125 lines
3.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""Script for automatic sync PRs handling in private repos"""
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
from commit_status_helper import get_commit, post_commit_status
|
|
from get_robot_token import get_best_robot_token
|
|
from github_helper import GitHub
|
|
from pr_info import PRInfo
|
|
from report import SUCCESS
|
|
from ci_config import CI
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
|
|
description="Script for handling sync PRs",
|
|
)
|
|
parser.add_argument(
|
|
"--merge",
|
|
action="store_true",
|
|
help="merge sync pr",
|
|
)
|
|
parser.add_argument(
|
|
"--status",
|
|
action="store_true",
|
|
help="check and set sync pr status",
|
|
)
|
|
args = parser.parse_args()
|
|
return args
|
|
|
|
|
|
def merge_sync_pr(gh, sync_pr):
|
|
if not sync_pr:
|
|
print("Sync PR not found - exiting")
|
|
return
|
|
|
|
if sync_pr.state == "closed":
|
|
print(f"Sync PR [{sync_pr.number}] already closed - exiting")
|
|
sys.exit(0)
|
|
|
|
if sync_pr.state != "open":
|
|
print(
|
|
f"WARNING: Unknown Sync PR [{sync_pr.number}] state [{sync_pr.state}] - exiting"
|
|
)
|
|
sys.exit(0)
|
|
|
|
print(f"Trying to merge Sync PR [{sync_pr.number}]")
|
|
if sync_pr.draft:
|
|
gh.toggle_pr_draft(sync_pr)
|
|
sync_pr.merge()
|
|
|
|
|
|
def set_sync_status(gh, pr_info, sync_pr):
|
|
# FIXME: uncomment posting red Sync status to prohibit merge in MQ if PR state fetching works good
|
|
if not sync_pr:
|
|
# post_commit_status(
|
|
# get_commit(gh, pr_info.sha), FAILURE, "", "Sync PR not found", StatusNames.SYNC
|
|
# )
|
|
return
|
|
|
|
# FIXME: fetch sync pr in a proper way
|
|
# retries = 0
|
|
# while sync_pr.mergeable_state == "unknown" and retries < 3:
|
|
# retries += 1
|
|
# print(f"Unknown status. Trying to fetch again [{retries}/3]")
|
|
# time.sleep(5)
|
|
# sync_pr = gh.get_pulls_from_search(
|
|
# query=f"head:sync-upstream/pr/{sync_pr.number} org:ClickHouse type:pr",
|
|
# repo="ClickHouse/clickhouse-private",
|
|
# )
|
|
|
|
if sync_pr.mergeable_state == "clean":
|
|
print(f"Sync PR [{sync_pr.number}] is clean")
|
|
post_commit_status(
|
|
get_commit(gh, pr_info.sha), SUCCESS, "", "", CI.StatusNames.SYNC
|
|
)
|
|
else:
|
|
print(
|
|
f"Sync PR [{sync_pr}] is not mergeable, state [{sync_pr.mergeable_state}]"
|
|
)
|
|
# post_commit_status(
|
|
# get_commit(gh, pr_info.sha),
|
|
# FAILURE,
|
|
# "",
|
|
# f"state: {sync_pr.mergeable_state}",
|
|
# StatusNames.SYNC,
|
|
# )
|
|
|
|
|
|
def main():
|
|
args = parse_args()
|
|
|
|
assert args.merge ^ args.status
|
|
|
|
gh = GitHub(get_best_robot_token())
|
|
|
|
pr_info = PRInfo()
|
|
assert pr_info.merged_pr, "BUG. merged PR number could not been determined"
|
|
|
|
prs = gh.get_pulls_from_search(
|
|
query=f"head:sync-upstream/pr/{pr_info.merged_pr} org:ClickHouse type:pr",
|
|
repo="ClickHouse/clickhouse-private",
|
|
)
|
|
|
|
sync_pr = None
|
|
|
|
if len(prs) > 1:
|
|
print(f"WARNING: More than one PR found [{prs}] - exiting")
|
|
elif len(prs) == 0:
|
|
print("WARNING: No Sync PR found")
|
|
else:
|
|
sync_pr = prs[0]
|
|
|
|
if args.merge:
|
|
merge_sync_pr(gh, sync_pr)
|
|
elif args.status:
|
|
set_sync_status(gh, pr_info, sync_pr)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|