python-3.x - Postgres 函数 - 从 Python 运行，不插入记录

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;

标签： python-3.xpostgresql

Answer 1

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

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

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