mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-25 17:12:03 +00:00
267 lines
8.3 KiB
Python
267 lines
8.3 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import logging
|
|
import re
|
|
from typing import Tuple
|
|
|
|
from github import Github
|
|
|
|
from commit_status_helper import (
|
|
get_commit,
|
|
post_labels,
|
|
remove_labels,
|
|
reset_mergeable_check,
|
|
)
|
|
from env_helper import GITHUB_RUN_URL, GITHUB_REPOSITORY, GITHUB_SERVER_URL
|
|
from get_robot_token import get_best_robot_token
|
|
from pr_info import FORCE_TESTS_LABEL, PRInfo
|
|
from workflow_approve_rerun_lambda.app import TRUSTED_CONTRIBUTORS
|
|
|
|
NAME = "Run Check"
|
|
|
|
TRUSTED_ORG_IDS = {
|
|
7409213, # yandex
|
|
28471076, # altinity
|
|
54801242, # clickhouse
|
|
}
|
|
|
|
OK_SKIP_LABELS = {"release", "pr-backport", "pr-cherrypick"}
|
|
CAN_BE_TESTED_LABEL = "can be tested"
|
|
DO_NOT_TEST_LABEL = "do not test"
|
|
SUBMODULE_CHANGED_LABEL = "submodule changed"
|
|
|
|
# They are used in .github/PULL_REQUEST_TEMPLATE.md, keep comments there
|
|
# updated accordingly
|
|
LABELS = {
|
|
"pr-backward-incompatible": ["Backward Incompatible Change"],
|
|
"pr-bugfix": [
|
|
"Bug Fix",
|
|
"Bug Fix (user-visible misbehaviour in official stable or prestable release)",
|
|
"Bug Fix (user-visible misbehavior in official stable or prestable release)",
|
|
],
|
|
"pr-build": [
|
|
"Build/Testing/Packaging Improvement",
|
|
"Build Improvement",
|
|
"Build/Testing Improvement",
|
|
"Build",
|
|
"Packaging Improvement",
|
|
],
|
|
"pr-documentation": [
|
|
"Documentation (changelog entry is not required)",
|
|
"Documentation",
|
|
],
|
|
"pr-feature": ["New Feature"],
|
|
"pr-improvement": ["Improvement"],
|
|
"pr-not-for-changelog": [
|
|
"Not for changelog (changelog entry is not required)",
|
|
"Not for changelog",
|
|
],
|
|
"pr-performance": ["Performance Improvement"],
|
|
}
|
|
|
|
CATEGORY_TO_LABEL = {c: lb for lb, categories in LABELS.items() for c in categories}
|
|
|
|
|
|
def pr_is_by_trusted_user(pr_user_login, pr_user_orgs):
|
|
if pr_user_login.lower() in TRUSTED_CONTRIBUTORS:
|
|
logging.info("User '%s' is trusted", pr_user_login)
|
|
return True
|
|
|
|
logging.info("User '%s' is not trusted", pr_user_login)
|
|
|
|
for org_id in pr_user_orgs:
|
|
if org_id in TRUSTED_ORG_IDS:
|
|
logging.info(
|
|
"Org '%s' is trusted; will mark user %s as trusted",
|
|
org_id,
|
|
pr_user_login,
|
|
)
|
|
return True
|
|
logging.info("Org '%s' is not trusted", org_id)
|
|
|
|
return False
|
|
|
|
|
|
# Returns whether we should look into individual checks for this PR. If not, it
|
|
# can be skipped entirely.
|
|
# Returns can_run, description, labels_state
|
|
def should_run_checks_for_pr(pr_info: PRInfo) -> Tuple[bool, str, str]:
|
|
# Consider the labels and whether the user is trusted.
|
|
print("Got labels", pr_info.labels)
|
|
if FORCE_TESTS_LABEL in pr_info.labels:
|
|
return True, f"Labeled '{FORCE_TESTS_LABEL}'", "pending"
|
|
|
|
if DO_NOT_TEST_LABEL in pr_info.labels:
|
|
return False, f"Labeled '{DO_NOT_TEST_LABEL}'", "success"
|
|
|
|
if CAN_BE_TESTED_LABEL not in pr_info.labels and not pr_is_by_trusted_user(
|
|
pr_info.user_login, pr_info.user_orgs
|
|
):
|
|
return False, "Needs 'can be tested' label", "failure"
|
|
|
|
if OK_SKIP_LABELS.intersection(pr_info.labels):
|
|
return (
|
|
False,
|
|
"Don't try new checks for release/backports/cherry-picks",
|
|
"success",
|
|
)
|
|
|
|
return True, "No special conditions apply", "pending"
|
|
|
|
|
|
def check_pr_description(pr_info) -> Tuple[str, str]:
|
|
lines = list(
|
|
map(lambda x: x.strip(), pr_info.body.split("\n") if pr_info.body else [])
|
|
)
|
|
lines = [re.sub(r"\s+", " ", line) for line in lines]
|
|
|
|
# Check if body contains "Reverts ClickHouse/ClickHouse#36337"
|
|
if [
|
|
True
|
|
for line in lines
|
|
if re.match(rf"\AReverts {GITHUB_REPOSITORY}#[\d]+\Z", line)
|
|
]:
|
|
return "", LABELS["pr-not-for-changelog"][0]
|
|
|
|
category = ""
|
|
entry = ""
|
|
|
|
i = 0
|
|
while i < len(lines):
|
|
if re.match(r"(?i)^[#>*_ ]*change\s*log\s*category", lines[i]):
|
|
i += 1
|
|
if i >= len(lines):
|
|
break
|
|
# Can have one empty line between header and the category
|
|
# itself. Filter it out.
|
|
if not lines[i]:
|
|
i += 1
|
|
if i >= len(lines):
|
|
break
|
|
category = re.sub(r"^[-*\s]*", "", lines[i])
|
|
i += 1
|
|
|
|
# Should not have more than one category. Require empty line
|
|
# after the first found category.
|
|
if i >= len(lines):
|
|
break
|
|
if lines[i]:
|
|
second_category = re.sub(r"^[-*\s]*", "", lines[i])
|
|
result_status = (
|
|
"More than one changelog category specified: '"
|
|
+ category
|
|
+ "', '"
|
|
+ second_category
|
|
+ "'"
|
|
)
|
|
return result_status[:140], category
|
|
|
|
elif re.match(
|
|
r"(?i)^[#>*_ ]*(short\s*description|change\s*log\s*entry)", lines[i]
|
|
):
|
|
i += 1
|
|
# Can have one empty line between header and the entry itself.
|
|
# Filter it out.
|
|
if i < len(lines) and not lines[i]:
|
|
i += 1
|
|
# All following lines until empty one are the changelog entry.
|
|
entry_lines = []
|
|
while i < len(lines) and lines[i]:
|
|
entry_lines.append(lines[i])
|
|
i += 1
|
|
entry = " ".join(entry_lines)
|
|
# Don't accept changelog entries like '...'.
|
|
entry = re.sub(r"[#>*_.\- ]", "", entry)
|
|
else:
|
|
i += 1
|
|
|
|
if not category:
|
|
return "Changelog category is empty", category
|
|
|
|
# Filter out the PR categories that are not for changelog.
|
|
if re.match(
|
|
r"(?i)doc|((non|in|not|un)[-\s]*significant)|(not[ ]*for[ ]*changelog)",
|
|
category,
|
|
):
|
|
return "", category
|
|
|
|
if not entry:
|
|
return f"Changelog entry required for category '{category}'", category
|
|
|
|
return "", category
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
pr_info = PRInfo(need_orgs=True, pr_event_from_api=True, need_changed_files=True)
|
|
can_run, description, labels_state = should_run_checks_for_pr(pr_info)
|
|
gh = Github(get_best_robot_token(), per_page=100)
|
|
commit = get_commit(gh, pr_info.sha)
|
|
|
|
description_error, category = check_pr_description(pr_info)
|
|
pr_labels_to_add = []
|
|
pr_labels_to_remove = []
|
|
if (
|
|
category in CATEGORY_TO_LABEL
|
|
and CATEGORY_TO_LABEL[category] not in pr_info.labels
|
|
):
|
|
pr_labels_to_add.append(CATEGORY_TO_LABEL[category])
|
|
|
|
for label in pr_info.labels:
|
|
if (
|
|
label in CATEGORY_TO_LABEL.values()
|
|
and category in CATEGORY_TO_LABEL
|
|
and label != CATEGORY_TO_LABEL[category]
|
|
):
|
|
pr_labels_to_remove.append(label)
|
|
|
|
if pr_info.has_changes_in_submodules():
|
|
pr_labels_to_add.append(SUBMODULE_CHANGED_LABEL)
|
|
elif SUBMODULE_CHANGED_LABEL in pr_info.labels:
|
|
pr_labels_to_remove.append(SUBMODULE_CHANGED_LABEL)
|
|
|
|
print(f"change labels: add {pr_labels_to_add}, remove {pr_labels_to_remove}")
|
|
if pr_labels_to_add:
|
|
post_labels(gh, pr_info, pr_labels_to_add)
|
|
|
|
if pr_labels_to_remove:
|
|
remove_labels(gh, pr_info, pr_labels_to_remove)
|
|
|
|
reset_mergeable_check(commit, "skipped")
|
|
|
|
if description_error:
|
|
print(
|
|
"::error ::Cannot run, PR description does not match the template: "
|
|
f"{description_error}"
|
|
)
|
|
logging.info(
|
|
"PR body doesn't match the template: (start)\n%s\n(end)\n" "Reason: %s",
|
|
pr_info.body,
|
|
description_error,
|
|
)
|
|
url = (
|
|
f"{GITHUB_SERVER_URL}/{GITHUB_REPOSITORY}/"
|
|
"blob/master/.github/PULL_REQUEST_TEMPLATE.md?plain=1"
|
|
)
|
|
commit.create_status(
|
|
context=NAME,
|
|
description=description_error[:139],
|
|
state="failure",
|
|
target_url=url,
|
|
)
|
|
sys.exit(1)
|
|
|
|
url = GITHUB_RUN_URL
|
|
if not can_run:
|
|
print("::notice ::Cannot run")
|
|
commit.create_status(
|
|
context=NAME, description=description, state=labels_state, target_url=url
|
|
)
|
|
sys.exit(1)
|
|
else:
|
|
print("::notice ::Can run")
|
|
commit.create_status(
|
|
context=NAME, description=description, state="pending", target_url=url
|
|
)
|