skia2/gn/copy_git_directory.py
Hal Canary 22814dbf6b GN tools: copy_git_directory.py better error message.
No-Try: true
Change-Id: Ifd92c47ae1beaaa53430e6fa8e658a795a398b55
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/255299
Reviewed-by: Hal Canary <halcanary@google.com>
Commit-Queue: Hal Canary <halcanary@google.com>
2019-11-19 17:04:18 +00:00

61 lines
2.2 KiB
Python
Executable File

#! /usr/bin/env python
# Copyright 2019 Google LLC.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import os
import shutil
import subprocess
import sys
def copy_git_directory(src, dst, out=None):
'''
Makes a copy of `src` directory in `dst` directory. If files already exist
and are identical, do not touch them. If extra files or directories exist,
remove them. Assume that `src` is a git directory so that `git ls-files` can
be used to enumerate files. This has the added benefit of ignoring files
not tracked by git. Also, if out is not None, write summary of actions to out.
If `dst` is a top-level git directory, the `.git` directory will be removed.
'''
if not os.path.isdir(src):
raise Exception('Directory "%s" does not exist.' % src)
if not os.path.isdir(dst):
os.makedirs(dst)
ls_files = subprocess.check_output(['git', 'ls-files', '-z', '.'], cwd=src)
src_files = set(p for p in ls_files.split('\0') if p)
abs_src = os.path.abspath(src)
cwd = os.getcwd()
try:
os.chdir(dst)
def output(out, sym, dst, path):
if out:
out.write('%s %s%s%s\n' % (sym, dst, os.sep, path))
for dirpath, dirnames, filenames in os.walk('.', topdown=False):
for filename in filenames:
path = os.path.normpath(os.path.join(dirpath, filename))
if path not in src_files:
output(out, '-', dst, path)
os.remove(path)
for filename in dirnames:
path = os.path.normpath(os.path.join(dirpath, filename))
if not os.listdir(path): # Remove empty subfolders.
output(out, '-', dst, path + os.sep)
os.rmdir(path)
for path in src_files:
src_path = os.path.join(abs_src, path)
if os.path.exists(path):
with open(path) as f1:
with open(src_path) as f2:
if f1.read() == f2.read():
continue
output(out, '+', dst, path)
shutil.copy2(src_path, path)
finally:
os.chdir(cwd)
if __name__ == '__main__':
if len(sys.argv) != 3:
sys.stderr.write('\nUsage:\n %s SRC_DIR DST_DIR\n\n' % sys.argv[0])
sys.exit(1)
copy_git_directory(sys.argv[1], sys.argv[2], sys.stdout)