python拷贝特定后缀名文件,并保留原始目录结构的实例

#!/usr/bin/python
# -*- coding: UTF-8 -*- 
import os
import shutil
def cp_tree_ext(exts,src,dest):
 """
 Rebuild the director tree like src below dest and copy all files like XXX.exts to dest 
 exts:exetens seperate by blank like "jpg png gif"
 """
 fp={}
 extss=exts.lower().split()
 for dn,dns,fns in os.walk(src):
 for fl in fns:
 if os.path.splitext(fl.lower())[1][1:] in extss:
 if dn not in fp.keys():
 fp[dn]=[]
 fp[dn].append(fl)
 for k,v in fp.items():
 relativepath=k[len(src)+1:]
 newpath=os.path.join(dest,relativepath)
 for f in v:
 oldfile=os.path.join(k,f)
 print("拷贝 ["+oldfile+"] 至 ["+newpath+"]")
 if not os.path.exists(newpath):
 os.makedirs(newpath)
 shutil.copy(oldfile,newpath)
#用法如下：
#
#cp_tree_ext(exts,src,dest)
#
#exts:以空格分隔的字符串，可多个拓展名，如"bat txt"
#src:原目录
#dest:目标目录，如果不存在，则建立
cp_tree_ext('doc docx','/home/lincoln/python/copy/source','/home/lincoln/python/copy/target')