python中不同参数类型的方法重载

Question

我正在用python编写一个预处理器，其中的一部分用于AST。

有一个render()方法负责将各种语句转换为源代码。

现在，我有这样一个（简称）：

def render(self, s):
    """ Render a statement by type. """

    # code block (used in structures)
    if isinstance(s, S_Block):
        # delegate to private method that does the work
        return self._render_block(s)

    # empty statement
    if isinstance(s, S_Empty):
        return self._render_empty(s)

    # a function declaration
    if isinstance(s, S_Function):
        return self._render_function(s)

    # ...

如您所见，它很乏味，容易出错，而且代码很长（我有更多种类的语句）。

理想的解决方案是（在Java语法中）：

String render(S_Block s)
{
    // render block
}

String render(S_Empty s)
{
    // render empty statement
}

String render(S_Function s)
{
    // render function statement
}

// ...

当然，python不能这样做，因为它有动态类型。当我寻找如何模仿方法重载时，所有的答案都只是说“你不想在python中那样做”。我想这在某些情况下是正确的，但是这里kwargs一点用处都没有。

如果上面所示的类型检查ifs，那么在python中，如果没有长达一公里的可怕序列，我将如何做到这一点？另外，最好是用“Python式”的方法？

注意：可以有多个“Renderer”实现，它们以不同的方式呈现语句。因此，我不能将呈现代码移到语句中，而只调用s.render()。必须在renderer类中完成。

（我找到了一些interesting "visitor" code，但我不确定它是否真的是我想要的东西）。

Answer 1

如果您使用的是Python3.4（或者愿意为Python2.6+安装backport），那么可以使用^{}来完成以下操作：

from functools import singledispatch

class S_Block(object): pass
class S_Empty(object): pass
class S_Function(object): pass


class Test(object):
    def __init__(self):
        self.render = singledispatch(self.render)
        self.render.register(S_Block, self._render_block)
        self.render.register(S_Empty, self._render_empty)
        self.render.register(S_Function, self._render_function)

    def render(self, s):
        raise TypeError("This type isn't supported: {}".format(type(s)))

    def _render_block(self, s):
        print("render block")

    def _render_empty(self, s):
        print("render empty")

    def _render_function(self, s):
        print("render function")


if __name__ == "__main__":
    t = Test()
    b = S_Block()
    f = S_Function()
    e = S_Empty()
    t.render(b)
    t.render(f)
    t.render(e)

输出：

render block
render function
render empty

_{*基于this gist.的代码}