Python--获取指定目录下的指定类型文件

获取目录（srcdir）下指定类型（fileext）的的文件，输出为一个list

def Dirfile(srcdir ,fileext):
    filelist = []
    dstlist = []
    filelist.append(srcdir)
    while len(filelist) != 0:
        tmpdir = filelist.pop()
        if os.path.isdir(tmpdir):
            tmpvec = os.listdir(tmpdir)
            for tmpitem in tmpvec:
                filelist.append(tmpdir + \"\\\\\" + tmpitem)
        else:
            if os.path.isfile(tmpdir):
                if os.path.splitext(tmpdir)[1] == fileext:
                    dstlist.append(tmpdir)
    return dstlist

获取目录（srcdir）下所有文件名为（dstdir）种指定类型（fileext）的的文件，输出为一个list

def Finddirfile(srcdir, dstdir, fileext):
    filelist = []
    dstlist = []
    resultlist = []
    filelist.append(srcdir)
    while len(filelist) != 0:
        tmpdir = filelist.pop(0)
        if os.path.isdir(tmpdir):
            tmpvec = os.listdir(tmpdir)
            for tmpitem in tmpvec:
                filelist.append(tmpdir + \"\\\\\" + tmpitem)
            if os.path.split(tmpdir)[1] == dstdir:
                for item in os.listdir(tmpdir):
                    dstlist.append(tmpdir + \"\\\\\" + item)
    while len(dstlist) != 0:
        tmpdst = dstlist.pop(0)
        if os.path.isdir(tmpdst):
            for tempdst in os.listdir(tmpdst):
                dstlist.append(tmpdst + \"\\\\\" + tempdst)
        else:
            if os.path.isfile(tmpdst):
                if os.path.splitext(tmpdst)[1] == fileext:
                    resultlist.append(tmpdst)
    if len(resultlist) != 0:
        return resultlist
    else:
        return resultlist