避免使用if else来实例化class - python

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

    def do_something(self):
        print("Doing something in ClassA")


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

    def do_something(self):
        print("Doing something in ClassB")


class ClassFactory:
    @staticmethod
    def create_class(class_name):
        if class_name == "ClassA":
            return ClassA()
        elif class_name == "ClassB":
            return ClassB()
        else:
            raise ValueError("Invalid class name")


# 调用方代码
class_name = "ClassA"
class_instance = ClassFactory.create_class(class_name)
class_instance.do_something()