import pickle
import random
 
saFlag = False  #<--- flag
 
saLessFile = raw_input("File to scan? ")
#opens the selected file
textf = open(saLessFile, 'r')
str1 = textf.read()
textf.close()
 
#makes every word in file a list
wordlist = str1.split(None)
 
for i in range(0, len(wordlist)):
  print wordlist[i],
 
print "\n\nScan loop follows:"
 
for i in range(0, len(wordlist)):
  if (wordlist[i].lower() == "sa001:"):
    print "SA001 found:"
    saFlag = True
    for j in range(i, (i+7)):
      print wordlist[j],
    print "..."
 
if saFlag:
  print "No altering needed\n"
 
else:
  print "SA not found turn document into the following?"
  wordlist.insert(0, "SA001: This is a test, blah blah blah blah blah!\n")
 
  #turns the list into a string with a space inbetween each word
  joinList = " ".join(wordlist)
  print joinList
 
  #asks user if they are sure they wish to modify
  sure = raw_input("Modify?(y or n): ")
 
  if sure == "y" or sure == "Y":
    #saves the new string over the old file's words, essentially modifying the contents
    file = open(saLessFile, "w")
    pickle.dump(joinList, file)
    file.close()
    print "modified..."
  elif sure == "n" or sure == "N":
    quit

pickle.dump(joinList, file)

file.write(joinList)

data = open(filename, "r").readlines()
found = 0
for rec in data:
   if ("sa001:" in rec) or ("SA001:") in rec:
      print "SA001 found"
      found =1
 
## file only changes if string is not found
if not found:
   fp = open(filename+".2", "w")
   fp.write( "SA001:\n")          ## add this line
   for rec in data:                       ## original data
      fp.write(rec)
   fp.close()

import re
mFilename = raw_input("File to scan? ")
mFile= open(mFilename).read()
 
 
print "Searching File..."
#the re.findall function
    #first parameter : string/regular expression/pattern to search for
    #second parameter: string to search in
    #third is optional: I passed in re.ignorecase to tell the module to ignore the case
if not re.findall('SA001:',mFile,re.IGNORECASE):
        #'SAOO1: is not in the file
      print "Text not found..."
      sure = raw_input("Modify?(y or n): ")
      if sure == "y" or sure == "Y":
          open(mFilename,'w').write("SA001:\n"+mFile)
          print "File Modified..."      
else:
    #'SAOO1: is already in the file
    print "No Altering needed."