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

时间:2012-04-13 08:21:53

标签: python oracle cx-oracle

我有一个SQL文件,我想使用cx_Oracle python库在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(';')拆分字符串(这里建议为execute a sql script file from cx_oracle?),因为注释将被拆分(并将导致错误)并且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))

1 个答案:

答案 0 :(得分:2)

可以同时执行多个语句,但它是半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
    >>>