1*c8dee2aaSAndroid Build Coastguard Worker#!/usr/bin/env python 2*c8dee2aaSAndroid Build Coastguard Worker# 3*c8dee2aaSAndroid Build Coastguard Worker# Copyright 2017 Google Inc. 4*c8dee2aaSAndroid Build Coastguard Worker# 5*c8dee2aaSAndroid Build Coastguard Worker# Use of this source code is governed by a BSD-style license that can be 6*c8dee2aaSAndroid Build Coastguard Worker# found in the LICENSE file. 7*c8dee2aaSAndroid Build Coastguard Worker 8*c8dee2aaSAndroid Build Coastguard Worker 9*c8dee2aaSAndroid Build Coastguard Worker"""Create the asset.""" 10*c8dee2aaSAndroid Build Coastguard Worker 11*c8dee2aaSAndroid Build Coastguard Worker 12*c8dee2aaSAndroid Build Coastguard Workerimport argparse 13*c8dee2aaSAndroid Build Coastguard Workerimport requests 14*c8dee2aaSAndroid Build Coastguard Workerimport subprocess 15*c8dee2aaSAndroid Build Coastguard Worker 16*c8dee2aaSAndroid Build Coastguard Worker 17*c8dee2aaSAndroid Build Coastguard WorkerVERSION = 'v1.25.4' 18*c8dee2aaSAndroid Build Coastguard WorkerDOWNLOAD_URL = 'https://dl.k8s.io/release/%s/bin/linux/amd64/kubectl' % VERSION 19*c8dee2aaSAndroid Build Coastguard WorkerSHA256_URL = 'https://dl.k8s.io/%s/bin/linux/amd64/kubectl.sha256' % VERSION 20*c8dee2aaSAndroid Build Coastguard Worker 21*c8dee2aaSAndroid Build Coastguard Worker 22*c8dee2aaSAndroid Build Coastguard Workerdef create_asset(target_dir): 23*c8dee2aaSAndroid Build Coastguard Worker """Create the asset.""" 24*c8dee2aaSAndroid Build Coastguard Worker subprocess.check_call(['curl', '-LO', DOWNLOAD_URL], cwd=target_dir) 25*c8dee2aaSAndroid Build Coastguard Worker sha256 = requests.get(SHA256_URL).text 26*c8dee2aaSAndroid Build Coastguard Worker p = subprocess.Popen( 27*c8dee2aaSAndroid Build Coastguard Worker ['sha256sum', '--check'], 28*c8dee2aaSAndroid Build Coastguard Worker cwd=target_dir, 29*c8dee2aaSAndroid Build Coastguard Worker stdout=subprocess.PIPE, 30*c8dee2aaSAndroid Build Coastguard Worker stderr=subprocess.STDOUT, 31*c8dee2aaSAndroid Build Coastguard Worker stdin=subprocess.PIPE) 32*c8dee2aaSAndroid Build Coastguard Worker output = p.communicate( 33*c8dee2aaSAndroid Build Coastguard Worker input=(sha256 + ' kubectl').encode('utf-8'))[0].decode('utf-8') 34*c8dee2aaSAndroid Build Coastguard Worker if 'OK' not in output: 35*c8dee2aaSAndroid Build Coastguard Worker raise ValueError('Got wrong checksum for kubectl: %s' % output) 36*c8dee2aaSAndroid Build Coastguard Worker subprocess.check_call(['chmod', 'a+x', 'kubectl'], cwd=target_dir) 37*c8dee2aaSAndroid Build Coastguard Worker 38*c8dee2aaSAndroid Build Coastguard Worker 39*c8dee2aaSAndroid Build Coastguard Workerdef main(): 40*c8dee2aaSAndroid Build Coastguard Worker parser = argparse.ArgumentParser() 41*c8dee2aaSAndroid Build Coastguard Worker parser.add_argument('--target_dir', '-t', required=True) 42*c8dee2aaSAndroid Build Coastguard Worker args = parser.parse_args() 43*c8dee2aaSAndroid Build Coastguard Worker create_asset(args.target_dir) 44*c8dee2aaSAndroid Build Coastguard Worker 45*c8dee2aaSAndroid Build Coastguard Worker 46*c8dee2aaSAndroid Build Coastguard Workerif __name__ == '__main__': 47*c8dee2aaSAndroid Build Coastguard Worker main() 48*c8dee2aaSAndroid Build Coastguard Worker 49