python 遍歷文件夾下的所有文件

基礎

import os


# 遍歷文件夾
def walkFile(file):
    for root, dirs, files in os.walk(file):

        # root 表示當前正在訪問的文件夾路徑
        # dirs 表示該文件夾下的子目錄名list
        # files 表示該文件夾下的文件list

        # 遍歷文件
        for f in files:
            print(os.path.join(root, f))

        # 遍歷所有的文件夾
        for d in dirs:
            print(os.path.join(root, d))


def main():
    walkFile("f:/ostest/")


if __name__ == '__main__':
    main()

進階

計算文件夾內py文件代碼的行數

total_num = 0
for base_path,folder_list,file_list in os.walk(target_path):
    for file_name in file_list:
        file_path = os.path.join(base_path,file_name)
        file_ext = file_path.rsplit('.',maxsplit=1)
        if len(file_ext) != 2:
            # 沒有后綴名
            continue
        if file_ext[1] != 'py':
            # 不是py文件
            continue
        file_num = 0
        with open(file_path,'rb') as f:
            for line in f:
                # 去空格
                line = line.strip()
                if not line:
                    continue
                # 去除 # 注釋
                if line.startswith(b'#'):
                    continue
                file_num += 1
        total_num += file_num