Question

我正在尝试使用Python来运行PostgreSQL函数，该函数将数据从一个表插入到另一个表中。如果我在pgAdmin中运行该函数，它可以工作，但如果我通过Python运行它，我会得到返回值，表示它已被执行，但它没有插入任何记录。

这是我的Python代码

import psycopg2
from parse_config import fn_parse_config

def run_function():

  conn = None
  try:
    # read database configuration
    login_params = fn_parse_config("etl_process.ini", "psycopg_login")
    # connect to the PostgreSQL database
    conn = psycopg2.connect(**login_params)
    # create a cursor object for execution
    cur = conn.cursor()
    # another way to call a stored procedure
    #cur.execute("SELECT * FROM usp_test();")
    cur.callproc('usp_test')
    # process the result set
    row = cur.fetchone()
    while row is not None:
        print(row)
        row = cur.fetchone()
    # close the communication with the PostgreSQL database server
    cur.close()
except (Exception, psycopg2.DatabaseError) as error:
    print(error)
finally:
    if conn is not None:
        conn.close()         
if __name__ == '__main__':
     run_function()

这是我的功能

CREATE OR REPLACE FUNCTION usp_test() 
    RETURNS integer AS $$

BEGIN 
INSERT INTO staging_raw_file (Site,
            StyleCode)
        select CAST(a.Site as VARCHAR(10)),
        CAST(a.StyleCode as VARCHAR(30))
from import_raw_file a;

return 5;

END; $$

LANGUAGE plpgsql;

Answer 1

找到答案，需要自动提交设置

conn = psycopg2.connect(**login_params)
conn.autocommit = True

没有它它实际上没有运行插入，但它将仅从没有那个的选择返回结果

Postgres函数 - 从Python运行，没有插入记录

1 个答案: