批量增加文件头注释的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