Python, wie man mehrere Argumente bei add magic Methode akzeptiert

Question

Ich versuche, mit Operator überladen zu spielen, und ich fand mich selbst versuchen mehr als 2 Argumente. Wie würde ich dies implementieren, um eine beliebige Anzahl von Argumenten zu akzeptieren.

class Dividend: 

    def __init__(self, amount): 
     self.amount = amount 

    def __add__(self, other_investment): 
     return self.amount + other_investment.amount 

investmentA = Dividend(150) 
investmentB = Dividend(50) 
investmentC = Dividend(25) 

print(investmentA + investmentB) #200 
print(investmentA + investmentB + investmentC) #error 

Answer 1

Das Problem ist nicht, dass Ihre __add__ Methode nicht mehrere Argumente akzeptiert, das Problem ist, dass es keinen Dividend zurückgibt. Der Additionsoperator ist immer ein binärer Operator, aber nach der ersten Addition versuchen Sie, einen numerischen Typ zu einem Dividend hinzuzufügen, anstatt zwei Dividenden hinzuzufügen. Sie sollten Ihre __add__ Methode den entsprechenden Typen zurückgeben, zum Beispiel haben:

def __add__(self, other_investment): 
    return Dividend(self.amount + other_investment.amount)