如何用Pythonic方式实现无重复代码的多层比较函数？

问题概述 #

原函数通过多轮调用不同的复杂比较函数，若各轮结果冲突（比如之前标记为-1，当前需要标记为1），则提前返回0表示不可比较；若结果兼容，则返回最终的比较状态。原代码存在大量重复逻辑，尝试用内部函数重构时，遇到内部函数无法直接触发外部函数提前返回的问题。

原重复代码实现 #

def compare(x, y):
  current = 0

  tmp = complicated_function(x, y)
  if tmp > 0:
    if current < 0:
      return 0 # incomparable
    current = -1
  elif tmp < 0:
    if current > 0:
      return 0 # incomparable
    current = 1

  tmp = complicated_function_2(x, y)
  if tmp > 0:
    if current < 0:
      return 0 # incomparable
    current = -1
  elif tmp < 0:
    if current > 0:
      return 0 # incomparable
    current = 1

  # ... 更多重复的比较逻辑

  return current

尝试的重构（存在缺陷） #

def compare(x, y):
  current = 0
  def inner_compare(z):
    if z > 0:
      if current == -1:
        return 0 # 此处返回仅终止内部函数，无法让外部函数提前退出
      current = 1
    if z < 0:
      if current == 1:
        return 0 # 同上，无法触发外部函数返回
      current = -1
  inner_compare(complicated_function(x, y))
  inner_compare(complicated_function_2(x, y))
  # ...

符合Python风格的解决方案 #

方案1：内部函数返回冲突状态，外部函数判断终止 #

利用nonlocal关键字让内部函数修改外部函数的current变量，同时让内部函数返回布尔值标记是否出现冲突，外部函数每次调用后检查状态，若冲突则直接返回0。

def compare(x, y):
    current = 0

    def update_current(z):
        nonlocal current
        if z > 0:
            if current == -1:
                return False  # 结果冲突，需要终止
            current = -1
        elif z < 0:
            if current == 1:
                return False
            current = 1
        return True  # 无冲突，继续执行后续比较

    # 依次执行各比较步骤
    if not update_current(complicated_function(x, y)):
        return 0
    if not update_current(complicated_function_2(x, y)):
        return 0
    # ... 添加更多比较步骤

    return current

方案2：批量处理比较函数（更易维护） #

如果有大量比较函数，可将它们存入列表循环处理，进一步简化代码，降低后续维护成本：

def compare(x, y):
    current = 0
    # 把所有需要执行的比较函数放入列表
    comparator_functions = [
        complicated_function,
        complicated_function_2,
        # ... 添加其他比较函数
    ]

    def update_current(z):
        nonlocal current
        if z > 0:
            if current == -1:
                return False
            current = -1
        elif z < 0:
            if current == 1:
                return False
            current = 1
        return True

    # 循环执行所有比较
    for func in comparator_functions:
        if not update_current(func(x, y)):
            return 0

    return current

方案优势 #

• 完全消除重复代码，逻辑统一维护
• 利用nonlocal清晰管理外部变量，符合Python作用域规范
• 提前返回的逻辑直观，代码可读性强
• 批量处理的方式让后续新增比较函数更便捷

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