python：使用正则表达式将文档字符串捕获为完全匹配

Question

假设我有以下功能

def func(self, arg1=None, arg2=None):
    """
    Returns the product for the specified arg1 and arg2.
    Args:
        arg1: argument 1
        arg2: argument 2
    Returns:
        product of arg1 for arg2.

    DAL Owner: Team IT
    Data Owner: Team A
    """
    return arg1*arg2

我只需要匹配三引号内的任何内容。我已经能够使用以下正则表达式将其作为一个组进行匹配

def func.*?\):.*?\"\"\"(.*)\"\"\"

但它不允许我在 python 中轻松地进行字符串替换。所以我需要将文档字符串捕获为匹配而不是组。我尝试了lookbehind (?<=def f2)，但随后遇到了lookbehind 问题中不允许的可变长度。任何有关如何解决此问题的想法都将不胜感激。

更新：

我真的很想知道这是否可以通过完全匹配来完成，即不必使用多个捕获组

Answer 1

如果您想仅匹配"""s 中的部分，一种选择是使用regex 模块，该模块支持\K 令牌，它可以有效地忘记所有内容那是以前匹配的。通过匹配（通常）def func 直到"""s 来利用它，然后使用\K 将匹配的起点重置为当前位置，然后延迟重复任何字符直到前瞻匹配另一个 """ ：

import regex
input  = '''
def func(self, arg1=None, arg2=None):
    """
    Returns the product for the specified arg1 and arg2.
    Args:
        arg1: argument 1
        arg2: argument 2
    Returns:
        product of arg1 for arg2.

    DAL Owner: Team IT
    Data Owner: Team A
    """
    return arg1*arg2
'''
replaced = regex.sub(r'(?s)def func\(.*?"""\K.*?(?=""")', 'replacement', input)
print(replaced)

输出：

def func(self, arg1=None, arg2=None):
    """replacement"""
    return arg1*arg2

\K，如果可用，可以更灵活地替代lookbehind。

Answer 2

如果您只想轻松替换，请将其他所有内容分组：

(def func.*?\):.*?\"\"\").*(\"\"\")

然后替换，你只需写$1xxx$2作为替换，xxx是你想要的实际替换。

试试here。