python 遍历文件夹及子文件夹，输出匹配文件

import os
import re
import sys

fistr_dir = os.path.abspath(sys.argv[1])
#print("fistr_dir is:\t{}".format(fistr_dir))

def get_all_file_from_dir(path_dir):
    # 遍历文件夹，输出文件及子文件夹内文件
    if os.path.exists(path_dir):
        path_dir = os.path.abspath(path_dir)
        for i in os.listdir(path_dir):
            path_i = os.path.join(path_dir, i)
            if os.path.isfile(path_i):
                if re.search('.bam$|bai$', path_i):
                    print(path_i)
            else:
                get_all_file_from_dir(path_i)

get_all_file_from_dir(fistr_dir)