Alex Gaynor | 5951f46 | 2014-11-16 09:08:42 -0800 | [diff] [blame] | 1 | # This file is dual licensed under the terms of the Apache License, Version |
| 2 | # 2.0, and the BSD License. See the LICENSE file in the root of this repository |
| 3 | # for complete details. |
Terry Chia | a0a47cd | 2014-07-06 17:51:26 +0800 | [diff] [blame] | 4 | |
Alex Gaynor | 0e10f57 | 2014-01-06 13:17:31 -0800 | [diff] [blame] | 5 | from __future__ import absolute_import, division, print_function |
| 6 | |
Alex Gaynor | 8620159 | 2014-02-19 14:01:06 -0800 | [diff] [blame] | 7 | import getpass |
Alex Gaynor | 7a2b358 | 2014-02-20 07:48:46 -0800 | [diff] [blame] | 8 | import os |
Alex Gaynor | a1dff05 | 2014-02-19 16:44:06 -0800 | [diff] [blame] | 9 | import time |
Alex Gaynor | 8620159 | 2014-02-19 14:01:06 -0800 | [diff] [blame] | 10 | |
Alex Gaynor | 6bc3af7 | 2014-01-06 12:04:53 -0800 | [diff] [blame] | 11 | import invoke |
| 12 | |
Alex Gaynor | 8620159 | 2014-02-19 14:01:06 -0800 | [diff] [blame] | 13 | import requests |
| 14 | |
| 15 | |
Paul Kehrer | 47e5df9 | 2014-03-15 08:24:45 -0400 | [diff] [blame] | 16 | JENKINS_URL = "https://jenkins.cryptography.io/job/cryptography-wheel-builder" |
Alex Gaynor | a1dff05 | 2014-02-19 16:44:06 -0800 | [diff] [blame] | 17 | |
| 18 | |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 19 | def wait_for_build_completed(session): |
Paul Kehrer | d755e66 | 2014-12-24 08:07:56 -0600 | [diff] [blame] | 20 | # Wait 20 seconds before actually checking if the build is complete, to |
Alex Gaynor | d6b0598 | 2014-12-16 09:32:59 -0800 | [diff] [blame] | 21 | # ensure that it had time to really start. |
Paul Kehrer | d755e66 | 2014-12-24 08:07:56 -0600 | [diff] [blame] | 22 | time.sleep(20) |
Alex Gaynor | a1dff05 | 2014-02-19 16:44:06 -0800 | [diff] [blame] | 23 | while True: |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 24 | response = session.get( |
Alex Gaynor | a1dff05 | 2014-02-19 16:44:06 -0800 | [diff] [blame] | 25 | "{0}/lastBuild/api/json/".format(JENKINS_URL), |
| 26 | headers={ |
| 27 | "Accept": "application/json", |
| 28 | } |
| 29 | ) |
| 30 | response.raise_for_status() |
| 31 | if not response.json()["building"]: |
Alex Gaynor | 7a2b358 | 2014-02-20 07:48:46 -0800 | [diff] [blame] | 32 | assert response.json()["result"] == "SUCCESS" |
Alex Gaynor | a1dff05 | 2014-02-19 16:44:06 -0800 | [diff] [blame] | 33 | break |
| 34 | time.sleep(0.1) |
Alex Gaynor | 6bc3af7 | 2014-01-06 12:04:53 -0800 | [diff] [blame] | 35 | |
Alex Gaynor | c9e4c6a | 2014-02-19 14:29:37 -0800 | [diff] [blame] | 36 | |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 37 | def download_artifacts(session): |
| 38 | response = session.get( |
Alex Gaynor | 7a2b358 | 2014-02-20 07:48:46 -0800 | [diff] [blame] | 39 | "{0}/lastBuild/api/json/".format(JENKINS_URL), |
| 40 | headers={ |
| 41 | "Accept": "application/json" |
| 42 | } |
| 43 | ) |
| 44 | response.raise_for_status() |
| 45 | assert not response.json()["building"] |
| 46 | assert response.json()["result"] == "SUCCESS" |
Alex Gaynor | 6ab3f46 | 2014-02-20 09:07:53 -0800 | [diff] [blame] | 47 | |
| 48 | paths = [] |
| 49 | |
Paul Kehrer | ee1ca74 | 2015-05-28 23:23:26 -0500 | [diff] [blame] | 50 | last_build_number = response.json()["number"] |
Alex Gaynor | 7a2b358 | 2014-02-20 07:48:46 -0800 | [diff] [blame] | 51 | for run in response.json()["runs"]: |
Paul Kehrer | ee1ca74 | 2015-05-28 23:23:26 -0500 | [diff] [blame] | 52 | if run["number"] != last_build_number: |
| 53 | print( |
| 54 | "Skipping {0} as it is not from the latest build ({1})".format( |
| 55 | run["url"], last_build_number |
| 56 | ) |
| 57 | ) |
| 58 | continue |
| 59 | |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 60 | response = session.get( |
Alex Gaynor | 7a2b358 | 2014-02-20 07:48:46 -0800 | [diff] [blame] | 61 | run["url"] + "api/json/", |
| 62 | headers={ |
| 63 | "Accept": "application/json", |
| 64 | } |
| 65 | ) |
| 66 | response.raise_for_status() |
| 67 | for artifact in response.json()["artifacts"]: |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 68 | response = session.get( |
Paul Kehrer | 58f1101 | 2014-05-03 20:48:03 -0500 | [diff] [blame] | 69 | "{0}artifact/{1}".format(run["url"], artifact["relativePath"]) |
Alex Gaynor | 7a2b358 | 2014-02-20 07:48:46 -0800 | [diff] [blame] | 70 | ) |
| 71 | out_path = os.path.join( |
| 72 | os.path.dirname(__file__), |
| 73 | "dist", |
| 74 | artifact["fileName"], |
| 75 | ) |
| 76 | with open(out_path, "wb") as f: |
| 77 | f.write(response.content) |
Alex Gaynor | 6ab3f46 | 2014-02-20 09:07:53 -0800 | [diff] [blame] | 78 | paths.append(out_path) |
| 79 | return paths |
Alex Gaynor | 7a2b358 | 2014-02-20 07:48:46 -0800 | [diff] [blame] | 80 | |
| 81 | |
Alex Gaynor | 6bc3af7 | 2014-01-06 12:04:53 -0800 | [diff] [blame] | 82 | @invoke.task |
| 83 | def release(version): |
| 84 | """ |
| 85 | ``version`` should be a string like '0.4' or '1.0'. |
| 86 | """ |
Paul Kehrer | dd32e95 | 2014-07-09 21:14:24 -0500 | [diff] [blame] | 87 | invoke.run("git tag -s {0} -m '{0} release'".format(version)) |
Alex Gaynor | 6b1235a | 2014-01-06 15:28:59 -0800 | [diff] [blame] | 88 | invoke.run("git push --tags") |
Alex Gaynor | 6bc3af7 | 2014-01-06 12:04:53 -0800 | [diff] [blame] | 89 | |
Alex Gaynor | fea893c | 2014-01-07 11:06:51 -0800 | [diff] [blame] | 90 | invoke.run("python setup.py sdist") |
Alex Stapleton | 3888a84 | 2014-03-24 23:05:53 +0000 | [diff] [blame] | 91 | invoke.run("cd vectors/ && python setup.py sdist bdist_wheel") |
Alex Stapleton | a39a319 | 2014-03-14 20:03:12 +0000 | [diff] [blame] | 92 | |
| 93 | invoke.run( |
| 94 | "twine upload -s dist/cryptography-{0}* " |
Alex Stapleton | 3888a84 | 2014-03-24 23:05:53 +0000 | [diff] [blame] | 95 | "vectors/dist/cryptography_vectors-{0}*".format(version) |
Alex Stapleton | a39a319 | 2014-03-14 20:03:12 +0000 | [diff] [blame] | 96 | ) |
Alex Gaynor | 8620159 | 2014-02-19 14:01:06 -0800 | [diff] [blame] | 97 | |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 98 | session = requests.Session() |
| 99 | |
Alex Gaynor | b44158e | 2014-09-28 12:00:49 -0400 | [diff] [blame] | 100 | # This tells the CDN to delete the cached response for the URL. We do this |
| 101 | # so that the Jenkins builders will see the new sdist immediately when they |
| 102 | # go to build the wheels. |
Alex Gaynor | 0f74a2c | 2014-09-26 23:31:19 -0400 | [diff] [blame] | 103 | response = session.request( |
| 104 | "PURGE", "https://pypi.python.org/simple/cryptography/" |
| 105 | ) |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 106 | response.raise_for_status() |
| 107 | |
Paul Kehrer | d1d6f0f | 2014-06-20 14:32:02 -0600 | [diff] [blame] | 108 | username = getpass.getpass("Input the GitHub/Jenkins username: ") |
Alex Gaynor | 6629fa5 | 2014-02-20 16:10:53 -0800 | [diff] [blame] | 109 | token = getpass.getpass("Input the Jenkins token: ") |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 110 | response = session.post( |
Alex Gaynor | a1dff05 | 2014-02-19 16:44:06 -0800 | [diff] [blame] | 111 | "{0}/build".format(JENKINS_URL), |
Paul Kehrer | d1d6f0f | 2014-06-20 14:32:02 -0600 | [diff] [blame] | 112 | auth=requests.auth.HTTPBasicAuth( |
| 113 | username, token |
| 114 | ), |
Alex Gaynor | 8620159 | 2014-02-19 14:01:06 -0800 | [diff] [blame] | 115 | params={ |
Alex Gaynor | 8620159 | 2014-02-19 14:01:06 -0800 | [diff] [blame] | 116 | "cause": "Building wheels for {0}".format(version) |
| 117 | } |
| 118 | ) |
Alex Gaynor | c9e4c6a | 2014-02-19 14:29:37 -0800 | [diff] [blame] | 119 | response.raise_for_status() |
Alex Gaynor | 9ca7f03 | 2014-09-26 23:30:31 -0400 | [diff] [blame] | 120 | wait_for_build_completed(session) |
| 121 | paths = download_artifacts(session) |
Alex Gaynor | 6ab3f46 | 2014-02-20 09:07:53 -0800 | [diff] [blame] | 122 | invoke.run("twine upload {0}".format(" ".join(paths))) |