在 python 中，有没有办法制作一个行为像函数和上下文管理器的函数/类？

Question

在 python 中，有没有办法让函数/类表现得像函数和上下文管理器？

注意：我需要该函数/类来返回一个没有__exit__ 方法的对象，并且我无法更改该对象（这就是我包装它的原因）。

所以仅仅用__enter__ 和__exit__ 来创建一个类是行不通的，因为我需要它也像一个函数一样工作。

我已经尝试过contextmanager 装饰器：

@contextmanager
def my_context_man(my_str):
    my_str = 'begging ' + my_str
    yield my_str+' after func'
    print('end')

它在上下文管理器中完美运行，但不是作为函数：

a = 'middle'
old_a = my_context_man(a)
print('old_a', old_a)

with my_context_man(a) as new_a:
    print('new_a', new_a)

输出：

old_a <contextlib._GeneratorContextManager object at 0x0000000004F832E8>
new_a begging middle after func
end

而期望的输出将是：

old_a begging middle after func
new_a begging middle after func
end

编辑： 我遇到的具体问题是psycopg2 模块。 我想使用不同的上下文管理器。它返回一个连接对象。

def connect(dsn=None, connection_factory=None, cursor_factory=None, **kwargs):
    *the connection code*    
    return conn

我正在尝试对其进行更改，以便人们能够将它与我的上下文管理器一起使用，但不会破坏代码。 我无法更改 conn 对象

Answer 1

您的__new__ 方法没有返回my_context_man 的实例，而是str 的实例，并且str 没有__enter__ 方法。在with 语句中，__enter__ 的返回值与as 之后的名称绑定。你想要的

class my_context_man:
    def __init__(self, my_str):
        self.msg = my_str
        print("beginning " + my_str)

    def __enter__(self):
        return self.msg

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('end')