是否可以制作依赖于上下文的功能？

Question

我使用以下脚本直接运行脚本，只是为了制作一个 bash 命令行以便在脚本之外运行它（例如作业调度程序）。

def qsubcommand(func):
    def wrapper(*args, **kwargs):
        if kwargs.get('test', False):
            cmdl = ' '.join(['this.py', func.__name__, *map(str, args)])
            return cmdl
        else:
            return func(*args, **kwargs)
    return wrapper

@qsubcommand
def calculate(value1, value2):
   # do something

if __name__ == '__main__':
    if len(sys.argv) > 1:
        func, args = sys.argv[1], sys.argv[2:]
        if func in locals().keys():
            locals()[func](*args)
        else:
            raise NotImplementedError

我有很多函数，比如“计算”。 我正在使用脚本来运行和测试程序。

# When I want to run directly:
>>> calculate(4, 5)

# When I want to just print command line:
>>> calculate(4, 5, test=True)
'this.py calculate 4 5'

但是，我想以与上下文相关的方式使用它，如下所示。

# When I want to run directly:
>>> test = False
>>> calculate(4, 5)

# When I want to just print command line:
>>> test = True
>>> calculate(4, 5)
'this.py calculate 4 5'

如何修改以让函数识别范围之外的变量。 是否可以访问函数外的变量？

提前感谢您的友好回答。

Answer 1

只需将其放在要检查变量的函数部分：

if 'test' in globals() and test:
    # do test
else:
    # do normal

函数总是可以访问函数范围之外的变量，如果你不使用 global 关键字，它们就无法编辑。