我正在使用python脚本将数据加载到数据库批量加载器.
加载程序的输入是stdin.我一直无法获得正确的语法来调用基于Unix的批量加载器,并传递要加载的python列表的内容.
我一直在阅读有关Popen和PIPE的信息,但它们的表现却不尽如人意.
python列表包含要批量加载的数据库记录.在linux中,它看起来类似于:
echo "this is the string being written to the DB" | sql -c "COPY table FROM stdin"
用此命令使用的python列表替换echo语句的正确方法是什么?
我没有此过程的示例代码,因为我已经使用一些非常简单的语法尝试了Popen和PIPE的功能,但未获得所需的结果.
任何帮助将不胜感激.
谢谢
解决方法:
如果您的数据又短又简单,则可以对整个列表进行预格式化,并使用如下所示的子流程进行简单处理:
import subprocess
data = ["list", "of", "stuff"]
proc = subprocess.Popen(["sql", "-c", "COPY table FROM stdin"], stdin=subprocess.PIPE)
proc.communicate("\n".join(data))
如果数据太大而无法像这样进行预格式化,那么您也可以尝试直接使用stdin管道,尽管如果您也需要从stdout / stderr读取数据,则在使用管道时子进程模块会出现问题.
for line in data:
print >>proc.stdin, line