| #!/usr/bin/env python3 |
| # Copyright 2012 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Copies test data files or directories into a given output directory.""" |
| |
| |
| import optparse |
| import os |
| import shutil |
| import sys |
| |
| classWrongNumberOfArgumentsException(Exception): |
| pass |
| |
| defEscapePath(path): |
| """Returns a path with spaces escaped.""" |
| return path.replace(" ","\\ ") |
| |
| defListFilesForPath(path): |
| """Returns a list of all the files under a given path.""" |
| output=[] |
| # Ignore revision control metadata directories. |
| if(os.path.basename(path).startswith('.git')or |
| os.path.basename(path).startswith('.svn')): |
| return output |
| |
| # Files get returned without modification. |
| ifnot os.path.isdir(path): |
| output.append(path) |
| return output |
| |
| # Directories get recursively expanded. |
| contents= os.listdir(path) |
| for itemin contents: |
| full_path= os.path.join(path, item) |
| output.extend(ListFilesForPath(full_path)) |
| return output |
| |
| defCalcInputs(inputs): |
| """Computes the full list of input files for a set of command-line arguments. |
| """ |
| # |inputs| is a list of paths, which may be directories. |
| output=[] |
| for inputin inputs: |
| output.extend(ListFilesForPath(input)) |
| return output |
| |
| defCopyFiles(relative_filenames, output_basedir): |
| """Copies files to the given output directory.""" |
| for filein relative_filenames: |
| relative_dirname= os.path.dirname(file) |
| output_dir= os.path.join(output_basedir, relative_dirname) |
| output_filename= os.path.join(output_basedir, file) |
| |
| # In cases where a directory has turned into a file or vice versa, delete it |
| # before copying it below. |
| if os.path.exists(output_dir)andnot os.path.isdir(output_dir): |
| os.remove(output_dir) |
| if os.path.exists(output_filename)and os.path.isdir(output_filename): |
| shutil.rmtree(output_filename) |
| |
| ifnot os.path.exists(output_dir): |
| os.makedirs(output_dir) |
| shutil.copy(file, output_filename) |
| |
| defDoMain(argv): |
| parser= optparse.OptionParser() |
| usage='Usage: %prog -o <output_dir> [--inputs] [--outputs] <input_files>' |
| parser.set_usage(usage) |
| parser.add_option('-o', dest='output_dir') |
| parser.add_option('--inputs', action='store_true', dest='list_inputs') |
| parser.add_option('--outputs', action='store_true', dest='list_outputs') |
| options, arglist= parser.parse_args(argv) |
| |
| if len(arglist)==0: |
| raiseWrongNumberOfArgumentsException('<input_files> required.') |
| |
| files_to_copy=CalcInputs(arglist) |
| escaped_files=[EscapePath(x)for xinCalcInputs(arglist)] |
| if options.list_inputs: |
| return'\n'.join(escaped_files) |
| |
| ifnot options.output_dir: |
| raiseWrongNumberOfArgumentsException('-o required.') |
| |
| if options.list_outputs: |
| outputs=[os.path.join(options.output_dir, x)for xin escaped_files] |
| return'\n'.join(outputs) |
| |
| CopyFiles(files_to_copy, options.output_dir) |
| return |
| |
| def main(argv): |
| try: |
| result=DoMain(argv[1:]) |
| exceptWrongNumberOfArgumentsExceptionas e: |
| print(e, file=sys.stderr) |
| return1 |
| if result: |
| print(result) |
| return0 |
| |
| if __name__=='__main__': |
| sys.exit(main(sys.argv)) |
