gcs_download.py - mozsearch

Enable keyboard shortcuts

# Copyright 2022 The Chromium Authors

# Use of this source code is governed by a BSD-style license that can be

# found in the LICENSE file.

import os

import sys

import tarfile

import tempfile

sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__),

                                             'test')))

from gs_util_wrapper import run_gsutil

def DownloadAndUnpackFromCloudStorage(url, output_dir):

  """Fetches a tarball from GCS and uncompresses it to |output_dir|."""

  tmp_file = 'image.tgz'

  with tempfile.TemporaryDirectory() as tmp_d:

    tmp_file_location = os.path.join(tmp_d, tmp_file)

    run_gsutil(['cp', url, tmp_file_location])

    tarfile.open(name=tmp_file_location,

                 mode='r|gz').extractall(path=output_dir)