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 cockroachDB asset.""" 10*c8dee2aaSAndroid Build Coastguard Worker 11*c8dee2aaSAndroid Build Coastguard Worker 12*c8dee2aaSAndroid Build Coastguard Workerimport argparse 13*c8dee2aaSAndroid Build Coastguard Workerimport os 14*c8dee2aaSAndroid Build Coastguard Workerimport shutil 15*c8dee2aaSAndroid Build Coastguard Workerimport subprocess 16*c8dee2aaSAndroid Build Coastguard Workerimport sys 17*c8dee2aaSAndroid Build Coastguard Worker 18*c8dee2aaSAndroid Build Coastguard WorkerFILE_DIR = os.path.dirname(os.path.abspath(__file__)) 19*c8dee2aaSAndroid Build Coastguard WorkerINFRA_BOTS_DIR = os.path.realpath(os.path.join(FILE_DIR, os.pardir, os.pardir)) 20*c8dee2aaSAndroid Build Coastguard Workersys.path.insert(0, INFRA_BOTS_DIR) 21*c8dee2aaSAndroid Build Coastguard Workerimport utils 22*c8dee2aaSAndroid Build Coastguard Worker 23*c8dee2aaSAndroid Build Coastguard Worker 24*c8dee2aaSAndroid Build Coastguard WorkerURL = "https://binaries.cockroachdb.com/cockroach-v20.2.8.linux-amd64.tgz" 25*c8dee2aaSAndroid Build Coastguard Worker 26*c8dee2aaSAndroid Build Coastguard Workerdef create_asset(target_dir): 27*c8dee2aaSAndroid Build Coastguard Worker """Create the asset.""" 28*c8dee2aaSAndroid Build Coastguard Worker with utils.tmp_dir(): 29*c8dee2aaSAndroid Build Coastguard Worker p1 = subprocess.Popen(["curl", URL], stdout=subprocess.PIPE) 30*c8dee2aaSAndroid Build Coastguard Worker p2 = subprocess.Popen(["tar", "-xzf" "-"], stdin=p1.stdout) 31*c8dee2aaSAndroid Build Coastguard Worker p1.stdout.close() # Allow p1 to receive a SIGPIPE if p2 exits. 32*c8dee2aaSAndroid Build Coastguard Worker _,_ = p2.communicate() 33*c8dee2aaSAndroid Build Coastguard Worker shutil.move('./cockroach-v20.2.8.linux-amd64/cockroach', target_dir) 34*c8dee2aaSAndroid Build Coastguard Worker 35*c8dee2aaSAndroid Build Coastguard Worker 36*c8dee2aaSAndroid Build Coastguard Workerdef main(): 37*c8dee2aaSAndroid Build Coastguard Worker parser = argparse.ArgumentParser() 38*c8dee2aaSAndroid Build Coastguard Worker parser.add_argument('--target_dir', '-t', required=True) 39*c8dee2aaSAndroid Build Coastguard Worker args = parser.parse_args() 40*c8dee2aaSAndroid Build Coastguard Worker create_asset(args.target_dir) 41*c8dee2aaSAndroid Build Coastguard Worker 42*c8dee2aaSAndroid Build Coastguard Worker 43*c8dee2aaSAndroid Build Coastguard Workerif __name__ == '__main__': 44*c8dee2aaSAndroid Build Coastguard Worker main() 45