如何从任何一个返回正确的类？

class ClassA:
    def __init__(self):
        self.name = "Class A"

    def method(self):
        print("This is Class A")

class ClassB:
    def __init__(self):
        self.name = "Class B"

    def method(self):
        print("This is Class B")

class ClassC:
    def __init__(self):
        self.name = "Class C"

    def method(self):
        print("This is Class C")

def get_class_instance(class_name):
    if class_name == "A":
        return ClassA()
    elif class_name == "B":
        return ClassB()
    elif class_name == "C":
        return ClassC()

# 调用工厂方法，传入正确的参数
instance = get_class_instance("A")
instance.method()  # 输出：This is Class A