import os
import sys
import string

#以指定模式打开指定文件，获取文件句柄
def getFileIns(filePath,model):
 print("打开文件")
 print(filePath)
 print(model)
 return open(filePath,model)

#获取需要处理的文件
def getProcFile(path):
 return os.listdir(path)

#判断是否满足某个条件，如果满足则执行
def isTrue(outFileIns,s):
 findStr1 = "LINE_COUNT_UPDATE INTEGER := 0;"
 writeStr1 = "LINE_COUNT_ERROR INTEGER := 0; --错误数据XX条"
 findStr2 = "DBMS_OUTPUT.PUT_LINE('处理完毕"
 writeStr2 = "DBMS_OUTPUT.PUT_LINE('错误数据['||LINE_COUNT_ERROR||']条.');"
 findStr3 = "DBMS_OUTPUT.PUT_LINE('插入数据['||CUR_RESULT.INT_ID||']时发生异常...');"
 writeStr3 = "LINE_COUNT_ERROR := LINE_COUNT_ERROR+1;"
 findStr4 = "DBMS_OUTPUT.PUT_LINE('更新数据['||CUR_RESULT.INT_ID||']时发生异常...');"
 
 if s.find(findStr1) != -1:
 outFileIns.write(s)
 outFileIns.write(writeStr1+"
")
 elif s.find(findStr2) != -1:
 outFileIns.write(s)
 outFileIns.write(writeStr2+"
")
 elif s.find(findStr3) != -1:
 outFileIns.write(s)
 outFileIns.write("	"+writeStr3+"
")
 elif s.find(findStr4) != -1:
 outFileIns.write(s)
 outFileIns.write("	"+writeStr3+"
")
 elif s.find("CS_OSLGIS") != -1:
 outFileIns.write(s.replace("CS_OSLGIS","CQ_RMW"))
 elif s.find("AND A.LONGITUDE >") != -1:
 outFileIns.write("	AND A.LONGITUDE IS NOT NULL
	AND A.LONGITUDE IS NOT NULL
	AND ROWNUM<2
")
 elif s.find(") LOOP") != -1:
 outFileIns.write("	) LOOP
")
 else:
 outFileIns.write(s.replace("||')',2","||')',3"))
 

 

#读取并处理文本
def getAndProc(inFileIns,outFileIns):
 lines = inFileIns.readlines()
 for s in lines:
 #print(s)
 isTrue(outFileIns,s)


if __name__=="__main__":
 
 inFileMod = "r"
 outFileMod = "w"
 path = "D:\rmsdata2gis"
 for tmpFile in os.listdir(path):
 inFilePath = path+"\"+tmpFile
 outFilePath = path+"\BAK_"+tmpFile
 inFileIns = getFileIns(inFilePath,inFileMod)
 outFileIns = getFileIns(outFilePath,outFileMod)
 getAndProc(inFileIns,outFileIns)
 inFileIns.close()
 outFileIns.close()