Bootstrap

Python - 多次检查后缀名(endwith)

在通过后缀名查找类型文件的时候, 多次使用endwith, 使用元组(tuple), 简化操作.
此类方式, 也能够应用于if语句多次类似检測.


代码

# 列出目录内全部代码
def list_dictionary_codes(root_dir):
    paths_list = []
    for parent, dirNames, fileNames in os.walk(root_dir):
        for name in fileNames:
            ext = ['.h', '.m', '.xib', '.json', '.c', '.cpp', '.mm', '.md']
            if name.endswith(tuple(ext)):
                paths_list.append(os.path.join(parent, name))
    return paths_list

;