什么是工厂函数(Factory Function)

什么是工厂函数(Factory Function)

不需要直接创建class,就可以引用class的function

·

1 min read

工厂函数指不需要直接创建class,就可以引用class的function。

举个例子:

假设有一个表示“交通工具”的场景。

使用类的构造函数

class Car:
    def __init__(self, brand):
        self.brand = brand

class Bike:
    def __init__(self, type):
        self.type = type

# 创建对象
my_car = Car("Toyota")
my_bike = Bike("Mountain")

在这个例子中,我们直接使用各自的构造函数来创建CarBike对象。

使用工厂函数

class Car:
    def __init__(self, brand):
        self.brand = brand

class Bike:
    def __init__(self, type):
        self.type = type

def vehicle_factory(vehicle_type, attribute):
    if vehicle_type == "car":
        return Car(attribute)
    elif vehicle_type == "bike":
        return Bike(attribute)

# 创建对象
my_vehicle1 = vehicle_factory("car", "Toyota")
my_vehicle2 = vehicle_factory("bike", "Mountain")

在这个例子中,vehicle_factory是一个工厂函数,根据传入的vehicle_type参数来决定创建Car还是Bike对象。这种方式更加灵活和抽象,因为调用者不需要直接与具体的CarBike类打交道。