使用 python 列表作为使用 stdin 作为输入的 linux 命令的输入

Question

我正在使用 python 脚本将数据加载到数据库批量加载器。

加载器的输入是 stdin。我无法获得正确的语法来调用基于 unix 的批量加载器并传递要加载的 python 列表的内容。

我一直在阅读有关 Popen 和 PIPE 的内容，但它们的表现并不符合我的预期。

python 列表包含要批量加载的数据库记录。在 Linux 中，它看起来类似于：

echo "this is the string being written to the DB" | sql -c "COPY table FROM stdin"

What would be the true way replacement the echo statements with a python list to be use with this command ?

我没有此过程的示例代码，因为我一直在使用一些非常简单的语法尝试 Popen 和 PIPE 的功能，但没有获得所需的结果。

任何帮助将非常感激。

谢谢

Answer 1

如果您的数据短而简单，您可以预格式化整个列表并使用 subprocess 进行简单操作，如下所示：

import subprocess
data = ["list", "of", "stuff"]
proc = subprocess.Popen(["sql", "-c", "COPY table FROM stdin"], stdin=subprocess.PIPE)
proc.communicate("\n".join(data))

如果数据太大而无法像这样预格式化，那么您可以尝试直接使用 stdin 管道，尽管 subprocess 模块很不稳定使用管道时，如果您也需要从 stdout/stderr 读取数据。

for line in data:
    print >>proc.stdin, line