使用python遍历文件夹取出特定的字符串

# -*- coding: utf-8 -*
import re
import os
# 需要处理的文件夹路径（绝对路径）
path = u"/Users/a140/Downloads/test2"
# 存放结果的文件路径（绝对路径）
results = u"/Users/a140/Downloads/result.txt"
# 载入文件列表
file_list = os.listdir( path )
# 文件排序
file_list.sort()
# 遍历所有文件
for file in file_list:
    # 打开要输出的文件
    fo = open( path + '/' + file, "r")
    # 输出文件名
    print("fileName: ", fo.name)
    # 打开要写入结果的文件
    resu = open(results, "a+")
    # 写入文件名和换行
    resu.write(fo.name + '
')
    # 遍历文件内的每一行
    for line in fo.readlines():
        # 判断当前行是否包含"TOTAL:"
        if len(re.findall('TOTAL:', line)) != 0:
            # 若包含则将此行赋值给变量tmp
            tmp = line
    # 关闭文件
    fo.close()
    # 取出TOTAL值
    TotalLine = tmp.split('TOTAL SWAP')[0].split(': ')
    # 输出TOTAL值
    print('Name: {}, TotalNum:{}'.format(TotalLine[0], TotalLine[1]))
    # 将TOTAL值写入结果的文件
    resu.write('Name: {}, TotalNum:{}'.format(TotalLine[0], TotalLine[1]) + '

')
    # 关闭文件
    resu.close()