blob: c3b8b5ce2dee7fce14e70039680c219fdb41d1e0 [file] [log] [blame]
# Copyright 2020 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Download a file from CIPD.
This is used for downloading a tool executable managed in CIPD.
CIPD URI format is locally defined for this script.
chromiumos/infra/tclint/linux-amd64 is the path sent to cipd tool,
and ${version} is the version defined in CIPD.
import os
import urllib.parse
from chromite.lib import commandline
from chromite.lib import constants
from chromite.lib import cros_build_lib
def GetParser():
"""Creates the argparse parser."""
parser = commandline.ArgumentParser(description=__doc__)
parser.add_argument('uri', type='cipd',
help='CIPD URI of a file to download.')
parser.add_argument('output', type='path',
help='Location to store the file.')
return parser
def ParseCipdUri(uri):
o = urllib.parse.urlparse(uri)
if o.scheme != 'cipd':
raise ValueError('wrong scheme: ', o.scheme)
if ':' not in o.path:
raise ValueError('version not specified')
pkgpath, version = o.path.rsplit(':', 1)
return (o.netloc + pkgpath, version)
def main(argv):
parser = GetParser()
options = parser.parse_args(argv)
(pkgpath, version) = ParseCipdUri(options.uri)
[os.path.join(constants.DEPOT_TOOLS_DIR, 'cipd'), 'pkg-fetch',
'-out', options.output, '-version', version, '-verbose', pkgpath],
except cros_build_lib.RunCommandError as ex:
# Hide the stack trace using Die.
cros_build_lib.Die('%s', ex)