build/copy_test_data_ios.py

   1 #!/usr/bin/env python3
   2 # Copyright 2012 The Chromium Authors
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Copies test data files or directories into a given output directory."""
   7
   8
   9 import optparse
  10 import os
  11 import shutil
  12 import sys
  13
  14 class WrongNumberOfArgumentsException(Exception):
  15   pass
  16
  17 def EscapePath(path):
  18   """Returns a path with spaces escaped."""
  19   return path.replace(" ", "\\ ")
  20
  21 def ListFilesForPath(path):
  22   """Returns a list of all the files under a given path."""
  23   output = []
  24   # Ignore revision control metadata directories.
  25   if (os.path.basename(path).startswith('.git') or
  26       os.path.basename(path).startswith('.svn')):
  27     return output
  28
  29   # Files get returned without modification.
  30   if not os.path.isdir(path):
  31     output.append(path)
  32     return output
  33
  34   # Directories get recursively expanded.
  35   contents = os.listdir(path)
  36   for item in contents:
  37     full_path = os.path.join(path, item)
  38     output.extend(ListFilesForPath(full_path))
  39   return output
  40
  41 def CalcInputs(inputs):
  42   """Computes the full list of input files for a set of command-line arguments.
  43   """
  44   # |inputs| is a list of paths, which may be directories.
  45   output = []
  46   for input in inputs:
  47     output.extend(ListFilesForPath(input))
  48   return output
  49
  50 def CopyFiles(relative_filenames, output_basedir):
  51   """Copies files to the given output directory."""
  52   for file in relative_filenames:
  53     relative_dirname = os.path.dirname(file)
  54     output_dir = os.path.join(output_basedir, relative_dirname)
  55     output_filename = os.path.join(output_basedir, file)
  56
  57     # In cases where a directory has turned into a file or vice versa, delete it
  58     # before copying it below.
  59     if os.path.exists(output_dir) and not os.path.isdir(output_dir):
  60       os.remove(output_dir)
  61     if os.path.exists(output_filename) and os.path.isdir(output_filename):
  62       shutil.rmtree(output_filename)
  63
  64     if not os.path.exists(output_dir):
  65       os.makedirs(output_dir)
  66     shutil.copy(file, output_filename)
  67
  68 def DoMain(argv):
  69   parser = optparse.OptionParser()
  70   usage = 'Usage: %prog -o <output_dir> [--inputs] [--outputs] <input_files>'
  71   parser.set_usage(usage)
  72   parser.add_option('-o', dest='output_dir')
  73   parser.add_option('--inputs', action='store_true', dest='list_inputs')
  74   parser.add_option('--outputs', action='store_true', dest='list_outputs')
  75   options, arglist = parser.parse_args(argv)
  76
  77   if len(arglist) == 0:
  78     raise WrongNumberOfArgumentsException('<input_files> required.')
  79
  80   files_to_copy = CalcInputs(arglist)
  81   escaped_files = [EscapePath(x) for x in CalcInputs(arglist)]
  82   if options.list_inputs:
  83     return '\n'.join(escaped_files)
  84
  85   if not options.output_dir:
  86     raise WrongNumberOfArgumentsException('-o required.')
  87
  88   if options.list_outputs:
  89     outputs = [os.path.join(options.output_dir, x) for x in escaped_files]
  90     return '\n'.join(outputs)
  91
  92   CopyFiles(files_to_copy, options.output_dir)
  93   return
  94
  95 def main(argv):
  96   try:
  97     result = DoMain(argv[1:])
  98   except WrongNumberOfArgumentsException as e:
  99     print(e, file=sys.stderr)
 100     return 1
 101   if result:
 102     print(result)
 103   return 0
 104
 105 if __name__ == '__main__':
 106   sys.exit(main(sys.argv))