mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-19 16:20:50 +00:00
108 lines
3.4 KiB
Python
Executable File
108 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Note: should work with python 2 and 3
|
|
|
|
from github import Github
|
|
import datetime
|
|
import tabulate
|
|
import argparse
|
|
from termcolor import colored
|
|
import sys
|
|
|
|
COLORMAP = {
|
|
"success": colored("success", "green"),
|
|
"failure": colored("failure", "red"),
|
|
"error": colored("error", "red"),
|
|
"pending": colored("pending", "yellow"),
|
|
"not run": colored("not run", "white"),
|
|
}
|
|
|
|
|
|
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 list(filt.values())
|
|
|
|
|
|
def get_filtered_statuses(commit):
|
|
return _filter_statuses(commit.get_statuses())
|
|
|
|
|
|
def get_commits(repo, since):
|
|
return sorted(repo.get_commits(since=since), key=lambda x: x.commit.author.date)
|
|
|
|
|
|
def process_one_commit(commit):
|
|
commit_statuses = get_filtered_statuses(commit)
|
|
|
|
# very dirty, but don't require additional dependencies
|
|
commit_modified = commit.commit.author.date + datetime.timedelta(hours=3)
|
|
commit_sha = commit.sha
|
|
checks_result = {}
|
|
for commit_status in commit_statuses:
|
|
state = commit_status.state
|
|
check_name = commit_status.context
|
|
checks_result[check_name] = state
|
|
|
|
return commit_sha, commit_modified, checks_result
|
|
|
|
|
|
if __name__ == "__main__":
|
|
three_days_ago = datetime.datetime.now() - datetime.timedelta(days=3)
|
|
parser = argparse.ArgumentParser("ClickHouse commits history parser")
|
|
parser.add_argument("--token", required=True)
|
|
parser.add_argument("--since", default=three_days_ago.strftime("%Y-%m-%d %H:%M:%S"))
|
|
parser.add_argument("--substr", default="Functional stateful")
|
|
|
|
args = parser.parse_args()
|
|
|
|
date_since = datetime.datetime.strptime(args.since, "%Y-%m-%d %H:%M:%S")
|
|
|
|
gh = Github(args.token)
|
|
repo = gh.get_repo("ClickHouse/ClickHouse")
|
|
commits = get_commits(repo, date_since)
|
|
|
|
longest_header = []
|
|
all_data = []
|
|
for num, commit in enumerate(commits):
|
|
commit_sha, commit_modified, check_results = process_one_commit(commit)
|
|
mapped_keys = [key for key in list(check_results.keys()) if args.substr in key]
|
|
if len(mapped_keys) > len(longest_header):
|
|
longest_header = mapped_keys
|
|
all_data.append((commit_modified, commit_sha, check_results))
|
|
if (num + 1) % 10 == 0:
|
|
print("Processed", num + 1, "commits")
|
|
|
|
longest_header = ["Date", "SHA"] + longest_header
|
|
|
|
result_data = []
|
|
for row in all_data:
|
|
current_result = [row[0].strftime("%Y-%m-%d %H:%M:%S"), row[1][0:7]]
|
|
for check_name in longest_header[2:]:
|
|
if check_name in row[2]:
|
|
check_result = row[2][check_name]
|
|
else:
|
|
check_result = "not run"
|
|
|
|
if sys.stdout.isatty():
|
|
current_result.append(COLORMAP[check_result])
|
|
else:
|
|
current_result.append(check_result)
|
|
result_data.append(current_result)
|
|
|
|
if sys.stdout.isatty():
|
|
longest_header = [colored(h, "white", attrs=["bold"]) for h in longest_header]
|
|
|
|
print(tabulate.tabulate(result_data, headers=longest_header, tablefmt="grid"))
|