从cx_oracle执行sql脚本文件?

时间:2011-09-23 17:11:42

标签: python database cx-oracle

有没有办法在python中使用cx_oracle执行sql脚本文件。

我需要在sql文件中执行我的create table脚本。

2 个答案:

答案 0 :(得分:11)

另一种选择是使用SQL * Plus(Oracle的命令行工具)来运行脚本。你可以使用subprocess模块从Python中调用它 - 这里有一个很好的演练:http://moizmuhammad.wordpress.com/2012/01/31/run-oracle-commands-from-python-via-sql-plus/

对于像tables.sql这样的脚本(注意故意的错误):

CREATE TABLE foo ( x INT );

CREATE TABLER bar ( y INT );

您可以使用以下功能:

from subprocess import Popen, PIPE

def run_sql_script(connstr, filename):
    sqlplus = Popen(['sqlplus','-S', connstr], stdin=PIPE, stdout=PIPE, stderr=PIPE)
    sqlplus.stdin.write('@'+filename)
    return sqlplus.communicate()

connstrcx_Oracle使用的连接字符串相同。 filename是脚本的完整路径(例如'C:\temp\tables.sql')。该函数打开一个SQL Plus会话(使用'-S'来静音其欢迎消息),然后排队“@filename”发送给它 - 这将告诉SQL Plus运行脚本。

sqlplus.communicate将命令发送到stdin,等待SQL * Plus会话终止,然后返回(stdout,stderr)作为元组。使用上面的tables.sql调用此函数将提供以下输出:

>>> output, error = run_sql_script(connstr, r'C:\temp\tables.sql')
>>> print output

Table created.

CREATE TABLER bar (
       *
ERROR at line 1:
ORA-00901: invalid CREATE command

>>> print error

这将需要一些解析,具体取决于您想要返回到程序的其余部分 - 如果它是交互式的,您可以向用户显示整个输出,或者如果您只是想要扫描单词“ERROR”检查它是否运行正常。

答案 1 :(得分:9)

cx_oracle试图遵守的

PEP-249实际上没有这样的方法。

但是,这个过程应该非常简单。将文件的内容拉成一个字符串,将其拆分为“;”字符,然后在结果数组的每个成员上调用.execute。我假设“;” character仅用于分隔文件中的oracle SQL语句。

f = open('tabledefinition.sql')
full_sql = f.read()
sql_commands = full_sql.split(';')

for sql_command in sql_commands:
    curs.execute(sql_command)