Python：去除目录下所有 .c .h 文件中多余的空格（二）

功能：去除 .c和.h文件中多余空格。此处的多余空格指的是行尾的空格和空行行首的空格。

环境：Windows 7旗舰版

Python：3.6.5

Version 2：（支持多及目录）

# -*- coding: utf-8 -*-
"""
Created on Fri Nov  9 08:50:54 2018

@author: sdc
"""

import sys
import os
import pprint

FILENAMEFILTERS = ('.c', '.h')

#处理过滤出来的文件
def handleCertainFiles(filenames):
    for filename in filenames:
        print(filename)
        #使用with 方式，不用调用close()函数
        with open(filename, 'r+', encoding='utf-8') as fileHandle:#不能以'a+'的方式打开
            fileContents = fileHandle.readlines()
            if len(fileContents) == 0:
                print('%s file is empty' % filename)
                continue
            #确保最后一行有换行符
            finalLineContent = fileContents[-1]
            if finalLineContent[-1] != '\n':
                finalLineContent = finalLineContent + '\n'
                fileContents[-1] = finalLineContent
                #print(fileContents)
            
            newFileContents = []	#存储修改后的内容，最终将其写入文件中
            for lineContent in fileContents:
                newLineContent = lineContent[:-1] #去除 \n
                if newLineContent.strip() != '': #如果去除首尾的空格后不为空，那么只应该去掉行尾的空格
                    newLineContent = newLineContent.rstrip() + '\n'
                else:
                    newLineContent = newLineContent.strip() + '\n'
                        
                newFileContents.append(newLineContent)
            fileHandle.seek(0, 0)	#将文件指针放在文件开头
            fileHandle.truncate()	#清空文件
            #print(newFileContents)
            fileHandle.writelines(newFileContents)	#写入修改后的内容

#查找符合要求的文件
#返回符合要求的文件列表
def findCertainFiles(filenames):
    
    result = []
    
    for filename in filenames:
        filenamePrefix, filenameSuffix = os.path.splitext(filename) #分离文件名和文件后缀
        if filenameSuffix in FILENAMEFILTERS:
            result.append(filename)
    return result

#在目录中找到所有文件
#返回所有的文件，包含文件的路径
def findFilesInDir(dir):
    
    result = []
    
    if not os.path.exists(dir):
        print('Wrong dir name')
        return result
    for mainDir, subDir, files in os.walk(dir):
        for file in files:
            filename = os.path.join(mainDir, file)
            result.append(filename)
    return result

def main():
    allFiles = []
    if len(sys.argv) < 2:
        print('Usage: delNeedlessSpace.py dirname')
        return
    allFiles = findFilesInDir(sys.argv[1])
    pprint.pprint(tuple(allFiles))
    allFiles = findCertainFiles(allFiles)
    pprint.pprint(tuple(allFiles))
    handleCertainFiles(allFiles)

if __name__ == '__main__':
    main()

使用方式：

python test.py dirname