1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
import io
import os
import sys
import zipfile
import requests
from urllib3.util.retry import Retry
def get_response(session, url, token):
response = session.get(url, headers={"Authorization": "token " + token})
if response.status_code != 200:
raise ValueError("Got HTTP {} fetching {}: ".format(
response.status_code, url
))
return response
def main(platform, target):
if platform == "windows":
workflow = "build-windows-openssl.yml"
path = "C:/"
elif platform == "macos":
workflow = "build-macos-openssl.yml"
path = os.environ["HOME"]
else:
raise ValueError("Invalid platform")
session = requests.Session()
adapter = requests.adapters.HTTPAdapter(
max_retries=Retry()
)
session.mount("https://", adapter)
session.mount("http://", adapter)
token = os.environ["GITHUB_TOKEN"]
print("Looking for: {}".format(target))
runs_url = (
"https://api.github.com/repos/pyca/infra/actions/workflows/"
"{}/runs?branch=master&status=success".format(workflow)
)
response = get_response(session, runs_url, token).json()
artifacts_url = response["workflow_runs"][0]["artifacts_url"]
response = get_response(session, artifacts_url, token).json()
for artifact in response["artifacts"]:
if artifact["name"] == target:
print("Found artifact")
response = get_response(
session, artifact["archive_download_url"], token
)
zipfile.ZipFile(io.BytesIO(response.content)).extractall(
os.path.join(path, artifact["name"])
)
return
if __name__ == "__main__":
main(sys.argv[1], sys.argv[2])
|