0

我正在尝试从标准输入读取行,并使用 plpythonu 存储过程将这些行中的数据插入到 PostgreSQL 数据库中。

当我在 Python 3 下调用该过程时,它会运行(为读取的每一行消耗一个序列值),但在 db 中不存储任何数据。当我从 psql 调用相同的过程时,它工作正常,在 db 中插入一行。

例如

操作:SELECT sl_insert_day('2017-01-02', '05:15');以用户 jazcap53 从 psql 中运行
结果:插入 day_id 为 1 的日期。

行动:python3 src/load/load_mcv.py < input.txt在命令行运行
结果:没有插入,但消耗了 2 个串行 day_id。

操作:SELECT sl_insert_day('2017-01-03', '06:15');以用户 jazcap53 从 psql 中运行
结果:插入 day_id 为 4 的日期。

文件:输入.txt:

DAY, 2017-01-05, 06:00
DAY, 2017-01-06, 07:00

输出:

('sl_insert_day() succeeded',)  
('sl_insert_day() succeeded',)  

我正在运行 Fedora 25、Python 3.6.0 和 PostgreSQL 9.5.6。

非常感谢任何可以帮助我的人!


下面是一个重现此行为的 MCV 示例。我希望我的问题出现在第 8 步或第 6 步中——为了完整性,包括了其他步骤。

用于创建 MCV 的步骤:

步骤 1) 创建数据库:

在 psql 中作为用户 postgres,
创建数据库 sl_test_mcv;

步骤 2) 数据库初始化:

文件:db/database_mcv.ini

[postgresql]
host=localhost
database=sl_test_mcv
user=jazcap53
password=*****

步骤 3) 运行数据库配置:

文件:db/config_mcv.py

from configparser import ConfigParser

def config(filename='db/database_mcv.ini', section='postgresql'):
    parser = ConfigParser()
    parser.read(filename)
    db = {}
    if parser.has_section(section):
        params = parser.items(section)
        for param in params:
            db[param[0]] = param[1]
    else:
        raise Exception('Section {} not found in the {} file'.format(section, filename))
    return db

步骤 4) 创建表:

文件:db/create_tables_mcv.sql

DROP TABLE IF EXISTS sl_day CASCADE;

CREATE TABLE sl_day (
    day_id SERIAL UNIQUE,
    start_date date NOT NULL,
    start_time time NOT NULL,
    PRIMARY KEY (day_id)
);

步骤 5) 创建语言:

CREATE LANGUAGE plpythonu;

步骤 6) 创建程序:

文件:db/create_procedures_mcv.sql

DROP FUNCTION sl_insert_day(date, time without time zone);

CREATE FUNCTION sl_insert_day(new_start_date date, 
    new_start_time time without time zone) RETURNS text AS $$
from plpy import spiexceptions
try:
    plan = plpy.prepare("INSERT INTO sl_day (start_date, start_time) \
            VALUES($1, $2)", ["date", "time without time zone"])
    plpy.execute(plan, [new_start_date, new_start_time])
except plpy.SPIError, e:
    return "error: SQLSTATE %s" % (e.sqlstate,)
else:
    return "sl_insert_day() succeeded"
$$ LANGUAGE plpythonu;

步骤 7) 授予权限:

文件:db/grant_privileges_mcv.sql

GRANT SELECT, UPDATE, INSERT, DELETE ON sl_day TO jazcap53;  
GRANT USAGE ON sl_day_day_id_seq TO jazcap53;

步骤 8) 以 python3 src/load/load_mcv.py < input.txt 运行程序:

文件:src/load/load_mcv.py

import sys 
import psycopg2
from spreadsheet_etl.db.config_mcv import config

def conn_exec():
    conn = None
    try:
        params = config()
        conn = psycopg2.connect(**params)
        cur = conn.cursor()
        last_serial_val = 0 
        while True:
            my_line = sys.stdin.readline()
            if not my_line:
                break
            line_list = my_line.rstrip().split(', ')
            if line_list[0] == 'DAY':
                cur.execute('SELECT sl_insert_day(\'{}\', \'{}\')'.
                            format(line_list[1], line_list[2]))
                print(cur.fetchone())
        cur.close()
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    finally:
        if conn is not None:
            conn.close()

if __name__ == '__main__':
    conn_exec()
4

1 回答 1

1

conn.commit()之后做cur.close()

于 2017-04-11T23:38:39.613 回答