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 |