scripts/sdk/common/files.py

#!/usr/bin/env python3.8
# Copyright 2018 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import errno
import os
import shutil


def make_dir(file_path):
    '''Creates the directory hierarchy for the given file and returns the
    given path.
    '''
    target = os.path.dirname(file_path)
    try:
        os.makedirs(target)
    except OSError as exception:
        if exception.errno == errno.EEXIST and os.path.isdir(target):
            pass
        else:
            raise
    return file_path


def copy_tree(src, dst, allow_overwrite=True):
    '''Recursively copies a directory into another.

    Differs with shutil.copytree in that it won't fail if the destination
    directory already exists.

    Args:
        src: The source directory.
        dst: The destination directory.
        allow_overwrite: True to allow files to be overwritten, otherwise raise
            an exception.

    Raises:
        Exception: If allow_overwrite is False and a destination file already
            exists.

    '''
    if not os.path.isdir(dst):
        os.makedirs(dst)
    for path, directories, files in os.walk(src):

        def get_path(name):
            source_path = os.path.join(path, name)
            dest_path = os.path.join(dst, os.path.relpath(source_path, src))
            return (source_path, dest_path)

        for dir in directories:
            source, dest = get_path(dir)
            if not os.path.isdir(dest):
                os.makedirs(dest)
        for file in files:
            source, dest = get_path(file)
            if not allow_overwrite and os.path.exists(dest):
                raise Exception(
                    "cannot copy file: file '%s': File exists" % dest)
            shutil.copy2(source, dest)