##^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^Lance Simms, Stanford University 2009
##^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^Lance Simms, Stanford University '09
##PURPOSE:
##  To append a signature to files I have written
##
import os


##Get all of the .py files in the directory
Files=os.listdir('./')
for File in Files:

  ##################################################PYTHON  
  #Check to see if it's a .py python script
  if File.find('.py') != -1 and File.find('.pyc') == -1:
    print 'Adding Signature to ' + File
    ReadHandle  = open(File, 'r')
    #Get all lines in the file as a list
    FileLines   = ReadHandle.readlines()
    ReadHandle.close()
    FileLines.insert(0, "##^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^Lance Simms, Stanford University 2009\n")
    WriteHandle = open(File, 'w')
    for Line in FileLines:
      WriteHandle.write(Line)
    WriteHandle.close()

  ##################################################IDL
  #Check to see if it'a .pro IDL file
  if File.find('.pro') !=-1 :

    print 'Adding Signature to ' + File
    ReadHandle  = open(File, 'r')
    #Get all lines in the file as a list
    FileLines   = ReadHandle.readlines()
    ReadHandle.close()
    FileLines.insert(0, ";^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^Lance Simms, Stanford University 2009\n")
    WriteHandle = open(File, 'w')
    for Line in FileLines:
      WriteHandle.write(Line)
    WriteHandle.close()