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