#!/usr/bin/env python3 import os import subprocess import logging from github import Github from commit_status_helper import post_commit_status from docker_pull_helper import get_image_with_version from env_helper import ( IMAGES_PATH, REPO_COPY, S3_DOWNLOAD, S3_TEST_REPORTS_BUCKET, TEMP_PATH, ) from get_robot_token import get_best_robot_token from pr_info import PRInfo from report import TestResult from s3_helper import S3Helper from stopwatch import Stopwatch from tee_popen import TeePopen from upload_result_helper import upload_results NAME = "Woboq Build" def get_run_command(repo_path, output_path, image): cmd = ( "docker run " + f"--volume={repo_path}:/repo_folder " f"--volume={output_path}:/test_output " f"-e 'DATA={S3_DOWNLOAD}/{S3_TEST_REPORTS_BUCKET}/codebrowser/data' {image}" ) return cmd if __name__ == "__main__": logging.basicConfig(level=logging.INFO) stopwatch = Stopwatch() gh = Github(get_best_robot_token(), per_page=100) pr_info = PRInfo() if not os.path.exists(TEMP_PATH): os.makedirs(TEMP_PATH) docker_image = get_image_with_version(IMAGES_PATH, "clickhouse/codebrowser") s3_helper = S3Helper() result_path = os.path.join(TEMP_PATH, "result_path") if not os.path.exists(result_path): os.makedirs(result_path) run_command = get_run_command(REPO_COPY, result_path, docker_image) logging.info("Going to run codebrowser: %s", run_command) run_log_path = os.path.join(TEMP_PATH, "run.log") with TeePopen(run_command, run_log_path) as process: retcode = process.wait() if retcode == 0: logging.info("Run successfully") else: logging.info("Run failed") subprocess.check_call(f"sudo chown -R ubuntu:ubuntu {TEMP_PATH}", shell=True) report_path = os.path.join(result_path, "html_report") logging.info("Report path %s", report_path) s3_path_prefix = "codebrowser" html_urls = s3_helper.fast_parallel_upload_dir( report_path, s3_path_prefix, "clickhouse-test-reports" ) index_html = ( '' "HTML report" ) test_result = TestResult(index_html, "Look at the report") report_url = upload_results(s3_helper, 0, pr_info.sha, [test_result], [], NAME) print(f"::notice ::Report url: {report_url}") post_commit_status(gh, pr_info.sha, NAME, "Report built", "success", report_url)