1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
|
#!/usr/bin/env python3
#
# Copyright 2014 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Copies files to a directory."""
import argparse
import filecmp
import itertools
import os
import shutil
import sys
from util import build_utils
import action_helpers # build_utils adds //build to sys.path.
def _get_all_files(base):
"""Returns a list of all the files in |base|. Each entry is relative to the
last path entry of |base|."""
result = []
dirname = os.path.dirname(base)
for root, _, files in os.walk(base):
result.extend([os.path.join(root[len(dirname):], f) for f in files])
return result
def CopyFile(f, dest, deps):
"""Copy file or directory and update deps."""
if os.path.isdir(f):
shutil.copytree(f, os.path.join(dest, os.path.basename(f)))
deps.extend(_get_all_files(f))
else:
if os.path.isfile(os.path.join(dest, os.path.basename(f))):
dest = os.path.join(dest, os.path.basename(f))
deps.append(f)
if os.path.isfile(dest):
if filecmp.cmp(dest, f, shallow=False):
return
# The shutil.copy() below would fail if the file does not have write
# permissions. Deleting the file has similar costs to modifying the
# permissions.
os.unlink(dest)
shutil.copy(f, dest)
def DoCopy(options, deps):
"""Copy files or directories given in options.files and update deps."""
files = list(
itertools.chain.from_iterable(
action_helpers.parse_gn_list(f) for f in options.files))
for f in files:
if os.path.isdir(f) and not options.clear:
print('To avoid stale files you must use --clear when copying '
'directories')
sys.exit(-1)
CopyFile(f, options.dest, deps)
def DoRenaming(options, deps):
"""Copy and rename files given in options.renaming_sources and update deps."""
src_files = list(
itertools.chain.from_iterable(
action_helpers.parse_gn_list(f) for f in options.renaming_sources))
dest_files = list(
itertools.chain.from_iterable(
action_helpers.parse_gn_list(f)
for f in options.renaming_destinations))
if len(src_files) != len(dest_files):
print('Renaming source and destination files not match.')
sys.exit(-1)
for src, dest in zip(src_files, dest_files):
if os.path.isdir(src):
print('renaming diretory is not supported.')
sys.exit(-1)
else:
CopyFile(src, os.path.join(options.dest, dest), deps)
def main(args):
args = build_utils.ExpandFileArgs(args)
parser = argparse.ArgumentParser()
action_helpers.add_depfile_arg(parser)
parser.add_argument('--dest', help='Directory to copy files to.')
parser.add_argument('--files', action='append', help='List of files to copy.')
parser.add_argument(
'--clear',
action='store_true',
help='If set, the destination directory will be deleted '
'before copying files to it. This is highly recommended to '
'ensure that no stale files are left in the directory.')
parser.add_argument('--stamp', help='Path to touch on success.')
parser.add_argument('--renaming-sources',
action='append',
help='List of files need to be renamed while being '
'copied to dest directory')
parser.add_argument('--renaming-destinations',
action='append',
help='List of destination file name without path, the '
'number of elements must match rename-sources.')
options = parser.parse_args(args)
if options.clear:
build_utils.DeleteDirectory(options.dest)
build_utils.MakeDirectory(options.dest)
deps = []
if options.files:
DoCopy(options, deps)
if options.renaming_sources:
DoRenaming(options, deps)
if options.depfile:
action_helpers.write_depfile(options.depfile, options.stamp, deps)
if options.stamp:
build_utils.Touch(options.stamp)
if __name__ == '__main__':
sys.exit(main(sys.argv[1:]))
|