python 文件夹复制加强版

shutil模块主要用于文件夹的操作。其中copytree用来对文件夹进行复制，但是比较遗憾的是，如果目标文件已经存在的话，该函数就会报错抛异常了，非常地不给力。求人不如求已，看到帮助文档中有此方法的源码，就修改了一下，使其默认可以支持文件和文件夹的覆盖。

import os

import os.path

import shutil

def copytree(src, dst, symlinks=False):

    names = os.listdir(src)

    if not os.path.isdir(dst):

        os.makedirs(dst)

          

    errors = []

    for name in names:

        srcname = os.path.join(src, name)

        dstname = os.path.join(dst, name)

        try:

            if symlinks and os.path.islink(srcname):

                linkto = os.readlink(srcname)

                os.symlink(linkto, dstname)

            elif os.path.isdir(srcname):

                copytree(srcname, dstname, symlinks)

            else:

                if os.path.isdir(dstname):

                    os.rmdir(dstname)

                elif os.path.isfile(dstname):

                    os.remove(dstname)

                shutil.copy2(srcname, dstname)

            # XXX What about devices, sockets etc.?

        except (IOError, os.error) as why:

            errors.append((srcname, dstname, str(why)))

        # catch the Error from the recursive copytree so that we can

        # continue with other files

        except OSError as err:

            errors.extend(err.args[0])

    try:

        copystat(src, dst)

    except WindowsError:

        # can't copy file access times on Windows

        pass

    except OSError as why:

        errors.extend((src, dst, str(why)))

    if errors:

        raise Error(errors)

if __name__ == '__main__':

    copytree('E:/book', 'E:/newbook')