skia2/infra/bots/assets/cockroachdb/create.py
Joe Gregorio 6dd3b73f1b [assets] Add CockroackDB as an asset for testing.
Change-Id: I3202bdedebb0f33968a739dd25f6dbcd7d5b4429
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/272997
Reviewed-by: Eric Boren <borenet@google.com>
Commit-Queue: Joe Gregorio <jcgregorio@google.com>
2020-02-24 17:28:15 +00:00

39 lines
922 B
Python
Executable File

#!/usr/bin/env python
#
# Copyright 2017 Google Inc.
#
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Create the cockroachDB asset."""
import argparse
import shutil
import subprocess
import utils
URL = "https://binaries.cockroachdb.com/cockroach-v19.2.4.linux-amd64.tgz"
def create_asset(target_dir):
"""Create the asset."""
with utils.tmp_dir():
p1 = subprocess.Popen(["curl", URL], stdout=subprocess.PIPE)
p2 = subprocess.Popen(["tar", "-xzf" "-"], stdin=p1.stdout)
p1.stdout.close() # Allow p1 to receive a SIGPIPE if p2 exits.
_,_ = p2.communicate()
shutil.move('./cockroach-v19.2.4.linux-amd64/cockroach', target_dir)
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--target_dir', '-t', required=True)
args = parser.parse_args()
create_asset(args.target_dir)
if __name__ == '__main__':
main()