如何实现基于参数类型的工厂模式？

# 定义接口或基类
class Product:
    def operation(self):
        pass

# 定义具体产品类
class ConcreteProductA(Product):
    def operation(self):
        print("ConcreteProductA operation")

class ConcreteProductB(Product):
    def operation(self):
        print("ConcreteProductB operation")

# 定义工厂类
class Factory:
    def create_product(self, product_type):
        if product_type == "A":
            return ConcreteProductA()
        elif product_type == "B":
            return ConcreteProductB()
        else:
            raise ValueError("Invalid product type")

# 客户端代码
factory = Factory()
product_a = factory.create_product("A")
product_a.operation()  # 输出：ConcreteProductA operation

product_b = factory.create_product("B")
product_b.operation()  # 输出：ConcreteProductB operation