访问者模式(Visitor)及代码实现

示例代码部分:

# -*- coding: utf-8 -*-
"""
(C) rgc
All rights reserved
create time '2020/12/29 19:31'

Usage:
访问者模式示例
购买各种礼物时 顾客和店主 对礼物的处理不同;顾客考虑礼物价格是否划算,大小是否合适,店主考虑礼物 根据价格和大小 应该怎么包装才合适
这个需求中 各种礼物是 元素;顾客和店主 是访问者;
假如现在有2种礼物,则顾客和店主需要 共需要实现 2*2=4种访问方法
   现在增加到了5种礼物,则顾客和店主需要 共需要实现 2*5=10种方法方法
因为顾客和店主 对每种礼物 都要分别进行相应操作;
"""
from abc import ABC, abstractmethod


class Gift(ABC):
    """
    礼物基类
    抽象元素角色
    """

    def __init__(self, price, height):
        self.price = price
        self.height = height

    @abstractmethod
    def accept(self, visitor):
        """

        :param visitor:
        :return:
        """
        pass


class Lighter(Gift):
    """
    打火机类
    具体元素角色
    """

    def accept(self, visitor):
        """

        :param visitor:
        :return:
        """
        return visitor.visit_lighter(self)


class Toys(Gift):
    """
    玩具类
    具体元素角色
    """

    def accept(self, visitor):
        """

        :param visitor:
        :return:
        """
        return visitor.visit_toys(self)


class Cup(Gift):
    """
    杯子类
    具体元素角色
    """

    def accept(self, visitor):
        """

        :param visitor:
        :return:
        """
        return visitor.visit_cup(self)


class Person(ABC):
    """
    人 类
    抽象访问者角色
    """

    @abstractmethod
    def visit_lighter(self, element: Lighter):
        """
        对 打火机礼物进行的操作
        :param element:
        :return:
        """
        pass

    @abstractmethod
    def visit_toys(self, element: Toys):
        """
        对 玩具礼物进行的操作
        :param element:
        :return:
        """
        pass

    @abstractmethod
    def visit_cup(self, element: Cup):
        """
        对 杯子 礼物进行的操作
        :param element:
        :return:
        """
        pass


class Customer(Person):
    """
    顾客 类
    具体访问者角色
    """

    def visit_lighter(self, element: Lighter):
        """
        对 打火机礼物进行的操作
        :param element:
        :return:
        """
        if element.height < 7:
            print('打火机大小还行')
        else:
            print('打火机太大了,不方便')
        if element.price > 2000:
            print('什么破打火机怎么这么贵')
        else:
            print('嗯...,这个打火机还行')
            return True
        return False

    def visit_toys(self, element: Toys):
        """
        对 玩具礼物进行的操作
        :param element:
        :return:
        """
        print('玩具不在乎高度问题')
        if element.price > 2000:
            print('天啊,这个玩具好贵啊')
            return False
        print('小意思,这个玩具可以考虑')
        return True

    def visit_cup(self, element: Cup):
        """
        对 杯子 礼物进行的操作
        :param element:
        :return:
        """
        print('杯子不考虑高度问题')
        if 100 < element.price < 300:
            print('这个杯子可以,不贵也不便宜')
            return True
        print('不考虑这个杯子')
        return False


class Clerk(Person):
    """
    店员 类
    具体访问者角色
    """

    def visit_lighter(self, element: Lighter):
        """
        对 打火机礼物进行的操作
        :param element:
        :return:
        """
        if element.height > 6:
            print('这个打火机要用长点的盒子包装')
        else:
            print('这个打火机用小盒子包装就行了')

    def visit_toys(self, element: Toys):
        """
        对 玩具礼物进行的操作
        :param element:
        :return:
        """
        if element.price > 1000:
            print('这个玩具挺贵的,用高档材料包装吧')
        else:
            print('这个玩具价格一般,随便包装一下吧')

    def visit_cup(self, element: Cup):
        """
        对 杯子 礼物进行的操作
        :param element:
        :return:
        """
        if element.height > 10:
            print('这个杯子有点大,用箱子包装吧')
        else:
            print('挺小一杯子,用小盒子吧,别浪费资源')


if __name__ == '__main__':
    # 各种礼物初始化
    gifts = [Lighter(3000, 6), Toys(2000, 50), Cup(200, 10)]

    print('客户挑选礼物')
    customer_visitor = Customer()
    # 礼物列表
    choose_gift_list = []
    # 查看每个礼物
    for gift in gifts:
        # 把心仪的礼物放到礼物列表中
        if gift.accept(customer_visitor):
            choose_gift_list.append(gift)

    print('*' * 10)

    print(f'店员看到到顾客选择了{len(choose_gift_list)}个礼物,现在开始包装')
    clerk_visitor = Clerk()
    # 店员对顾客选择的礼物进行包装
    for gift in choose_gift_list:
        gift.accept(clerk_visitor)

运行结果:

总结: 

注意 在访问者类 中操作 元素对象时,除非需要,否则尽量不要修改现有对象属性值(可以通过 原型模式进行克隆),防止 影响后续操作;

这种模式 打破了 程序员编写代码时的思想禁锢, 一般都是 数据对象增加新功能时 都是在类中直接增加新的方法; 而此模式 将 新功能放在单独的类中;

相关链接:

https://refactoringguru.cn/design-patterns/visitor

https://refactoringguru.cn/design-patterns/visitor/python/example

http://c.biancheng.net/view/1397.html

https://www.cnblogs.com/liuqingzheng/articles/10039702.html