匹配 lib2to3 的 1-or-2-arg 函数调用的模式

Question

我有一个脚本可以重写 Python 模块，以便将所有出现的 func(a) 转换为 func2(a is None)。我现在也想支持 func(a, msg) 变成 func2(a is None, msg)，但我找不到能做到这一点的模式。以下显示了我的尝试：

from lib2to3 import refactor, fixer_base
from textwrap import dedent

PATTERN_ONE_ARG = """power< 'func' trailer< '(' arglist< obj1=any > ')' > >"""

PATTERN_ONE_OR_TWO_ARGS = """power< 'func' trailer< '(' arglist< obj1=any [',' obj2=any] > ')' > >"""


class TestFixer(fixer_base.BaseFix):
    def __init__(self, options, fixer_log):
        # self.PATTERN = PATTERN_ONE_ARG
        self.PATTERN = PATTERN_ONE_OR_TWO_ARGS

        super().__init__(options, fixer_log)

    def transform(self, node, results):
        print("found it")
        return node


class TestRefactoringTool(refactor.MultiprocessRefactoringTool):
    def get_fixers(self):
        fixer = TestFixer(self.options, self.fixer_log)
        return [fixer], []


def test():
    test_script = """
        log.print("hi")
        func(a, "12345")
        func(a, msg="12345")
        func(a)
        """
    refac.refactor_string(dedent(test_script), 'script')


flags = dict(print_function=True)
refac = TestRefactoringTool([], flags)
test()

对于在 test_script 字符串中找到的每个 func，我应该看到一个 "found it"，所以总共有 3 个，但我只看到打印了 2 个，这意味着 func(a) 没有被模式匹配器找到。我的模式基于 lib2to3.fixes 中可用的修复程序，但我一定遗漏了一个微妙之处。任何人都知道谁可以修复 PATTERN_ONE_OR_TWO_ARGS 以便找到所有 3 个函数？

我知道我可以创建一个单独的修复器实例，但是使用该模式可以使我免于编写大量代码（我有几十个这样的修复器，那么总数将是 24 个！）。

Answer 1

找到了：

PATTERN_ONE_OR_TWO_ARGS = """
    power< 'func' trailer< '('
        ( not(arglist | argument<any '=' any>) obj1=any
        | arglist< obj1=any ',' obj2=any > )
    ')' > >
    """

如果 transform() 是：

def transform(self, node, results):
    if 'obj2' in results:
        print("found 2", results['obj1'], results['obj2'])
    else:
        print("found 1", results['obj1'])
    return node

然后

test_script = """
    log.print("hi")
    func(a, "12345")
    func(a, msg="12345")
    func(a)
    func(k=a)
    """

输出是

found 2 a  "12345"
found 2 a  msg="12345"
found 1 a

我还发现 http://python3porting.com/fixers.html#fixers-chapter 这表明我可以覆盖 match() 方法而不是使用模式。感兴趣的四种模式是：

PATTERN_ONE_ARG_OR_KWARG   = """power< 'func' trailer< '(' not(arglist) obj1=any                         ')' > >"""
PATTERN_ONE_ARG            = """power< 'func' trailer< '(' not(arglist | argument<any '=' any>) obj1=any ')' > >"""
PATTERN_ONE_KWARG          = """power< 'func' trailer< '(' obj1=argument< any '=' any >                  ')' > >"""
PATTERN_TWO_ARGS_OR_KWARGS = """power< 'func' trailer< '(' arglist< obj1=any ',' obj2=any >              ')' > >"""

匹配 lib2to3 的 1-or-2-arg 函数调用的模式

pattern to match 1-or-2-arg function call for lib2to3

python

pattern-matching

python-2to3