zip.py - mozsearch

Enable keyboard shortcuts

#!/usr/bin/env python3

# Copyright 2014 The Chromium Authors. All rights reserved.

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

# found in the LICENSE file.

"""Archives a set of files."""

import argparse

import os

import sys

import zipfile

from util import build_utils

def main(args):

  args = build_utils.ExpandFileArgs(args)

  parser = argparse.ArgumentParser(args)

  parser.add_argument('--input-files', help='GN-list of files to zip.')

  parser.add_argument(

      '--input-files-base-dir',

      help='Paths in the archive will be relative to this directory')

  parser.add_argument('--input-zips', help='GN-list of zips to merge.')

  parser.add_argument(

      '--input-zips-excluded-globs',

      help='GN-list of globs for paths to exclude.')

  parser.add_argument('--output', required=True, help='Path to output archive.')

  compress_group = parser.add_mutually_exclusive_group()

  compress_group.add_argument(

      '--compress', action='store_true', help='Compress entries')

  compress_group.add_argument(

      '--no-compress',

      action='store_false',

      dest='compress',

      help='Do not compress entries')

  build_utils.AddDepfileOption(parser)

  options = parser.parse_args(args)

  with build_utils.AtomicOutput(options.output) as f:

    with zipfile.ZipFile(f.name, 'w') as out_zip:

      depfile_deps = None

      if options.input_files:

        files = build_utils.ParseGnList(options.input_files)

        build_utils.DoZip(

            files,

            out_zip,

            base_dir=options.input_files_base_dir,

            compress_fn=lambda _: options.compress)

      if options.input_zips:

        files = build_utils.ParseGnList(options.input_zips)

        depfile_deps = files

        path_transform = None

        if options.input_zips_excluded_globs:

          globs = build_utils.ParseGnList(options.input_zips_excluded_globs)

          path_transform = (

              lambda p: None if build_utils.MatchesGlob(p, globs) else p)

        build_utils.MergeZips(

            out_zip,

            files,

            path_transform=path_transform,

            compress=options.compress)

  # Depfile used only by dist_jar().

  if options.depfile:

    build_utils.WriteDepfile(options.depfile,

                             options.output,

                             inputs=depfile_deps)

if __name__ == '__main__':

  main(sys.argv[1:])