如何在不破坏遗留代码的情况下增加函数的返回值？

Question

def foo(a): 
    return a

amount = 4 + foo(a)

def foo(a)
   return a, a + 5

当我这样做的时候，变量'amount'不是正确的答案，因为a现在是，用pylint的话说是'tuple'，所以amount将从返回a+4，变为返回(a，a+5)+4

我如何将'a+5'添加到foo中，同时仍然允许amount是单个标量值，而不是元组？

Answer 1

也许这样的方法可以奏效：

def foo(a, new_usage=False):
    if not new_usage:
        return a

    return a + 5

然后针对所有新的使用情况，这样调用您的函数：

foo(a, True)

例如,您的旧代码仍然可以工作：

In [40]: amount = 4 + foo(4)

In [41]: amount
Out[41]: 8

In [42]: amount = 4 + foo(4, True)

In [43]: amount
Out[43]: 13