使用上下文管理器作为函数

Question

我想创建一个函数，如果用with语句调用它也可以充当上下文管理器。 示例用法是：

# Use as function    
set_active_language("en")

# Use as context manager
with set_active_language("en"):
    ...

这与标准函数open的使用方式非常相似。

这是我想出的解决方案：

active_language = None  # global variable to store active language

class set_active_language(object):

    def __init__(self, language):
        global active_language
        self.previous_language = active_language
        active_language = language

    def __enter__(self):
        pass

    def __exit__(self, *args):
        global active_language
        active_language = self.previous_language

这段代码不是线程安全的，但这与问题无关。

我不喜欢这个解决方案的一点是，类构造函数伪装成一个简单的函数，仅用于它的副作用。

有一个更好的方法吗？

请注意，我尚未测试此解决方案。

更新：我不想将函数和上下文管理器拆分为单独的实体的原因是命名。 函数和上下文管理器基本上做同样的事情，所以对两者使用一个名称似乎是合理的。 如果我想将它分开，命名上下文处理器将是有问题的。 应该是什么？ active_language ？ 该名称可能（并且将会）与变量名称发生冲突。 override_active_language可能会起作用。

Answer 1

从技术上讲，不，你不能这样做。 但是你可以很好地伪造它，以至于人们（没有想太多）不会注意到。

def set_active_language(language):
    global active_language
    previous_language = active_language
    active_language = language

    class ActiveScope(object):
        def __enter__(self):
            pass

        def __exit__(self, *args):
            global active_language
            active_language = previous_language

    return ActiveScope()

当用作函数时， ActiveScope类只是一个稍微浪费的空操作。

Answer 2

希望有人能证明我是错的，但我认为答案是否定的：没有其他办法。 而且，您选择的方法的另一个缺点是，在 a with a, b, c:语句中与其他上下文管理器一起使用时，它可能会出现错误行为。 CM 的预期副作用是在对象构造上执行的，而不是像预期的那样在__enter__方法中执行。

为了能够做你想做的事情，你必须从类构造函数内部知道它是在with语句中初始化为上下文管理器，还是简单地作为函数调用。 据我所知，即使使用inspect模块也无法收集到它。