Python-工具-批量插入INSERT语句(将sql文件导入数据库)

import cx_Oracle

db = cx_Oracle.connect(‘tssh/rdd3sjtest@sjtest‘)
dblj = db.cursor()
count = 0
ycts  = 0
cgts  = 0
str1 =‘‘
while(1 == 1):
    file_name = input("请输入文件名\n")
    print(2)
    fn = file_name[-3:].upper()
    if(fn != ‘SQL‘):
        print("文件名无效，必须是sql文件\n")
        continue
    fileobject = open(file_name)
    for line in fileobject:
        st = line[0:6].upper()
        print(line[0:6].upper())
        if(st ==‘INSERT‘ or st == ‘VALUES‘):
            count = count + 1
            str1 = str1 + line
            if(count == 2):
                str1 = str1.replace(‘;‘,‘‘)
                print(str1)
                try:
                    dblj.execute(str1)
                    dblj.commit()
                    cgts = cgts + 1
                except IOError:
                    print("Error: 没有找到文件或读取文件失败")
                else:
                    ycts = ycts + 1                
                count = 0
                str1 = ‘‘
    print("成功"+cgts+"条","失败"+ycts+"条")
    fileobject.close()
    dblj.close()