OLD | NEW |
(Empty) | |
| 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)) |
OLD | NEW |