Python使用方法重載實現(xiàn)訪問者模式

更新時間：2024年12月17日 10:23:28 作者：三叔木卯

這篇文章主要為大家詳細介紹了Python如何使用方法重載實現(xiàn)訪問者模式,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下

Python上的訪問者模式，看了一下網上其他人的例子，一般都是類似下面的代碼。

from abc import ABC, abstractmethod  
  
# 抽象訪問者  
class AnimalVisitor(ABC):  
    @abstractmethod  
    def visit_dog(self, dog: "Dog"):  
        pass  
  
    @abstractmethod  
    def visit_cat(self, cat: "Cat"):  
        pass  
  
# 抽象動物類  
class Animal(ABC):  
    def __init__(self, name: str):  
        self.name = name  
  
    @abstractmethod  
    def accept(self, visitor: AnimalVisitor):  
        pass  
  
# 具體動物類  
class Dog(Animal):  
    def accept(self, visitor):  
        visitor.visit_dog(self)  
  
class Cat(Animal):  
    def accept(self, visitor):  
        visitor.visit_cat(self)  
  
# 具體訪問者：喂食訪問者  
class FeedingVisitor(AnimalVisitor):  
    def visit_dog(self, dog):  
        print(f"給{dog.name}喂狗糧")  
  
    def visit_cat(self, cat):  
        print(f"給{cat.name}喂貓糧")  
  
# 具體訪問者：檢查健康訪問者  
class HealthCheckVisitor(AnimalVisitor):  
    def visit_dog(self, dog):  
        print(f"檢查{dog.name}的疫苗接種情況")  
  
    def visit_cat(self, cat):  
        print(f"檢查{cat.name}是否需要洗澡")

使用示例如下：

def main():  
    # 創(chuàng)建動物  
    dog = Dog("旺財")  
    cat = Cat("咪咪")  
  
    # 創(chuàng)建訪問者  
    feeding_visitor = FeedingVisitor()  
    health_visitor = HealthCheckVisitor()  
  
    # 創(chuàng)建動物列表  
    animals = [dog, cat]  
  
    # 執(zhí)行不同的操作  
    print("=== 喂食時間 ===")  
    for animal in animals:  
        animal.accept(feeding_visitor)  
  
    print("\n=== 健康檢查 ===")  
    for animal in animals:  
        animal.accept(health_visitor)
  
  
if __name__ == "__main__":  
    main()

以上實現(xiàn)的訪問者模式，訪問者的接口類（抽象類）一般通過定義不同的方法（帶visit前綴的方法）來對不同的被訪問者進行訪問。有些奇怪，為什么不通過方法重載的方式來實現(xiàn)訪問者模式呢？

通過定義不同方法來實現(xiàn)訪問者模式，明顯違背一些設計原則的。訪問者的接口類需要負責訪問不同方法，而這些方法之間相關性不大（刪除某個方法對其他方法沒有影響），這明顯違背了單一職責原則（SRP）；如果要添加一種動物（被訪問者），那么就要修改訪問者接口類以添加相應的方法，這是違背開閉原則（OCP）的；訪問者的接口類的方法還依賴了具體類（比如Dog、Cat類），這違背了依賴倒轉原則（DIP）。

使用方法重載實現(xiàn)的訪問者模式，則沒有上面的問題，而且代碼也更簡單明了。Python沒有傳統(tǒng)的方法重載方式，不過在functools模塊里有個singledispatchmethod單分派裝飾器，這里可以借用它來實現(xiàn)“方法重載”。

使用方法重載的方式代碼如下：

from abc import ABC, abstractmethod  
from functools import singledispatchmethod  
  
# 抽象訪問者  
class AnimalVisitor(ABC):  
    @abstractmethod  
    def visit(self, animal: "Animal"):  
        pass  
  
# 抽象動物類  
class Animal(ABC):  
    def __init__(self, name: str):  
        self.name = name  
  
    def accept(self, visitor: AnimalVisitor):  
        visitor.visit(self)  
  
# 具體動物類  
class Dog(Animal): ...  
  
class Cat(Animal): ...  
  
# 具體訪問者：喂食訪問者  
class FeedingVisitor(AnimalVisitor):  
    @singledispatchmethod  
    def visit(self, animal: Animal):  
        raise NotImplementedError(f"{type(animal)} 未重載 visit方法")  
  
    @visit.register(Dog)  
    def _(self, dog):  
        print(f"給{dog.name}喂狗糧")  
  
    @visit.register(Cat)  
    def _(self, cat):  
        print(f"給{cat.name}喂貓糧")  
  
# 具體訪問者：檢查健康訪問者  
class HealthCheckVisitor(AnimalVisitor):  
  
    @singledispatchmethod  
    def visit(self, animal: Animal):  
        raise NotImplementedError(f"{type(animal)} 未重載 visit方法")  
  
    @visit.register(Dog)  
    def _(self, dog):  
        print(f"檢查{dog.name}的疫苗接種情況")  
  
    @visit.register(Cat)  
    def _(self, cat):  
        print(f"檢查{cat.name}是否需要洗澡")

可以看到，這一種方式訪問者抽象類AnimalVisitor的方法只有一個，就是visit。AnimalVisitor只負責一個職責，那就是訪問動物，符合單一職責原則；添加其他種動物的時候，不用修改抽象類，符合開閉原則；AnimalVisitor現(xiàn)在只依賴Animal這個抽象類，符合依賴倒轉原則。

有人可能擔心，如果訪問者抽象類沒有把訪問動物類的相應方法都列出來，會導致具體訪問者類漏實現(xiàn)一些方法重載。這個問題在上面的代碼中考慮到了，在singledispatchmethod裝飾的visit方法里使用NotImplementedError異常進行防御，如果某個具體動物類沒有重載visit方法，將拋出異常。

到此這篇關于Python使用方法重載實現(xiàn)訪問者模式的文章就介紹到這了,更多相關Python訪問者模式內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: