使用python中的cx_Oracle使用PL/SQL和DML/DDL解析SQL文件

Question

我有一个SQL文件,我想使用cx_Oraclepython库在oracle中解析和执行.SQL文件包含经典的DML/DDL和PL/SQL,例如.它看起来像这样:

create.sql:

-- This is some ; malicious comment
CREATE TABLE FOO(id numeric);

BEGIN
  INSERT INTO FOO VALUES(1);
  INSERT INTO FOO VALUES(2);
  INSERT INTO FOO VALUES(3);
END;
/
CREATE TABLE BAR(id numeric);

如果我在SQLDeveloper或SQL*Plus中使用此文件,它将被拆分为3个查询并执行.

但是,cx_Oracle.connect(...).cursor().execute(...)一次只能获取一个查询,而不是整个文件.我不能简单地拆分字符串string.split(';')(这里建议从cx_oracle执行sql脚本文件？),因为两个注释都将被拆分(并且会导致错误)并且PL/SQL块不会作为单个命令执行,因此导致错误.

在Oracle论坛(https://forums.oracle.com/forums/thread.jspa?threadID=841025)上,我发现cx_Oracle本身不支持解析整个文件.我的问题是 - 有一个工具可以帮我吗？例如.我可以调用一个python库将我的文件拆分成查询？

编辑:最好的解决方案似乎直接使用SQL*Plus.我用过这段代码:

# open the file
f = open(file_path, 'r')
data = f.read()
f.close()

# add EXIT at the end so that SQL*Plus ends (there is no --no-interactive :(
data = "%s\n\nEXIT" % data

# write result to a temp file (required, SQL*Plus takes a file name argument)
f = open('tmp.file', 'w')
f.write(data)
f.close()

# execute SQL*Plus
output = subprocess.check_output(['sqlplus', '%s/%s@%s' % (db_user, db_password, db_address), '@', 'tmp.file'])

# if an error was found in the result, raise an Exception
if output.find('ERROR at line') != -1:
    raise Exception('%s\n\nStack:%s' % ('ERROR found in SQLPlus result', output))

Answer 1

可以同时执行多个语句，但它是半 hacky 的。您需要包装语句并一次执行一个。

>>> import cx_Oracle
>>>
>>> a = cx_Oracle.connect('schema/pw@db')
>>> curs = a.cursor()
>>> SQL = (("""create table tmp_test ( a date )"""),
... ("""insert into tmp_test values ( sysdate )""")
... )
>>> for i in SQL:
...     print i
...
create table tmp_test ( a date )
insert into tmp_test values ( sysdate )
>>> for i in SQL:
...     curs.execute(i)
...
>>> a.commit()
>>>

正如您所指出的，这并不能解决分号问题，该问题没有简单的答案。在我看来，你有 3 个选择：

1. 编写一个过于复杂的解析器，我认为这根本不是一个好的选择。

2. 不要从 Python 执行 SQL 脚本；将代码放在单独的 SQL 脚本中，这样解析就很容易，放在单独的 Python 文件中，嵌入到 Python 代码中，放在数据库的过程中......等等。这可能是我的首选选项。

3. 以这种方式使用subprocess和调用脚本。这是最简单、最快的选项，但cx_Oracle根本没有用。

   >>> import subprocess
   >>> cmdline = ['sqlplus','schema/pw@db','@','tmp_test.sql']
   >>> subprocess.call(cmdline)
   
   SQL*Plus: Release 9.2.0.1.0 - Production on Fri Apr 13 09:40:41 2012
   
   Copyright (c) 1982, 2002, Oracle Corporation.  All rights reserved.
   
   
   Connected to:
   Oracle Database 11g Release 11.2.0.1.0 - 64bit Production
   
   SQL>
   SQL> CREATE TABLE FOO(id number);
   
   Table created.
   
   SQL>
   SQL> BEGIN
     2    INSERT INTO FOO VALUES(1);
     3    INSERT INTO FOO VALUES(2);
     4    INSERT INTO FOO VALUES(3);
     5  END;
     6  /
   
   PL/SQL procedure successfully completed.
   
   SQL> CREATE TABLE BAR(id number);
   
   Table created.
   
   SQL>
   SQL> quit
   Disconnected from Oracle Database 11g Release 11.2.0.1.0 - 64bit Production
   0
   >>>