我將一些查詢保存在單個.sql腳本中，而不是直接將它們寫入.py腳本中的變量中?，F在我想使用Python閱讀這個sql腳本，用分號拆分每個sql語句，然后將每個腳本傳遞給Psycopg2游標進行順序執行。python 腳本似乎正確讀取了文件，但是當我嘗試執行語句時，會引發“無法執行空查詢”錯誤。也許問題是：在這個sql腳本中有許多換行符。其中的語句寫如下：DROP TABLE IF EXISTS target_schema.some_table;SELECT columnFROM schema.table;下面是蟒蛇代碼：import psycopg2import pathlibconn = 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()任何人都可以幫我解決這個問題嗎？
1 回答
波斯汪
TA貢獻1811條經驗 獲得超4個贊
將 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()
添加回答
舉報
0/150
提交
取消