Python工厂方法优化咨询：多CSV类按文件名实例化的改进方案

需要处理多个周期性CSV对象，每个CSV因属性不同对应独立类，仅能通过文件名区分类型。当前采用如下工厂方法按文件名实例化对象：

class CSVFileFactory:
     @staticmethod
     def obj_builder(fname: str):
         if "type1" in fname:
             return TypeOneCSV(filename=fname)
         elif "type2" in fname:
             return TypeTwoCSV(filename=fname)
         elif "type3" in fname:
             return TypeThreeCSV(filename=fname)
         ...
         ...

但存在大量if-elif分支。请问这是否为最简洁的实现方式？有哪些未遵循的规范？如何降低代码复杂度、优化冗余分支？

是否为最简洁的实现方式？ #

显然不是。这种写法在CSV类型较少时勉强可用，但随着类型增多，代码会快速臃肿，维护成本陡增——新增或修改类型必须改动obj_builder方法，完全不符合扩展性要求。

未遵循的规范/设计原则 #

• 开闭原则：对扩展开放、对修改关闭是面向对象设计的核心原则之一，当前实现新增类型必须修改工厂方法代码，直接违反了这一点。
• 单一职责原则：obj_builder同时承担了类型判断、实例化逻辑，职责过重，一旦某段分支出错，整个工厂方法都会受影响。
• 可读性与可维护性：大量分支会让代码变得冗长，后续开发者需要逐行梳理才能理清类型映射关系，排查问题效率极低。

优化方案：用映射表替代if-elif分支 #

最直接有效的优化是将「文件名标识-对应类」的映射关系抽离为字典，通过查找字典完成实例化，彻底消除分支逻辑。

实现方式1：静态映射表（适合固定类型场景） #

class CSVFileFactory:
    # 集中管理类型映射，新增类型仅需添加键值对
    TYPE_MAPPING = {
        "type1": TypeOneCSV,
        "type2": TypeTwoCSV,
        "type3": TypeThreeCSV,
    }

    @staticmethod
    def obj_builder(fname: str):
        for identifier, cls in CSVFileFactory.TYPE_MAPPING.items():
            if identifier in fname:
                return cls(filename=fname)
        # 处理未匹配到的情况，可根据业务需求调整（如返回默认对象或抛出异常）
        raise ValueError(f"Unsupported CSV type for file: {fname}")

优势：

• 新增类型无需修改obj_builder逻辑，仅需在映射表中添加条目，符合开闭原则。
• 映射关系集中可见，可读性和维护性大幅提升。

实现方式2：自动注册机制（适合动态扩展场景） #

如果需要支持从外部模块动态导入CSV类型，可以实现注册装饰器，让每个CSV类自行完成注册：

class CSVFileFactory:
    TYPE_MAPPING = {}

    @staticmethod
    def register(identifier):
        def decorator(cls):
            CSVFileFactory.TYPE_MAPPING[identifier] = cls
            return cls
        return decorator

    @staticmethod
    def obj_builder(fname: str):
        for identifier, cls in CSVFileFactory.TYPE_MAPPING.items():
            if identifier in fname:
                return cls(filename=fname)
        raise ValueError(f"Unsupported CSV type for file: {fname}")

# 用装饰器注册CSV类
@CSVFileFactory.register("type1")
class TypeOneCSV:
    def __init__(self, filename):
        self.filename = filename

@CSVFileFactory.register("type2")
class TypeTwoCSV:
    def __init__(self, filename):
        self.filename = filename

优势：

• 类的注册逻辑与类定义绑定，无需集中修改工厂映射表，模块解耦性更好。
• 支持插件式动态加载类，扩展性更强。

额外优化：处理多标识冲突 #

如果存在文件名包含多个标识的情况（如type1_type2.csv），可以在映射表中按优先级排序，确保先匹配高优先级的标识，或在注册时指定优先级参数，匹配时按优先级顺序查找。

内容的提问来源于stack exchange，提问作者Marco