python-appimage/python_appimage/utils/fs.py

import errno
import os
import shutil

from .log import debug


__all__ = ['copy_file', 'copy_tree', 'make_tree', 'remove_file', 'remove_tree']


# Wrap some file system related functions
def make_tree(path):
    '''Create directories recursively if they don't exist
    '''
    debug('MKDIR', path)
    return os.makedirs(path, exist_ok=True)


def copy_file(source, destination, update=False, verbose=True):
    '''
    '''
    name = os.path.basename(source)
    if verbose:
        debug('COPY', '%s from %s', name, os.path.dirname(source))
    if os.path.exists(source) and (
        not update
        or (
            not os.path.exists(destination)
            or (os.path.getmtime(source) > os.path.getmtime(destination))
        )
    ):
        shutil.copy(source, destination)


def copy_tree(source, destination):
    '''Copy (or update) a directory preserving symlinks
    '''
    if not os.path.exists(source):
        raise OSError(errno.ENOENT, 'No such file or directory: ' + source)

    name = os.path.basename(source)
    debug('COPY', '%s from %s', name, os.path.dirname(source))

    for root, _, files in os.walk(source):
        relpath = os.path.relpath(root, source)
        dirname = os.path.join(destination, relpath)
        os.makedirs(dirname, exist_ok=True)
        for file_ in files:
            src = os.path.join(root, file_)
            dst = os.path.join(dirname, file_)
            if os.path.islink(src):
                try:
                    os.remove(dst)
                except OSError:
                    pass
                linkto = os.readlink(src)
                os.symlink(linkto, dst)
            else:
                copy_file(src, dst, update=True, verbose=False)


def remove_file(path):
    '''remove a file if it exists
    '''
    name = os.path.basename(path)
    debug('REMOVE', '%s from %s', name, os.path.dirname(path))
    try:
        os.remove(path)
    except OSError:
        pass


def remove_tree(path):
    '''remove a directory if it exists
    '''
    name = os.path.basename(path)
    debug('REMOVE', '%s from %s', name, os.path.dirname(path))
    try:
        shutil.rmtree(path)
    except OSError:
        pass