python 匹配正则表达式返回组（如果存在）或为每个缺失组返回“无”

Posted 2021-05-10

import collections
import re
import sys

STRING = basestring if sys.version_info.major < 3 else str


def get_group(match_, group_id):
    try:
        return match_[group_id]
    except IndexError:
        pass


def match(regex, text, groups=0):
    result = regex.search(text)
    if result:
        is_sequence = not isinstance(groups, STRING) and collections.Sequence
        return ([get_group(i) for i in groups] if is_sequence
                else get_group(groups))