python - 使用上下文管理器控制流程

Question

我想知道在 python 中是否可以实现类似的功能(3.2，如果相关的话)。

with assign_match('(abc)(def)', 'abcdef') as (a, b):
    print(a, b)

行为在哪里:

• 如果正则表达式匹配，则将正则表达式组分配给 a 和 b
  • 如果那里不匹配，它会抛出异常
• 如果匹配为None，它会完全绕过上下文

我的目标基本上是一种极其简洁的方式来执行上下文行为。

我尝试制作以下上下文管理器:

import re

class assign_match(object):
    def __init__(self, regex, string):
        self.regex = regex
        self.string = string
    def __enter__(self):
        result = re.match(self.regex, self.string)
        if result is None:
            raise ValueError
        else:
            return result.groups()
    def __exit__(self, type, value, traceback):
        print(self, type, value, traceback) #testing purposes. not doing anything here.

with assign_match('(abc)(def)', 'abcdef') as (a, b):
    print(a, b) #prints abc def
with assign_match('(abc)g', 'abcdef') as (a, b): #raises ValueError
    print(a, b)

当正则表达式匹配时，它实际上完全按照预期工作，但是，如您所见，如果不匹配，它会抛出 ValueError。有什么办法可以让它“跳转”到退出序列吗？

谢谢!!

Answer 1

啊!也许在 SO 上解释它可以为我澄清问题。我只是使用 if 语句而不是 with 语句。

def assign_match(regex, string):
    match = re.match(regex, string)
    if match is None:
        raise StopIteration
    else:
        yield match.groups()

for a in assign_match('(abc)(def)', 'abcdef'):
    print(a)

给出了我想要的行为。把它留在这里以防其他人想从中受益。 (Mods，如果不相关，请随意删除等)

编辑:实际上，这个解决方案有一个相当大的缺陷。我在 for 循环中执行此行为。所以这阻止了我做:

for string in lots_of_strings:
    for a in assign_match('(abc)(def)', string):
        do_my_work()
        continue # breaks out of this for loop instead of the parent
    other_work() # behavior i want to skip if the match is successful

因为 continue 现在跳出这个循环而不是父 for 循环。如果有人有任何建议，我很乐意听取!

EDIT2:好吧，这次真的想通了。

from contextlib import contextmanager
import re

@contextmanager
def assign_match(regex, string):
    match = re.match(regex, string)
    if match:
        yield match.groups()

for i in range(3):
    with assign_match('(abc)(def)', 'abcdef') as a:
#    for a in assign_match('(abc)(def)', 'abcdef'):
        print(a)
        continue
    print(i)

抱歉发帖 - 我发誓在发帖之前我真的感觉卡住了。 :-) 希望其他人会发现这很有趣!