如何使用 Python 中的装饰器模式根据字典中的数据向 class 动态添加方法？

Question

考虑这样一种情况，您有一个基本的 class Animal ，并且您希望根据存储在字典中的数据向 class 添加新方法。 字典中的数据将方法名映射到函数，你想使用装饰器模式将这些方法动态添加到class。

例如，假设字典定义如下：

methods = {
    "walk": lambda self: print(f"{self.name} is walking"),
    "run": lambda self: print(f"{self.name} is running"),
}

基数 class 定义为：

class Animal:
    def __init__(self, name):
        self.name = name

如何使用装饰器模式将字典中定义的方法动态添加到Animal class 中？

我尝试使用标准装饰器向class添加方法，但这种方法一次只能添加一个方法，无法根据字典中的数据动态添加多个方法。

我希望找到一种方法，根据字典中的数据，将方法动态添加到 class 中，这样我就可以在一行代码中将多个方法添加到 class 中。 我一直在寻找一种使用装饰器模式的解决方案，因为它是向 Python 中的类添加行为的常用方法。

Answer 1

methods = {
    "walk": lambda self: print(f"{self.name} is walking"),
    "run": lambda self: print(f"{self.name} is running"),
}


def my_decorator(cls: type):
    for name, method in methods.items():
        setattr(cls, name, method)
    return cls


@my_decorator
class Animal:
    def __init__(self, name):
        self.name = name


animal = Animal("name")
animal.run()

替代方案（装饰工厂）

from typing import Callable


methods = {
    "walk": lambda self: print(f"{self.name} is walking"),
    "run": lambda self: print(f"{self.name} is running"),
}


def my_decorator(methods: dict[str, Callable]):
    def decorator(cls: type):
        for name, method in methods.items():
            setattr(cls, name, method)
        return cls
    return decorator


@my_decorator(methods)
class Animal:
    def __init__(self, name):
        self.name = name


animal = Animal("name")
animal.run()