diff options
| author | arcadia-devtools <arcadia-devtools@yandex-team.ru> | 2022-03-24 16:16:08 +0300 |
|---|---|---|
| committer | arcadia-devtools <arcadia-devtools@yandex-team.ru> | 2022-03-24 16:16:08 +0300 |
| commit | 68dacc71ebba0395bdfc0d518c0b72bbb592fe31 (patch) | |
| tree | 5b827a5b720a8fb8e10109d554b1a687b8dbe046 /build/scripts/copy_docs_files.py | |
| parent | fc131012dada714b26d872d3532bde52e4aba2b1 (diff) | |
| download | ydb-68dacc71ebba0395bdfc0d518c0b72bbb592fe31.tar.gz | |
intermediate changes
ref:128a8f4d9d8b1f87d5a730a31e6abaecd194bfe9
Diffstat (limited to 'build/scripts/copy_docs_files.py')
| -rw-r--r-- | build/scripts/copy_docs_files.py | 76 |
1 files changed, 76 insertions, 0 deletions
diff --git a/build/scripts/copy_docs_files.py b/build/scripts/copy_docs_files.py new file mode 100644 index 0000000000..8c6c064a03 --- /dev/null +++ b/build/scripts/copy_docs_files.py @@ -0,0 +1,76 @@ +import argparse +import errno +import os +import process_command_files as pcf +import shutil +import sys + + +def parse_args(): + parser = argparse.ArgumentParser() + parser.add_argument('--build-root', required=True) + parser.add_argument('--dst-dir', required=True) + parser.add_argument('--existing', choices=('skip', 'overwrite'), default='overwrite') + parser.add_argument('--source-root', required=True) + parser.add_argument('--src-dir', required=None) + parser.add_argument('files', nargs='*') + return parser.parse_args(pcf.get_args(sys.argv[1:])) + + +def makedirs(dirname): + try: + os.makedirs(dirname) + except OSError as e: + if e.errno == errno.EEXIST and os.path.isdir(dirname): + pass + else: + raise + + +def copy_file(src, dst, overwrite=False, orig_path=None, generated=False): + if os.path.exists(dst) and not overwrite: + return + + makedirs(os.path.dirname(dst)) + + with open(src, 'r') as fsrc, open(dst, 'w') as fdst: + if (orig_path or generated) and src.endswith('.md'): + fdst.write('---\n{}\n\n---\n'.format('generated: true' if generated else 'vcsPath: {}'.format(orig_path))) + shutil.copyfileobj(fsrc, fdst) + + +def main(): + args = parse_args() + + source_root = os.path.normpath(args.source_root) + os.path.sep + build_root = os.path.normpath(args.build_root) + os.path.sep + + dst_dir = os.path.normpath(args.dst_dir) + assert dst_dir.startswith(build_root) + makedirs(dst_dir) + + src_dir = os.path.normpath(args.src_dir) + os.path.sep + assert src_dir.startswith(source_root) + + if src_dir.startswith(source_root): + root = source_root + is_from_source_root = True + elif src_dir.startswith(build_root): + root = build_root + is_from_source_root = False + else: + assert False, 'src_dir [{}] should start with [{}] or [{}]'.format(src_dir, source_root, build_root) + + is_overwrite_existing = args.existing == 'overwrite' + + for f in [os.path.normpath(f) for f in args.files]: + src_file = os.path.join(src_dir, f) + dst_file = os.path.join(dst_dir, f) + if src_file == dst_file: + continue + rel_path = src_file[len(root):] if is_from_source_root else None + copy_file(src_file, dst_file, overwrite=is_overwrite_existing, orig_path=rel_path) + + +if __name__ == '__main__': + main() |