python - 你如何将这个正则表达式习语从 Perl 翻译成 Python？

Question

大约一年前我从 Perl 切换到 Python 并且没有回头。我发现只有 一个 成语我在 Perl 中比在 Python 中更容易做到:

if ($var =~ /foo(.+)/) {
  # do something with $1
} elsif ($var =~ /bar(.+)/) {
  # do something with $1
} elsif ($var =~ /baz(.+)/) {
  # do something with $1
}

相应的 Python 代码并不那么优雅，因为 if 语句不断嵌套:

m = re.search(r'foo(.+)', var)
if m:
  # do something with m.group(1)
else:
  m = re.search(r'bar(.+)', var)
  if m:
    # do something with m.group(1)
  else:
    m = re.search(r'baz(.+)', var)
    if m:
      # do something with m.group(2)

有没有人有一种优雅的方式在 Python 中重现这种模式？我见过使用匿名函数调度表，但对于少数正则表达式，这些表对我来说似乎有点笨拙......

Answer 1

使用命名组和调度表:

r = re.compile(r'(?P<cmd>foo|bar|baz)(?P<data>.+)')

def do_foo(data):
    ...

def do_bar(data):
    ...

def do_baz(data):
    ...

dispatch = {
    'foo': do_foo,
    'bar': do_bar,
    'baz': do_baz,
}


m = r.match(var)
if m:
    dispatch[m.group('cmd')](m.group('data'))

通过一些自省(introspection)，您可以自动生成正则表达式和调度表。