迹忆客 专注技术分享

一个简单的函数，其目的是创建一个类并返回它，称为类工厂。 类工厂作为强大的模式之一，在 Python 中被广泛使用。

本篇文章介绍了创建类工厂的不同方法。

如何在 Python 中创建类工厂

有两种设计类工厂的方法； 一个在编码时创建一个类，而另一个在运行时创建一个类。

前者使用 class 关键字，后者使用 type 关键字。 这两种方法在下面的文章中进行了解释和对比。

使用 class 关键字在 Python 中创建类工厂

我们可以使用 class 关键字创建一个类工厂。 为此，我们需要创建一个函数并在函数定义中包含一个类。

以下代码使用 class 关键字在 Python 中创建类工厂。

def ballfun():
    class Ball(object):
        def __init__(self, color):
            self.color = color
        def getColor(self):
            return self.color
    return Ball
Ball = ballfun()
ballObj = Ball('green')
print(ballObj.getColor())

上面的代码提供了以下输出。

green

使用 type 关键字在 Python 中创建类工厂

type 关键字允许动态创建类。 我们需要利用 type 关键字在 Python 中创建一个类工厂。

但是，我们应该注意，使用 type 关键字时，函数将只留在名称空间中，与类一起。

以下代码使用 type 关键字在 Python 中创建动态类。

def init(self, color):
    self.color = color
def getColor(self):
    return self.color
Ball = type('Ball', (object,), {
    '__init__': init,
    'getColor': getColor,
})
ballGreen = Ball(color='green')
print(ballGreen.getColor())

上面的代码提供了以下输出。

green

type 关键字允许动态类和在运行时有效创建和缺点。 正如您在上面的代码中看到的，init 和 getColor 函数都在命名空间中杂乱无章。

此外，当使用 type 关键字创建动态类时，这些函数的可重用性前景也会丢失。

一个简单的解决方案是引入类工厂。 它对两种方式都有帮助，因为它减少了代码的混乱并提高了函数的可重用性。

以下代码使用 type 关键字在 Python 中创建类工厂。

def create_ball_class():
    def init(self, color):
        self.color = color
    def getColor(self):
        return self.color
    return type('Ball', (object,), {
        '__init__': init,
        'getColor': getColor,
    })
Ball = create_ball_class()
ballObj = Ball('green')
print(ballObj.getColor())

上面的代码提供了以下输出。

green

既然我们已经了解了如何创建类工厂，那么区分何时使用和何时不使用新学习的类工厂概念也很重要。

通常，当我们不知道在编码发生时要分配哪些属性时，类工厂很有用。