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')