mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-22 23:52:03 +00:00
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
import os
|
|
import logging
|
|
import ast
|
|
|
|
from report import create_test_html_report
|
|
|
|
def process_logs(s3_client, additional_logs, s3_path_prefix, test_results):
|
|
proccessed_logs = {}
|
|
# Firstly convert paths of logs from test_results to urls to s3.
|
|
for test_result in test_results:
|
|
if len(test_result) <= 3:
|
|
continue
|
|
|
|
# Convert from string repr of list to list.
|
|
test_log_paths = ast.literal_eval(test_result[3])
|
|
test_log_urls = []
|
|
for log_path in test_log_paths:
|
|
if log_path in proccessed_logs:
|
|
test_log_urls.append(proccessed_logs[log_path])
|
|
elif log_path:
|
|
url = s3_client.upload_test_report_to_s3(
|
|
log_path,
|
|
s3_path_prefix + "/" + os.path.basename(log_path))
|
|
test_log_urls.append(url)
|
|
proccessed_logs[log_path] = url
|
|
|
|
test_result[3] = test_log_urls
|
|
|
|
additional_urls = []
|
|
for log_path in additional_logs:
|
|
if log_path:
|
|
additional_urls.append(
|
|
s3_client.upload_test_report_to_s3(
|
|
log_path,
|
|
s3_path_prefix + "/" + os.path.basename(log_path)))
|
|
|
|
return additional_urls
|
|
|
|
def upload_results(s3_client, pr_number, commit_sha, test_results, additional_files, check_name, with_raw_logs=True):
|
|
s3_path_prefix = f"{pr_number}/{commit_sha}/" + check_name.lower().replace(' ', '_').replace('(', '_').replace(')', '_').replace(',', '_')
|
|
additional_urls = process_logs(s3_client, additional_files, s3_path_prefix, test_results)
|
|
|
|
branch_url = "https://github.com/ClickHouse/ClickHouse/commits/master"
|
|
branch_name = "master"
|
|
if pr_number != 0:
|
|
branch_name = f"PR #{pr_number}"
|
|
branch_url = f"https://github.com/ClickHouse/ClickHouse/pull/{pr_number}"
|
|
commit_url = f"https://github.com/ClickHouse/ClickHouse/commit/{commit_sha}"
|
|
|
|
task_url = f"https://github.com/ClickHouse/ClickHouse/actions/runs/{os.getenv('GITHUB_RUN_ID')}"
|
|
|
|
if additional_urls:
|
|
raw_log_url = additional_urls[0]
|
|
additional_urls.pop(0)
|
|
else:
|
|
raw_log_url = task_url
|
|
|
|
html_report = create_test_html_report(check_name, test_results, raw_log_url, task_url, branch_url, branch_name, commit_url, additional_urls, with_raw_logs)
|
|
with open('report.html', 'w', encoding='utf-8') as f:
|
|
f.write(html_report)
|
|
|
|
url = s3_client.upload_test_report_to_s3('report.html', s3_path_prefix + ".html")
|
|
logging.info("Search result in url %s", url)
|
|
return url
|