批量增加文件头注释的python小程序

 

import sys
import os
import glob
import string

def isBlankLine(line):
    for ch in line:
        if ch in [' ',' ',' '] :
            continue
        else:
            return False
    return True

sys.argv.append('test*.txt')
sys.argv.append('comment.txt')
if len(sys.argv) < 3:
    exit()

filenames = glob.glob(sys.argv[1])
commentfilename = sys.argv[2]

# read the comments
f = file(commentfilename)
commentLines = f.readlines()
commentLines += ' '
f.close()
print string.join(commentLines, '')

# add the comment into each source file
for srcfilename in filenames:
    f = file(srcfilename)
    srcfileLines = f.readlines()
    f.close()
    # filter out the previous comment header
    for line in srcfileLines:
        if isBlankLine(line) :
            continue
        if line[0] == '#' :
            srcfileLines = commentLines + srcfileLines
            f = file(srcfilename, 'w')
            f.writelines(srcfileLines)
            print 'Add comment to '+srcfilename
        break