如何重构Python多if-elif-else语句的函数以降低认知复杂度

针对你给出的代码，最简洁易维护的重构方式是使用字典映射替代嵌套的if-elif-else结构，核心思路是把重复的条件判断转化为键值对查找，既消除重复代码，也让后续维护更方便。

重构后的代码（保持原逻辑） #

def get_uid(uid_type, key):
    # 定义类型与前缀的映射关系
    TYPE_PREFIX_MAP = {
        'orange_fruit': 'zof',
        'grape': 'zzg',
        'kiwi_healthy': 'zki',
        'anar_tasty': 'zat',
        'apple': 'zpp',
        'cherry': 'zch'
    }
    # 定义key与后缀的映射关系，默认后缀为'x'
    KEY_SUFFIX_MAP = {'juice': 'j', 'salad': 's'}
    
    # 获取当前类型对应的前缀
    prefix = TYPE_PREFIX_MAP.get(uid_type)
    if prefix:
        suffix = KEY_SUFFIX_MAP.get(key, 'x')
        # 处理anar_tasty的特殊else分支
        if uid_type == 'anar_tasty' and suffix == 'x':
            uid = 'yx' + uuidV().hex[:X]
        else:
            uid = f"{prefix}{suffix}" + uuidV().hex[:X]
        return uid
    
    # 类型不在映射中的默认返回
    return uuidV().hex[:Y]

进一步优化：减少重复的uuid调用 #

原代码中每个分支都会调用uuidV()生成新的随机片段，如果业务允许同一个UID使用同一段随机字符串，可以提前生成一次，提升效率：

def get_uid(uid_type, key):
    TYPE_PREFIX_MAP = {
        'orange_fruit': 'zof',
        'grape': 'zzg',
        'kiwi_healthy': 'zki',
        'anar_tasty': 'zat',
        'apple': 'zpp',
        'cherry': 'zch'
    }
    KEY_SUFFIX_MAP = {'juice': 'j', 'salad': 's'}
    
    # 提前生成随机片段，避免多次调用uuidV()
    uuid_segment = uuidV().hex[:X]
    
    prefix = TYPE_PREFIX_MAP.get(uid_type)
    if prefix:
        suffix = KEY_SUFFIX_MAP.get(key, 'x')
        if uid_type == 'anar_tasty' and suffix == 'x':
            return 'yx' + uuid_segment
        return f"{prefix}{suffix}" + uuid_segment
    
    return uuidV().hex[:Y]

额外的最佳实践 #

原函数参数type是Python的内置关键字，建议修改为uid_type之类的名称，避免语法冲突和代码歧义，上面的重构代码已经做了这个调整。

重构的优势 #

• 消除重复代码：把大量重复的嵌套判断转化为字典查找，代码量大幅减少
• 易于维护：新增类型或key规则时，只需修改映射字典，无需添加新的if-elif分支
• 可读性提升：直接通过字典就能清晰看到所有类型和key对应的前缀/后缀规则

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