我正在尝试从标准输入读取行,并使用 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()