File: upload_appveyor_builds.py

package info (click to toggle)
libsass-python 0.17.0-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 492 kB
  • sloc: python: 3,080; cpp: 622; makefile: 175
file content (145 lines) | stat: -rwxr-xr-x 4,135 bytes parent folder | download
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#!/usr/bin/env python3
# TODO: Upload to GitHub releases
# TODO: .pypirc configuration
import argparse
import json
import os
import os.path
import shutil
import subprocess
from urllib.parse import urljoin
from urllib.request import urlopen

from twine.commands import upload


APPVEYOR_API_BASE_URL = 'https://ci.appveyor.com/api/'
APPVEYOR_API_PROJECT_URL = urljoin(
    APPVEYOR_API_BASE_URL,
    'projects/asottile/libsass-python/',
)
APPVEYOR_API_BUILDS_URL = urljoin(
    APPVEYOR_API_PROJECT_URL,
    'history?recordsNumber=50&branch=master',
)
APPVEYOR_API_JOBS_URL = urljoin(
    APPVEYOR_API_PROJECT_URL,
    'build/',
)
APPVEYOR_API_JOB_URL = urljoin(APPVEYOR_API_BASE_URL, 'buildjobs/')


def ci_builds():
    response = urlopen(APPVEYOR_API_BUILDS_URL)
    projects = json.load(response)
    response.close()
    return projects['builds']


def ci_tag_build(tag):
    builds = ci_builds()
    commit_id = git_tags().get(tag)
    for build in builds:
        if build['status'] != 'success':
            continue
        if build['commitId'] == commit_id:
            return build
        elif build['isTag'] and build['tag'] == tag:
            return build


def git_tags():
    try:
        tags = subprocess.check_output(['git', 'tag'])
    except subprocess.CalledProcessError:
        return {}

    def read(tag):
        command = ['git', 'rev-list', tag]
        p = subprocess.Popen(command, stdout=subprocess.PIPE)
        try:
            firstline = p.stdout.readline()
        finally:
            p.terminate()
        return firstline.decode().strip()
    return {tag: read(tag) for tag in tags.decode().split()}


def ci_jobs(build):
    url = urljoin(APPVEYOR_API_JOBS_URL, build['version'])
    response = urlopen(url)
    build = json.load(response)
    response.close()
    return build['build']['jobs']


def ci_artifacts(job):
    url = urljoin(
        urljoin(APPVEYOR_API_JOB_URL, job['jobId'] + '/'),
        'artifacts/',
    )
    response = urlopen(url)
    files = json.load(response)
    response.close()
    for file_ in files:
        file_['url'] = urljoin(url, file_['fileName'])
    return files


def download_artifact(artifact, target_dir, overwrite=False):
    print('Downloading {}...'.format(artifact['fileName']))
    response = urlopen(artifact['url'])
    filename = os.path.basename(artifact['fileName'])
    target_path = os.path.join(target_dir, filename)
    if os.path.isfile(target_path) and \
       os.path.getsize(target_path) == artifact['size']:
        if overwrite:
            print(artifact['fileName'], ' already exists; overwrite...')
        else:
            print(artifact['fileName'], ' already exists; skip...')
            return target_path
    with open(target_path, 'wb') as f:
        shutil.copyfileobj(response, f)
        assert f.tell() == artifact['size']
    response.close()
    return target_path


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--overwrite', action='store_true', default=False,
        help='Overwrite files if already exist',
    )
    parser.add_argument(
        '--dist-dir', default='./dist/',
        help='The temporary directory to download artifacts',
    )
    parser.add_argument(
        'tag',
        help=(
            'Git tag of the version to upload.  If it has a leading slash, '
            'it means AppVeyor build number rather than Git tag.'
        ),
    )
    args = parser.parse_args()
    if args.tag.startswith('/'):
        build = {'version': args.tag.lstrip('/')}
    else:
        build = ci_tag_build(args.tag)
    jobs = ci_jobs(build)
    if not os.path.isdir(args.dist_dir):
        print(args.dist_dir, 'does not exist yet; creating a new directory...')
        os.makedirs(args.dist_dir)
    dists = []
    for job in jobs:
        artifacts = ci_artifacts(job)
        for artifact in artifacts:
            dist = download_artifact(artifact, args.dist_dir, args.overwrite)
            dists.append(dist)
    print('Uploading {} file(s)...'.format(len(dists)))
    upload.main(('-r', 'pypi') + tuple(dists))


if __name__ == '__main__':
    main()