有没有办法将标志传递给runpy？

Question

我正在寻找一种使用runpy或其他工具传递杂项选项的方法。

特别是，我想在另一个未优化的 python 脚本中获取优化的 python 脚本的输出。

python -O tobeoptimized.py

我曾尝试使用子进程，但我无法像在 runpy 中那样提取我需要的对象。

from subprocess import PIPE, run
command = ['python','-O','tobeoptimized.py']
result = run(command, stdout=PIPE, stderr=PIPE, universal_newlines=True)

Answer 1

不是，runpy因为它没有产生一个新进程，（exec()）。如果可以这样做，这个问题就会有答案。因此，我们几乎陷入了 C 代码库或扩展世界，其中CPython的核心被修改为至少可以从 C API 访问，如果不能从 Python API 访问。

而是尝试使用子进程来传递标志。可以调整输出。

如果问题是您想从进程中提取一个对象（因此子进程“失败”），您需要查看pickle模块或简单地将其放入文件/缓冲区/套接字并以自定义方式检索数据+重新组装.

Subprocess 并不是真正强制性的，但是如果有任何其他实现，它不会“破解它”，因为阻碍的是编译的核心，而不是可以被猴子修补的东西。

我想我们需要核心开发人员的新功能？:)

关于子流程输出，这对我来说很好用。

# hello.py
print("Hello")

# main.py
from subprocess import Popen, PIPE

process = Popen(
    ["python", "-O", "hello.py"],
    stdout=PIPE, stderr=PIPE, universal_newlines=True
)
print(process.stdout.read())

Answer 2

这是可行的。内置函数采用
优化参数。它的值为0、1和2。如果是，则类似于命令行上的without 、is和is 。 要优化运行模块/路径，必须对其进行修补。在此之前，我将定义两个函数进行说明。compile0-O1-O2-OO
runpy

这是一个测试模块。如果它在没有优化的情况下运行，它将不会打印。
应用程序.py

assert 0, "assert zero"
print("Optimized")

这两个函数并没有像以前那样完成所有细节runpy。所以不会完全发挥作用。

import importlib
import runpy

optimize = 1


def run_module(mod_name):
    spec = importlib.util.find_spec(mod_name)
    source = spec.loader.get_source(spec.name)
    code = compile(source, spec.name + ".py", "exec", optimize=optimize)
    d = {}
    exec(code, d)
    return d


def run_path(path):
    with open(path) as f:
        source = f.read()
    code = compile(source, path, "exec", optimize=optimize)
    d = {}
    exec(code, d)
    return d

这是修补一个函数，runpy该函数runpy.run_module用于获取要运行的模块的代码对象。该补丁提供了优化的代码对象。

import runpy

optimize = 1

def _get_module_details_wrapper(func):
    def tmp(*args, **kwargs):
        mod_name, spec, _ = func(*args, **kwargs)
        source = spec.loader.get_source(spec.name)
        code = compile(source, spec.name + ".py", "exec", optimize=optimize)
        return mod_name, spec, code

    return tmp

runpy._get_module_details = _get_module_details_wrapper(runpy._get_module_details)
runpy.run_module('app')

UPDATE
runpy.run_path可以在打开优化的情况下运行。

def optimize_compile(o):
    from functools import partial
    import builtins

    builtins.compile = partial(compile, optimize=o)


optimize_compile(1)
runpy.run_path("app.py")


optimize_compile(0)
try:
    runpy.run_path("app.py")
except AssertionError:
    print("assertion")

optimize_compile(1)
runpy.run_path("app.py")