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)

对不起该帖子 - 我发誓，在发布之前，我感到非常困惑。 :-)希望别人会觉得这很有趣！

使用Context Manager进行控制流程

1 个答案: