如何在Python中将多个if-elif语句重构为策略模式？

编程语言 2026-07-10

问题： 我正在构建一个简单的数据处理工具。根据文件格式（JSON、XML或 CSV）需要应用不同的解析逻辑。现在，我在一个类中使用了一长串 if-elif 语句。随着要处理的格式增多，这个类变得难以维护，并且违反了单一职责原则。

研究进展： 我已了解了 策略模式 和 基于接口的设计。我知道应该把每种算法封装到一个独立的类中，但作为初学者，我在如何在Python中动态选择正确的策略方面遇到困难，并且担心在工厂里再引入一堆 "if-elif" 的混乱。

class DataProcessor:
    def process(self, format_type, data):
        if format_type == "json":
            return f"Parsing JSON: {data}"
        elif format_type == "xml":
            return f"Parsing XML: {data}"
        elif format_type == "csv":
            return f"Parsing CSV: {data}"
        else:
            raise ValueError("Unsupported format")

# Usage
processor = DataProcessor()
print(processor.process("json", "{'key': 'value'}"))

我想实现的目标： 我想把它转换成一个干净的架构，其中：

每个解析器都是一个独立的类/策略。
当新增格式时，DataProcessor 不需要修改。

我已尝试： 我尝试创建一个类字典，但我不确定这是否是实现策略模式的“Pythonic”做法，或者是否有使用ABC（抽象基类）的更标准方法。

我该如何正确实现以保持代码的可扩展性？

解决方案

总有需要做出选择的时候，这是不可避免的。不过这个思路如何：

FORMAT_REGISTRY = {}

class JsonProcessor:
    def process(data):
        return f"Parsing JSON: {data}"

FORMAT_REGISTRY['json'] = JsonProcessor

class XmlProcessor:
    def process(data):
        return f"Parsing XML: {data}"

FORMAT_REGISTRY['xml'] = XmlProcessor

class CsvProcessor:
    def process(data):
        return f"Parsing CSV: {data}"

FORMAT_REGISTRY['csv'] = CsvProcessor

# Usage
# 1. Get class by type
try:
    processor_cls = FORMAT_REGISTRY['json']
except KeyError:
    raise ValueError("Unsupported format")
# 2. Instantiate processor. Shared set of constructor arguments go here
processor = processor_cls()
# 3. process data
print(processor.process("{'key': 'value'}"))

如果全局注册表不适合你的用例，你可以添加一个工厂类。但要权衡，增加的复杂性是否比一个简单函数更值得。

class ProcessorFactory:
    def __init__(self):
        self.format_registry = {}
        self.add_type('json', JsonProcessor)
        self.add_type('xml', XmlProcessor)
        self.add_type('csv', CsvProcessor)

    def add_type(self, label, factory_functor):
        self.format_registry[label] = factory_functor

    def make_processor(self, label):
        try:
            factory_functor = self.format_registry[label]
        except KeyError:
            raise ValueError("Unsupported format")
        return factory_functor()

Python