如何使类方法中的两个方向减去和相乘成为可能

时间:2015-04-28 23:05:29

标签: class python-3.x methods

所以我正在尝试做一个可以添加,减去和乘以体积(包括幅度和单位)的程序,我使__add__工作__rsub____mult__我还没弄明白如何让它们发挥作用。我希望能够执行Volume * 22 * Volume等操作。任何关于如何去做的建议将不胜感激!

def __add__ (self, v2): #adds to magnitude using an int or another volume
    if Volume.is_valid(self):
        if isinstance(v2, Volume):
            if Volume.is_valid(v2): #check if V2 is valid
                if self.__units == v2.__units:
                    return Volume(self.__magnitude + v2.__magnitude, 
                                  self.__units)
                else:
                    if v2.__units == 'ml':
                        v3 = v2.customary()
                        return Volume(self.__magnitude + v3.__magnitude, 
                                      self.__units)
                    elif v2.__units == 'oz':
                        v3 = v2.__units
                        return Volume(self.__magnitude + v3.__magnitude, 
                                      self.__units)

        elif isinstance(v2,(int,float)):
            #create new Volume with same units as self
            v3 = Volume(v2, self.__units)
            return Volume(self.__magnitude + v3.__magnitude, self.__units)

def __radd__ (self, v2): #adds ability to do i.e. 2 + a
    return self.__add__(v2)

def __sub__ (self, v2): #same as __add__ but this subtracts
    if Volume.is_valid(self):
        if isinstance(v2, Volume):
            if Volume.is_valid(v2):
                if self.__units == v2.__units:
                    return Volume(self.__magnitude - v2.__magnitude, 
                                  self.__units)
                else:
                    if v2.__units == 'ml':
                        v3 = v2.customary()
                        return Volume(self.__magnitude - v3.__magnitude, 
                                      self.__units)
                    elif v2.__units == 'oz':
                        v3 = v2.__units
                        return Volume(self.__magnitude - v3.__magnitude, 
                                      self.__units)
        elif isinstance(v2,(int,float)):
            v3 = Volume(v2, self.__units)
            return Volume(self.__magnitude - v3.__magnitude, self.__units)

def __rsub__ (self, v2):
    return Volume(v2.__magnitude - self.__magnitude, self.__units)

def __mult__ (self, v2):#multiplies a volume by an int only (not other vol)
    if Volume.is_valid(self):
        if isinstance(v2, Volume):
            if Volume.is_valid(v2):
                if self.__units == v2.__units:
                    return Volume(self.__magnitude * v2.__magnitude, 
                                  self.__units)
                else:
                    if v2.__units == 'ml':
                        v3 = v2.customary()
                        return Volume(self.__magnitude * v3.__magnitude, 
                                      self.__units)
                    elif v2.__units == 'oz':
                        v3 = v2.__units
                        return Volume(self.__magnitude * v3.__magnitude, 
                                      self.__units)
        elif isinstance(v2,(int,float)):
            v3 = Volume(v2, self.__units)
            return Volume(self.__magnitude * v3.__magnitude, self.__units)
        #return Volume(self.__magnitude * v2, self.__units)

def __rmult__ (self, v2):
    return self.__mult__(v2)

1 个答案:

答案 0 :(得分:0)

乘法的神奇方法是__mul__,而不是__mult__

__rsub__似乎没有那么多工作__sub__。也许是这样的事情:

def __rsub__(self, v2):
    return Volume(v2) - self

会更好。