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

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值