skia2/infra/bots/assets/cockroachdb/create.py
Kevin Lubick 3a7b02553b [infra] Add CIPD asset for mockery.
Of note, importing the common.py file first is important
if you want to use any modules in the parent directories.

This took a while to figure out, so I made sure to import
it by default and include a comment about this
"load bearing import". I am not sure how things worked
previously.

Change-Id: I908c136c547670592d5eadbc9ba2c918adcdab8f
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/341040
Commit-Queue: Kevin Lubick <kjlubick@google.com>
Reviewed-by: Eric Boren <borenet@google.com>
2020-12-11 13:17:08 +00:00

39 lines
962 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 common # fixes python import path
import argparse
import shutil
import subprocess
import utils
URL = "https://binaries.cockroachdb.com/cockroach-v20.1.3.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-v20.1.3.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()