ab1b8d5ae2
Previous approach had drawbacks: - every step had to implement exporting of results in fixed format - if step failed then failure will not be detected Now report step will fetch results directly from Buildkite. Agents have to be updated to have BUILDKITE_API_TOKEN env.
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
import logging
|
|
import os
|
|
import re
|
|
import subprocess
|
|
from typing import Optional
|
|
import requests
|
|
|
|
|
|
def upload_file(base_dir: str, file: str):
|
|
"""
|
|
Uploads artifact to buildkite and returns URL to it
|
|
"""
|
|
r = subprocess.run(f'buildkite-agent artifact upload "{file}"', shell=True, capture_output=True, cwd=base_dir)
|
|
logging.debug(f'upload-artifact {r}')
|
|
match = re.search('Uploading artifact ([^ ]*) ', r.stderr.decode())
|
|
logging.debug(f'match {match}')
|
|
if match:
|
|
url = f'https://buildkite.com/organizations/llvm-project/pipelines/premerge-checks/builds/{os.getenv("BUILDKITE_BUILD_NUMBER")}/jobs/{os.getenv("BUILDKITE_JOB_ID")}/artifacts/{match.group(1)}'
|
|
logging.info(f'uploaded {file} to {url}')
|
|
return url
|
|
else:
|
|
logging.warning(f'could not find artifact {base_dir}/{file}')
|
|
return None
|
|
|
|
|
|
class BuildkiteApi:
|
|
def __init__(self, token: str, organization: str):
|
|
self.token = token
|
|
self.organization = organization
|
|
|
|
def get_build(self, pipeline: str, build_number: str):
|
|
authorization = f'Bearer {self.token}'
|
|
# https://buildkite.com/docs/apis/rest-api/builds#get-a-build
|
|
url = f'https://api.buildkite.com/v2/organizations/{self.organization}/pipelines/{pipeline}/builds/{build_number}'
|
|
response = requests.get(url, headers={'Authorization': authorization})
|
|
if response.status_code != 200:
|
|
raise Exception(f'Builkite responded with non-OK status: {re.status_code}')
|
|
return response.json()
|
|
|
|
|
|
def format_url(url: str, name: Optional[str] = None):
|
|
if name is None:
|
|
name = url
|
|
return f"\033]1339;url='{url}';content='{name}'\a\n"
|