13

我正在尝试编写支持以下语义的代码：

with scope('action_name') as s:
  do_something()
  ...
do_some_other_stuff()

除其他事项（设置、清理）外，范围应决定此部分是否应运行。
例如，如果用户将程序配置为绕过“action_name”，那么在评估 Scope() 之后，do_some_other_stuff() 将在不先调用 do_something() 的情况下执行。
我尝试使用这个上下文管理器来做到这一点：

@contextmanager
def scope(action):
  if action != 'bypass':
    yield

但有RuntimeError: generator didn't yield异常（什么时候action）'bypass'。
我正在寻找一种方法来支持这一点，而不会退回到更冗长的可选实现： 

with scope('action_name') as s:
  if s.should_run():
    do_something()
    ...
do_some_other_stuff()

有谁知道我怎么能做到这一点？
谢谢！

PS我正在使用python2.7

编辑：
解决方案不一定要依赖with语句。没有它，我只是不知道如何表达它。本质上，我想要上下文形式的东西（支持设置和自动清理，与包含的逻辑无关）并允许基于传递给设置方法并在配置中选择的参数进行条件执行。
我还考虑了使用装饰器的可能解决方案。例子： 

@scope('action_name') # if 'action_name' in allowed actions, do:
                      #   setup()
                      #   do_action_name()
                      #   cleanup()
                      # otherwise return
def do_action_name()
  do_something()

但我不想基于这些范围强制执行过多的内部结构（即代码如何划分为函数）。
有没有人有一些创造性的想法？

4

4 回答 4

7

您正在尝试修改基本语言结构的预期行为。这绝不是一个好主意，它只会导致混乱。

您的解决方法没有任何问题，但您可以稍微简化一下。

@contextmanager 
def scope(action): 
  yield action != 'bypass'

with scope('action_name') as s: 
  if s: 
    do_something() 
    ... 
do_some_other_stuff()

相反，您scope可以是一个类，其__enter__方法返回一个有用的对象，或者None以相同的方式使用它。

于 2010-11-30T22:23:41.373 回答
4

以下似乎有效：

from contextlib import contextmanager

@contextmanager
def skippable():
    try:
        yield
    except RuntimeError as e:
        if e.message != "generator didn't yield":
            raise

@contextmanager
def context_if_condition():
    if False:
        yield True

with skippable(), context_if_condition() as ctx:
    print "won't run"

注意事项：

  • 需要有人想出更好的名字
  • context_if_condition没有就不能使用，skippable但没有办法强制执行/删除冗余
  • 它可以从比预期更深的函数中捕获并抑制 RuntimeError（自定义异常可能会有所帮助，但这会使整个构造更加混乱）
  • 这并不比仅使用@Mark Ransom 的版本更清晰
于 2016-10-29T12:17:51.147 回答
1

我不认为这是可以做到的。我尝试将上下文管理器实现为一个类，但无法强制块引发异常，该异常随后会被该__exit__()方法压制。

于 2010-11-30T14:19:45.287 回答
-2

我有与您相同的用例，并且遇到了自您发布问题以来有人帮助开发的条件库。

从该站点来看，其用途如下：

with conditional(CONDITION, CONTEXTMANAGER()):
    BODY()
于 2014-12-11T16:08:47.477 回答