如何重构Python中处理多格式颜色转HEX的嵌套try-except代码块

重构方案 #

我们可以通过策略模式把不同的颜色转换逻辑抽为可遍历的规则列表，彻底消除嵌套的try-except结构，同时保持原有逻辑完全兼容，扩展性也更强。

重构后代码 #

import svgwrite
from typing import Union, Tuple, Callable
from webcolors import name_to_hex, normalize_integer_triplet, rgb_percent_to_hex, normalize_hex, normalize_percent_triplet, rgb_to_hex
             
dwg = svgwrite.Drawing('qr.svg')

class Bit:
    def __init__(self, bit_color: Union[str, Tuple[int, int, int], Tuple[str, str, str]], spec:str='css3'):
        self.bit_type = 1
        self.spec = spec
        
        # 按优先级定义颜色转换策略，格式为(转换函数, 可忽略的异常类型)
        conversion_strategies: list[Tuple[Callable, tuple[type[Exception], ...]]] = [
            # 1. 命名色转HEX
            (lambda color: name_to_hex(color, spec=self.spec), (ValueError, AttributeError)),
            # 2. RGB整数三元组转HEX
            (lambda color: rgb_to_hex(normalize_integer_triplet(color)), (ValueError, TypeError)),
            # 3. RGB百分比三元组转HEX
            (lambda color: rgb_percent_to_hex(normalize_percent_triplet(color)), (ValueError,)),
            # 4. 标准化HEX格式
            (lambda color: normalize_hex(color), (ValueError,))
        ]
        
        for convert_func, allowed_exceptions in conversion_strategies:
            try:
                self.bit_color = convert_func(bit_color)
                break
            except allowed_exceptions:
                continue
        else:
            # 所有转换策略都失败时抛出异常
            raise ValueError('Color not recognized!')

优化点说明 #

• 完全消除嵌套结构，转换优先级一目了然，调整顺序只需要修改conversion_strategies的元素排列
• 扩展性更强，后续新增支持的颜色格式（比如HSL、RGBA转HEX）只需要在策略列表新增对应条目即可
• 修复了原代码中类型标注的错误：原参数bit_color标注为str，但实际支持三元组输入，新标注更准确
• 逻辑和原有实现完全等价，所有异常捕获规则、转换优先级都没有变化，不需要修改上层调用代码

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