批量增加檔案頭注釋的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