python 复制文件夹_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')