Python argparse 与 stdin 互斥是选项之一 [英] Python argparse mutually exclusive with stdin being one of the options
本文介绍了Python argparse 与 stdin 互斥是选项之一的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我希望我的脚本接收这些互斥的输入选项:
I would like my script to receive these mutually exclusive input options:
- 一个包含 JSON 的输入文件 (
script.py -i input.json
); - 一个包含 JSON 的字符串 (
script.py '{"a":1}'
); - 来自标准输入的 JSON(
echo '{"a":1}' | script.py
或cat input.json | script.py
).
- an input file containing a JSON (
script.py -i input.json
); - a string containing a JSON (
script.py '{"a":1}'
); - a JSON from stdin (
echo '{"a":1}' | script.py
orcat input.json | script.py
).
和这些互斥的输出选项:
- 一个包含 JSON 的输出文件;
- 标准输出中的 JSON.
所以我尝试使用此代码
import json,sys,argparse
parser = argparse.ArgumentParser(description='Template for python script managing JSON as input/output format')
group = parser.add_mutually_exclusive_group()
group.add_argument('--input-file', '-i', type=str, help='Input file name containing a valid JSON.', default=sys.stdin)
group.add_argument('json', nargs='?', type=str, help='Input string containing a valid JSON.' , default=sys.stdin)
parser.add_argument('--output-file', '-o',type=str, help='Output file name.')
args = parser.parse_args()
if not sys.stdin.isatty():
data = sys.stdin.read()
else:
# args = parser.parse_args()
if args.input_file :
data=open(args.input_file).read()
elif args.json :
data=args.json
datain=json.loads(data)
dataout=json.dumps(datain, indent=2)
if args.output_file :
output_file=open(args.output_file, 'w')
output_file.write(dataout+'\n')
output_file.close()
else:
print (dataout)
但它不适用于 stdin,因为它至少需要两个 group
选项之一.
But it does not work with stdin as it requires at least one of the two group
options.
如何在输入选项列表中添加标准输入?
How can I add stdin in the list of input options?
如果我这样称呼它,添加 default=sys.stdin
参数是可行的
Adding the default=sys.stdin
argument works if I call it like that
echo '{}' | ./script.py -
但不是那样:
echo '{}' | ./script.py
推荐答案
我会利用 argparse.FileType
和 sys.stdin
的默认值.>
I would take advantage of argparse.FileType
with a default value of sys.stdin
.
import json,sys,argparse
parser = argparse.ArgumentParser(description='Template for python script managing JSON as input/output format')
group = parser.add_mutually_exclusive_group()
group.add_argument(
'--input-file', '-i',
type=argparse.FileType('r'),
default=sys.stdin,
help='Input file name containing a valid JSON.')
group.add_argument(
'json',
nargs='?',
type=str,
help='Input string containing a valid JSON.')
parser.add_argument(
'--output-file', '-o',
type=argparse.FileType('w'),
help='Output file name.',
default=sys.stdout)
args = parser.parse_args()
data = args.json or args.input_file.read()
datain=json.loads(data)
dataout=json.dumps(datain, indent=2)
args.output_file.write(dataout)
这篇关于Python argparse 与 stdin 互斥是选项之一的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文