Question

我必须创建一个可以与Linux管道一起使用的python脚本

我想运行一个脚本，其中一些参数可以通过管道或同一行发送

将脚本与预期输出一起使用的一些示例：

echo "a" > list.txt
echo "b" >> list.txt

./run.py p1 p2   # ['p1', 'p2'] expected output
cat list.txt | ./run.py  # ['a', 'b'] expected output
cat list.txt | ./run.py p1 p2 # ['p1', 'p2', 'a', 'b'] expected output

我试过了：

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('args', nargs=argparse.REMAINDER)
args = parser.parse_args().args
print args

它仅适用于同一行中的参数：

./run.py p1 p2  #['p1', 'p2'] OK
cat list.txt | ./run.py  # []  Not OK
cat list.txt | ./run.py p1 p2 # ['p1', 'p2'] expected output

Answer 1

我发现xargs在这种情况下很有用。

我没有尝试过自己，但也许

cat list.txt | xargs ./run.py p1 p2

适合你吗？

如果您需要特定参数的位置，可以使用xargs占位符选项-J：

cat list.txt | xargs -J{} ./run.py p1 {} p2

将＆＃34; a b＆＃34;介于＆＃34; p1＆＃34;和＆＃34; p2＆＃34;。

Answer 2

仅使用argparse

的解决方案

import argparse
import sys

parser = argparse.ArgumentParser()
parser.add_argument('args', nargs=argparse.REMAINDER)
parser.add_argument('stdin', nargs='?', type=argparse.FileType('r'), default=sys.stdin)
args = parser.parse_args().args

if not sys.stdin.isatty():
    stdin = parser.parse_args().stdin.read().splitlines()
else:
    stdin = []

print(args + stdin)

nargs='?'使stdin可选，sys.stdin.isatty()检查sys.stdin是否为空

如何使用argparse通过python脚本中的管道发送参数？

2 个答案: