|
#!/usr/bin/env python
|
|
|
|
# Upload a file to a GitHub draft release. --id and --file are required.
|
|
# Optimized for CircleCI
|
|
|
|
import json
|
|
import os
|
|
import re
|
|
import argparse
|
|
import mimetypes
|
|
import httplib
|
|
from base64 import b64encode
|
|
|
|
|
|
def request(baseurl, path, mimetype, mimeencoding, data):
|
|
user_and_pass = b64encode(b"{0}:{1}".format(os.environ['GITHUB_USERNAME'], os.environ['GITHUB_TOKEN'])).decode("ascii")
|
|
|
|
headers = {
|
|
'User-Agent': 'tenderbot',
|
|
'Accept': 'application/vnd.github.v3.raw+json',
|
|
'Authorization': 'Basic %s' % user_and_pass,
|
|
'Content-Type': mimetype,
|
|
'Content-Encoding': mimeencoding
|
|
}
|
|
|
|
conn = httplib.HTTPSConnection(baseurl, timeout=5)
|
|
conn.request('POST', path, data, headers)
|
|
response = conn.getresponse()
|
|
if response.status < 200 or response.status > 299:
|
|
print(response)
|
|
conn.close()
|
|
raise IOError(response.reason)
|
|
responsedata = response.read()
|
|
conn.close()
|
|
return json.loads(responsedata)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--id", help="GitHub release ID", required=True, type=int)
|
|
parser.add_argument("--file", default="/tmp/workspace/tendermint_{0}_{1}_{2}.zip".format(os.environ.get('CIRCLE_TAG'),os.environ.get('GOOS'),os.environ.get('GOARCH')), help="File to upload")
|
|
parser.add_argument("--return-id-only", help="Return only the release ID after upload to GitHub.", action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
if not os.environ.has_key('GITHUB_USERNAME'):
|
|
raise parser.error('GITHUB_USERNAME not set.')
|
|
|
|
if not os.environ.has_key('GITHUB_TOKEN'):
|
|
raise parser.error('GITHUB_TOKEN not set.')
|
|
|
|
mimetypes.init()
|
|
filename = os.path.basename(args.file)
|
|
mimetype,mimeencoding = mimetypes.guess_type(filename, strict=False)
|
|
if mimetype is None:
|
|
mimetype = 'application/zip'
|
|
if mimeencoding is None:
|
|
mimeencoding = 'utf8'
|
|
|
|
with open(args.file,'rb') as f:
|
|
asset = f.read()
|
|
|
|
result = request('uploads.github.com', '/repos/tendermint/tendermint/releases/{0}/assets?name={1}'.format(args.id, filename), mimetype, mimeencoding, asset)
|
|
|
|
if args.return_id_only:
|
|
print(result['id'])
|
|
else:
|
|
print(result['browser_download_url'])
|
|
|