0

我有 2 个文件，它们与子进程或 exec 一起runner.py运行。 它们都有命令行选项。target.py

如果 runner 使用子进程运行目标，则可以：

$ python runner.py
run target.py with subprocess...
target.py: running with dummy = False

如果 runner 使用exec运行目标代码（带有-e选项）：

$ python runner.py -e
run target.py with exec...
usage: runner.py [-h] [-d]
runner.py: error: unrecognized arguments: -e

命令行参数-e被target.py代码“看到”（只接受一个--dummy选项）并引发错误。

使用 exec 运行脚本时，如何将 args 隐藏到 argparse？

这是代码：

跑步者.py

import subprocess
import argparse


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-e", "--exec", help="run with exec", action="store_true")
    args = parser.parse_args()

    target_filename = "target.py"

    if args.exec:
        print("run target.py with exec...")
        source_code = open(target_filename).read()
        compiled = compile(source_code, filename=target_filename, mode="exec")
        exec(compiled) # OPTION 1 - error on argparse
        # exec(compiled, {}) # OPTION 2 - target does not go inside "if main"
        # exec(compiled, dict(__name__="__main__")) # OPTION 3 - same error as OPTION 1 
    else:
        print("run target.py with subprocess...")
        subprocess.run(["python3", target_filename])

我试图用上面注释的选项隐藏全局变量，但没有运气。
似乎与argparse的工作方式有关。

目标.py

import argparse

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-d", "--dummy", help="a dummy option", action="store_true")
    args = parser.parse_args()

    print(f"target.py: running with dummy = {args.dummy}")
4

1 回答 1

0

有 argparse conflict_handler选项，因此可以写入
argparse.ArgumentParser(conflict_handler='resolve')目标脚本。
resolve删除冲突的选项，但这不能很好地处理类似的情况，即选项在 runner 和 target 中具有相同的名称，或者您不能或不想更改目标文件的情况。

这是我找到的解决方案。
内部 argparse 用于sys.argv检索使用命令行设置的选项。

您可以直接设置sys.argv = [target_filename]which 删除选项，但更改sys会带来很多其他问题。

使用unittest.mock.patch(python3.4+)sys.argv可以像这样安全地更改：

from unittest.mock import patch

# [...]    

source_code = open(target_filename).read()
compiled = compile(source_code, filename=target_filename, mode="exec")

# remove command args
with patch('sys.argv', [target_filename]):
    exec(compiled)

因此，也可以使用以下选项运行目标脚本代码：

# run target
with patch('sys.argv', [target_filename]):  
    exec(compiled)

# run target with -d
with patch('sys.argv', [target_filename, "-d"]):
    exec(compiled)
于 2020-06-02T11:44:22.260 回答