| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2017, the Dart project authors. Please see the AUTHORS file | 2 # Copyright (c) 2017, the Dart project authors. Please see the AUTHORS file |
| 3 # for details. All rights reserved. Use of this source code is governed by a | 3 # for details. All rights reserved. Use of this source code is governed by a |
| 4 # BSD-style license that can be found in the LICENSE file. | 4 # BSD-style license that can be found in the LICENSE file. |
| 5 | 5 |
| 6 import argparse | 6 import argparse |
| 7 import os | 7 import os |
| 8 import re | 8 import re |
| 9 import shutil | 9 import shutil |
| 10 import sys | 10 import sys |
| 11 | 11 |
| 12 def ParseArgs(args): | 12 def ParseArgs(args): |
| 13 args = args[1:] | 13 args = args[1:] |
| 14 parser = argparse.ArgumentParser( | 14 parser = argparse.ArgumentParser( |
| 15 description='A script to copy a file tree somewhere') | 15 description='A script to copy a file tree somewhere') |
| 16 | 16 |
| 17 parser.add_argument('--dry-run', '-d', |
| 18 dest='dryrun', |
| 19 default=False, |
| 20 action='store_true', |
| 21 help='Print the paths of the source files, but do not copy anything.') |
| 17 parser.add_argument('--exclude_patterns', '-e', | 22 parser.add_argument('--exclude_patterns', '-e', |
| 18 type=str, | 23 type=str, |
| 19 help='Patterns to exclude [passed to shutil.copytree]') | 24 help='Patterns to exclude [passed to shutil.copytree]') |
| 20 parser.add_argument('--from', '-f', | 25 parser.add_argument('--from', '-f', |
| 21 dest="copy_from", | 26 dest="copy_from", |
| 22 type=str, | 27 type=str, |
| 23 required=True, | 28 required=True, |
| 24 help='Source tree root') | 29 help='Source directory') |
| 25 parser.add_argument('--to', '-t', | 30 parser.add_argument('--to', '-t', |
| 26 type=str, | 31 type=str, |
| 27 required=True, | 32 required=True, |
| 28 help='Destination') | 33 help='Destination directory') |
| 29 | 34 |
| 30 return parser.parse_args(args) | 35 return parser.parse_args(args) |
| 31 | 36 |
| 32 | 37 |
| 33 def ValidateArgs(args): | 38 def ValidateArgs(args): |
| 34 if not os.path.isdir(args.copy_from): | 39 if not os.path.isdir(args.copy_from): |
| 35 print "--from argument must refer to a directory" | 40 print "--from argument must refer to a directory" |
| 36 return False | 41 return False |
| 37 return True | 42 return True |
| 38 | 43 |
| 39 | 44 |
| 45 def CopyTree(src, dst, dryrun=False, symlinks=False, ignore=None): |
| 46 names = os.listdir(src) |
| 47 if ignore is not None: |
| 48 ignored_names = ignore(src, names) |
| 49 else: |
| 50 ignored_names = set() |
| 51 |
| 52 if not dryrun: |
| 53 os.makedirs(dst) |
| 54 errors = [] |
| 55 for name in names: |
| 56 if name in ignored_names: |
| 57 continue |
| 58 srcname = os.path.join(src, name) |
| 59 dstname = os.path.join(dst, name) |
| 60 try: |
| 61 if os.path.isdir(srcname): |
| 62 CopyTree(srcname, dstname, dryrun, symlinks, ignore) |
| 63 else: |
| 64 if dryrun: |
| 65 print srcname |
| 66 else: |
| 67 shutil.copy(srcname, dstname) |
| 68 except (IOError, os.error) as why: |
| 69 errors.append((srcname, dstname, str(why))) |
| 70 # catch the Error from the recursive CopyTree so that we can |
| 71 # continue with other files |
| 72 except Error as err: |
| 73 errors.extend(err.args[0]) |
| 74 try: |
| 75 if not dryrun: |
| 76 shutil.copystat(src, dst) |
| 77 except WindowsError: |
| 78 # can't copy file access times on Windows |
| 79 pass |
| 80 except OSError as why: |
| 81 errors.extend((src, dst, str(why))) |
| 82 if errors: |
| 83 raise Error(errors) |
| 84 |
| 85 |
| 40 def Main(argv): | 86 def Main(argv): |
| 41 args = ParseArgs(argv) | 87 args = ParseArgs(argv) |
| 42 if not ValidateArgs(args): | 88 if not ValidateArgs(args): |
| 43 return -1 | 89 return -1 |
| 90 |
| 44 if os.path.exists(args.to): | 91 if os.path.exists(args.to): |
| 45 shutil.rmtree(args.to) | 92 shutil.rmtree(args.to) |
| 46 if args.exclude_patterns == None: | 93 if args.exclude_patterns == None: |
| 47 shutil.copytree(args.copy_from, args.to) | 94 CopyTree(args.copy_from, args.to, dryrun=args.dryrun) |
| 48 else: | 95 else: |
| 49 patterns = args.exclude_patterns.split(',') | 96 patterns = args.exclude_patterns.split(',') |
| 50 shutil.copytree(args.copy_from, args.to, | 97 CopyTree(args.copy_from, args.to, dryrun=args.dryrun, |
| 51 ignore=shutil.ignore_patterns(tuple(patterns))) | 98 ignore=shutil.ignore_patterns(*patterns)) |
| 52 return 0 | 99 return 0 |
| 53 | 100 |
| 54 | 101 |
| 55 if __name__ == '__main__': | 102 if __name__ == '__main__': |
| 56 sys.exit(Main(sys.argv)) | 103 sys.exit(Main(sys.argv)) |
| OLD | NEW |