我可以编写一个函数来检测它是否是从“with”语句调用的吗？

Question

更具体地说，我可以检测函数是否在 with EXPR: BLOCK 语句中被调用为 EXPR 吗？ 我试图让自己熟悉 python 中 with 语句的用法。作为第一步，我重新实现了一个示例，该示例生成标记文本，出现在 contextlib.contextmanager 的参考（暂时忽略异常处理）。

class Markup(object):
    def __init__(self):
        self.tags = []
        self.append = self.tags.append
        self.pop = self.tags.pop

    def tag(self, name):
        self.append(name)
        return self

    def __enter__(self):
        print('<{}>'.format(self.tags[-1]))

    def __exit__(self, exc_type, exc_value, traceback):
        print('</{}>'.format(self.pop()))

>>> m = Markup()
>>> with m.tag('ELEMENT'):
...     print('sample text')
...
<ELEMENT>
sample text
</ELEMENT>

这按预期工作。然后，我必须考虑 tag() 是否也可以生成空元素：

>>> m = Markup()

# if a function appears as EXPR of "with EXPR: BLOCK", 'ELEMENT' is a container element. .
>>> with m.tag('ELEMENT'):
...     print('sample text')
...
<ELEMENT>
sample text
</ELEMENT> 

# in other cases, ELEMENT is an empty element.
>>> m.tag('ELEMENT')
<ELEMENT/>

在我天真的看来，如果被调用者可以检测到它是否是从 with 语句调用的，这似乎是可行的或不。但是，我不知道这种检测是否可能。有没有这样的方法，如果有的话，如何实现？

Answer 1

您实际上并没有从 with 语句“调用”tag()。您正在调用 tag()，然后将 tag() 的返回值传递到 with 语句，该语句随后调用 __enter__ 传入的值，然后运行正文，然后调用 __exit__。

所以不，在实际调用之前（在调用 tag() 之后），您无法检测到 with 语句。