在 Jupyter Notebook 中欺骗 argparse

Posted 2023-03-25

【中文标题】在 Jupyter Notebook 中欺骗 argparse

【英文标题】：Spoof argparse in Jupyter Notebook

【发布时间】：2020-11-08 11:58:39

【问题描述】：

我正在尝试在 Jupyter Notebook 中使用用 argparse 编写的函数。为此，我试图“欺骗”argparse，但不知道该怎么做。我找到了this useful SO question/answer，但我仍然对 argparse 有所了解。到目前为止，这是我所得到的：

import sys
import argparse

sys.argv = ['--config-file', "my_config"]


def argument_parser():

    parser = argparse.ArgumentParser()
    parser.add_argument("--config-file", default="", help="path to config file")
    parser.add_argument(
        "opts",
        help="Modify config options",
        default=None,
        nargs=argparse.REMAINDER,
    )
    return parser

然后当我打电话给argument_parser() 我得到

ArgumentParser(prog='--config-file', usage=None, description=None, formatter_class=<class 'argparse.HelpFormatter'>, conflict_handler='error', add_help=True)

然后我尝试解析参数：

args = argument_parser().parse_args()
print("Commandline Args:", args)

输出是：

Commandline Args: Namespace(config_file='', opts=['my_config'])

我的目标是得到这样的输出：

Commandline Args: Namespace(config_file='my_config')

我该怎么做？

【参考方案1】：

import sys
import argparse

sys.argv = ['--config-file','my_config']


def argument_parser():

    parser = argparse.ArgumentParser()
    parser.add_argument("--config-file", default="", help="path to config file")
    parser.add_argument(
        "opts",
        help="Modify config options",
        default=None,
        nargs=argparse.REMAINDER,
    )
    return parser

args = argument_parser().parse_args(sys.argv)
print("Commandline Args:", args)

您应该将sys.argv 传递给parse_args 方法。

输出：

Commandline Args: Namespace(config_file='my_config', opts=[])