有条件地跳过Python With语句的主体

Question

我有一个With语句，如果某个<condition>满足，我想跳过它。就是我写,

With MyContext() as mc:
  do_something()

及

class MyContext(object):
...
def __enter__(self,...):
  if <condition>:
    JumpToExit()
def __exit__(self,...):
  print('goodbye')

我希望do_something()只在某些条件下执行，否则我希望JumpToExit()完全跳过主体，只完成块

谢谢

Answer 1

我不确定这是否可能__enter__是在with语句引入的try块之外执行的，因此我看不到直接从__enter__跳到__exit__的方法

with基本上（简化）是这样的：

with context as x:
    do_something()

进入

x = context.__enter__()
try:
    do_something()
finally:
    context.__exit__()

您需要在do_something()中抛出一个异常，或者成功完成它以进入__exit__。如果您确实从do_something()抛出异常，您可以在退出函数中（使用传递给它的一些参数）抑制异常，这样您就看不到它了。但是它必须是with块中的代码，它以某种方式导致跳转到__exit__

也许，如果您可以通过在__enter__中设置一些值来确保do_something立即抛出异常，那么您就可以使它正常工作。但对我来说，这听起来不是个好主意