设计模式-访问者模式-visitor-python

def

封装一些作用于某种数据结构中的各元素的操作， 它可以在不改变数据结构的前提下定义作用于这些元素的新的操作。

usage

● 符合单一职责原则
具体元素角色也就是Employee抽象类的两个子类负责数据的加载， 而Visitor类则负责报表的展现， 两个不同的职责非常明确地分离开来， 各自演绎变化。
● 优秀的扩展性
由于职责分开， 继续增加对数据的操作是非常快捷的， 例如， 现在要增加一份给大老板的报表， 这份报表格式又有所不同， 直接在Visitor中增加一个方法， 传递数据后进行整理打印。
● 灵活性非常高
例如， 数据汇总， 就以刚刚我们说的Employee的例子， 如果我现在要统计所有员工的工资之和， 怎么计算？ 把所有人的工资for循环加一遍？ 是个办法， 那我再提个问题， 员工工资×1.2， 部门经理×1.4， 总经理×1.8， 然后把这些工资加起来， 你怎么处理？ 1.2， 1.4， 1.8是什么？ 不是吧？ ！ 你没看到领导不论什么时候都比你拿得多， 工资奖金就不说了， 就是过节发个慰问券也比你多， 就是这个系数在作祟。 我们继续说你想怎么统计？ 使用for循环， 然后使用instanceof来判断是员工还是经理？ 这可以解决， 但不是个好办法， 好办法是通过访问者模式来实现， 把数据扔给访问者， 由访问者来进行统计计算

code

class Node(object):
    pass


class A(Node):
    pass


class B(Node):
    pass


class C(A, B):
    pass


class Visitor(object):

    def visit(self, node, *args, **kwargs):
        meth = None
        for cls in node.__class__.__mro__:
            meth_name = 'visit_' + cls.__name__
            meth = getattr(self, meth_name, None)
            if meth:
                break

        if not meth:
            meth = self.generic_visit
        return meth(node, *args, **kwargs)

    def generic_visit(self, node, *args, **kwargs):
        print('generic_visit ' + node.__class__.__name__)

    def visit_B(self, node, *args, **kwargs):
        print('visit_B ' + node.__class__.__name__)


a = A()
b = B()
c = C()
visitor = Visitor()
visitor.visit(a)
visitor.visit(b)
visitor.visit(c)

### OUTPUT ###
# generic_visit A
# visit_B B
# visit_B C