20行Python代码执行SQL文件

关于Python如何连接数据库并执行SQL语句,几乎所有的Python教程都会讲,教程里基本只介绍了执行单条SQL语句的方法,但是实际生产过程中可不只是执行一两条语句,动辄几十条甚至上百条的SQL语句,一个.SQL的SQL文件正常都会包含多条SQL语句,如果把这些SQL语句直接贴进Python代码文件,势必会影响代码的可读性和美感,代码量也冗余,那么Python能不能像oracle的PL/SQL客户端或者navicat一样读取SQL文件并且执行会是一件非常有趣的事情。

经过不断百度和尝试,一段优雅的Python代码终于敲出来,作为开源的忠实粉丝,技术始于分享,现将Python执行SQL文件的代码分享在CSDN,代码实例是oracle数据库,其他数据库(如MySQL)可参考。

import cx_Oracle
import os
try:
    os.environ['NLS_LANG'] = 'SIMPLIFIED CHINESE_CHINA.utf8'
    db = cx_Oracle.connect('数据库用户名', '数据库密码', '数据库IP地址/数据库实例')
    c = db.cursor()
    ##读取SQL文件,获得sql语句的list
    with open(u'E:\\test.sql', 'r+') as f:
        sql_list = f.read().split(';')[:-1]  # sql文件最后一行加上;
        sql_list = [x.replace('\n', ' ') if '\n' in x else x for x in sql_list]  # 将每段sql里的换行符改成空格
    ##执行sql语句,使用循环执行sql语句
    for sql_item  in sql_list:
        # print (sql_item)
        c.execute(sql_item)
except cx_Oracle.Error as e:
    print e
finally:
    c.close()
    db.commit()
    db.close()

你可能感兴趣的:(Python)