保存python argparse文件

Question

Argparse具有一个称为的功能，该功能fromfile-prefix-chars似乎正好满足我所寻找的一半。如何从当前命令行参数为该功能创建文件？

我有一个带有解析器的复杂脚本，该解析器使代码构建的一部分自动化。用例是第一次正确设置命令行，然后基本上使用相同的参数重新运行。

使用文件并从中加载似乎是实现我所需要的一种好方法。似乎缺少的是一种将初始命令行写出到文件中的简单方法，因此现有文件解析将正常进行。

Answer 1

from argparse import ArgumentParser
import json

parser = ArgumentParser()
parser.add_argument('--seed', type=int, default=8)
parser.add_argument('--resume', type=str, default='a/b/c.ckpt')
parser.add_argument('--surgery', type=str, default='190', choices=['190', '417'])
args = parser.parse_args()

with open('commandline_args.txt', 'w') as f:
    json.dump(args.__dict__, f, indent=2)

parser = ArgumentParser()
args = parser.parse_args()
with open('commandline_args.txt', 'r') as f:
    args.__dict__ = json.load(f)

print(args)

Answer 2

文件格式记录得很清楚：

默认情况下，从文件中读取的参数每行必须为[...]

文档中的代码示例显示他们编写的文件之间使用换行符：

>>> with open('args.txt', 'w') as fp:
...     fp.write('-f\nbar')

如果要保存当前命令行，只需写入sys.argv[1:]文件，并在参数之间使用换行符即可：

with open('commandline_args.txt', 'w') as f:
    f.write('\n'.join(sys.argv[1:]))

演示：

>>> from argparse import ArgumentParser
>>> import sys
>>> parser = ArgumentParser(fromfile_prefix_chars='@')
>>> parser.add_argument('--foo')
_StoreAction(option_strings=['--foo'], dest='foo', nargs=None, const=None, default=None, type=None, choices=None, help=None, metavar=None)
>>> parser.add_argument('bar', nargs='?')
_StoreAction(option_strings=[], dest='bar', nargs='?', const=None, default=None, type=None, choices=None, help=None, metavar=None)
>>> sys.argv[1:] = ['--foo', 'spam', 'barbaz']
>>> parser.parse_args()
Namespace(bar='barbaz', foo='spam')
>>> with open('commandline_args.txt', 'w') as f:
...     f.write('\n'.join(sys.argv[1:]))
...
17
>>> parser.parse_args(['@commandline_args.txt'])
Namespace(bar='barbaz', foo='spam')
>>> sys.argv[1:] = ['--foo=spam', 'barbaz']  # using alternate syntax
>>> parser.parse_args()
Namespace(bar='barbaz', foo='spam')
>>> with open('commandline_args.txt', 'w') as f:
...     f.write('\n'.join(sys.argv[1:]))
...
17
>>> parser.parse_args(['@commandline_args.txt'])
Namespace(bar='barbaz', foo='spam')