Python代码优化：解决Pylint对多if-raise语句的报错问题

问题场景 #

我编写了以下Python代码用于参数校验与计算：

if a1 in list1:
    if b1 == 1 and c1>1:
        raise Exception('invalid c1 for b1')
    elif b1 == 2 and c1>2:
        raise Exception('invalid c1 for b1')
    elif b1 == 3 and c1>3:
        raise Exception('invalid c1 for b1')
    else:
        z = c1 * d * f
else:
    raise Exception('invalid a1')

运行Pylint检测时收到报错：Unnecessary "elif" after "raise". change "elif" to "if"

按提示修改为以下代码后：

if a1 in list1:
    if b1 == 1 and c1>1:
        raise Exception('invalid c1 for b1')
    if b1 == 2 and c1>2:
        raise Exception('invalid c1 for b1')
    if b1 == 3 and c1>3:
        raise Exception('invalid c1 for b1')
    else:
        z = c1 * d * f
else:
    raise Exception('invalid a1')

Pylint又提示：Unnecessary "else" after "raise", remove the "else" and de-indent the code inside it

我不想直接移除else（两个else对应不同校验场景），也不想通过#pylint disable=...禁用规则，希望找到更合理的代码实现方式。

优化方案 #

方案1：提前校验，扁平化逻辑

将异常触发的逻辑提前，避免嵌套else，同时让代码逻辑更清晰：

if a1 not in list1:
    raise Exception('invalid a1')

if b1 == 1 and c1 > 1:
    raise Exception('invalid c1 for b1')
if b1 == 2 and c1 > 2:
    raise Exception('invalid c1 for b1')
if b1 == 3 and c1 > 3:
    raise Exception('invalid c1 for b1')

z = c1 * d * f

这种方式把外层的else转成提前校验，内层的else直接取消——只要前面所有校验都不触发异常，就会执行计算逻辑，和原代码逻辑完全一致，且符合Pylint的代码规范。

方案2：用字典映射简化校验逻辑

如果b1的取值是固定的有限集合，可以用字典映射阈值，减少重复代码的同时规避报错：

if a1 not in list1:
    raise Exception('invalid a1')

b_thresholds = {1: 1, 2: 2, 3: 3}
if b1 in b_thresholds and c1 > b_thresholds[b1]:
    raise Exception('invalid c1 for b1')

z = c1 * d * f

这种写法不仅更简洁，后续新增b1的取值时只需修改字典即可，维护性更强，同时自然不会触发Pylint的相关规则报错。

方案3：保留嵌套结构但调整内层逻辑

如果一定要保留原有的嵌套结构，可以去掉内层的else，直接在所有校验通过后执行计算：

if a1 in list1:
    if b1 == 1 and c1 > 1:
        raise Exception('invalid c1 for b1')
    if b1 == 2 and c1 > 2:
        raise Exception('invalid c1 for b1')
    if b1 == 3 and c1 > 3:
        raise Exception('invalid c1 for b1')
    # 所有校验通过后执行计算，无需else
    z = c1 * d * f
else:
    raise Exception('invalid a1')

这里的逻辑和原代码完全一致：只有当a1在list1中，且所有b1相关的校验都不触发异常时，才会执行z = c1 * d * f，不会触发Pylint的报错。

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