Python：检查异常引发的位置

Question

采取这段代码：

def A():
   try:
      B()
   except Exception:
      pass

def B():
   C()

def C():
   print exception_handling_pointer()

A()

函数exception_handling_pointer应该返回一个指向函数的指针，在该函数中将首先检查此特定异常是否被处理。即，在这种情况下，我期望输出是……。例如：

<function A ...>

如何实现函数exception_handling_pointer？

Answer 1

如果不实际引发异常，您无法决定在何处处理异常。这很容易在这里看到：

try: 
    raise input('Raise which?')
except input('Catch which?') as e: 
    pass`

任何执行您想要的操作的函数都必须在此处预测用户输入。整个努力是徒劳的，Python 也不支持它。

无论如何，我希望你只是出于兴趣才问......

Answer 2

这是一件非常愚蠢的事情，大多数人会说这是不可能完成的（THC4k 为一般情况提供了令人信服的证据），但它听起来确实很有趣，并且在许多实际用例中应该是完全可行的。

第 1 步。你需要退一步穿过框架。使用 sys._getframe 或 inspect.currentframe 获取第一个（不要告诉任何人，第二个似乎是第一个的别名）。然后，您可以使用 f.f_back 第 2 步 迭代它们

。每个指令都有一个 f.f_lasti 指令。这是帧中执行的最后一条指令。你必须保存它。现在回溯字节码 - f.f_code.co_code - 并查找 SETUP_EXCEPT 操作码，其参数跳转到 f.f_lasti 之后 `。跳转点就是异常处理。

第 3 步。这就是事情变得更加模糊的地方。关键是实际的比较操作将是一个以 10 作为参数的 COMPARE_OP。在我见过的所有情况下，它后面都会跟着一个 POP_JUMP_IF_FALSE。这将跳转到下一个 except 子句或 finally 子句。其前面是加载异常到堆栈的代码。如果只有一个，那么它将是一个直接的 LOAD_GLOBAL 或 LOAD_GLOBAL 或 LOAD_FAST （取决于有例外的模块是全局的还是local) 后跟 LOAD_ATTR。如果匹配多个异常，则将出现一系列加载操作，后跟 BUILD_TUPLE （惯用）或 BUILD_LIST （其他一些奇怪或非惯用的情况）。

关键是您可以浏览 LOAD_X 指令并将名称与您匹配的异常进行比较。请注意，您仅比较名称。如果他们重新分配了名称，那么您就是 SOL。

第 4 步。假设您找到了匹配项。现在您需要函数对象。我能想到的最好方法如下（我保留更新的权利）：f.f_code 将具有 co_filename 属性。您可以循环遍历sys.modules，每个模块都将具有__name__属性。您可以比较两者，记住您应该使用 __name__.endswith(co_filename)。当您获得匹配项时，您可以循环遍历模块函数并将其 f.func_code.co_firstlineno 属性与框架 f.f_lineno 属性进行比较。当你获得匹配时，你就有了你的功能。您还应该循环访问模块中每个类的方法。处理可能发生在某些嵌套函数中，在这种情况下，我目前无法想到明智的做法。 （这将是一个完全不同的字节码黑客，并且本身就很不稳定）

第 5 步。利润。

这应该能让您大致了解如何进行此操作。在各种各样的极端情况下，您将无法做到这一点，但在任何正常的用例中，您应该能够完成它。如果您编写的代码依赖于能否完成它，那么它将会崩溃。这有点像“因为我能做所以做”之类的事情。