Psycopg2 “无法执行空查询”

Question

我将一些查询保存在单个.sql脚本中，而不是直接将它们写入.py脚本中的变量中。

Now I want to read this sql script using Python, split each sql statement by semicolon and then pass each of these scripts to Psycopg2 cursor for sequential execution.

python 脚本似乎可以正确读取文件，但是当我尝试执行语句时会抛出“无法执行空查询”错误。

也许问题是：这个 sql 脚本中有许多换行符。 其中的语句如下所示：

DROP TABLE IF EXISTS
    target_schema.some_table
;

SELECT
    column
FROM
    schema.table
;

这是 python 代码：

import psycopg2
import pathlib

conn = psycopg2.connect(
     user=user
    ,password=password
    ,host=host
    ,port=port
    ,database=database
)

pg_cursor = conn.cursor()

scriptContents = Path('my_folder\my_sql_script.sql').read_text(encoding='utf-8')

sqlStatements = scriptContents.split(sep=';')

for statement in sqlStatements:    

    try:
        pg_cursor.execute(f'{statement}')
        conn.commit()
    except psycopg2.Error as errorMsg:
        print(errorMsg)        
        conn.rollback()

谁能帮我解决这个问题？

Answer 1

当您将 sql 文件拆分为; ，创建的列表将有空字符串作为最后一个元素，它将无法作为有效查询执行。 为了避免这种情况，让文件中的所有 sql 都有; 最后，尝试以下操作：

for statement in sqlStatements[:-1]:
# it will slice out the last element of your sqlStatements list
    try:
        pg_cursor.execute(f'{statement}')
        conn.commit()
    except psycopg2.Error as errorMsg:
        print(errorMsg)        
        conn.rollback()

另一种解决方案是将您的语句以新行分隔，然后按以下方式逐个读取它们：

my_sql_script.sql：

DROP TABLE IF EXISTS target_schema.some_table
SELECT column FROM schema.table

为了运行语句，请使用：

with open('my_folder\my_sql_script.sql','r', encoding='utf-8') as f:
    for statement in f.readlines():    
        try:
            pg_cursor.execute(f'{statement.rstrip()}')
            conn.commit()
        except psycopg2.Error as errorMsg:
            print(errorMsg)        
            conn.rollback()