【python提高】01-运算符的重载

一直认为运算符 ±*/ > < = 是int才有的专利，但是发现只要在类中重写
__add__ __gt__等魔法函数，也可以进行相应的±><运算。

看一下int里面的函数

>>> help(int)
Help on class int in module builtins:
Methods defined here:
# 实现 a+b
__add__(self, value, /)
	Return self+value.
# 实现 a<b
__lt__(self, value, /)
	Return self<value.
# 其余的都差不多， 大家写之前看一下好了

代码实现运算符重载

class Foo():
    def __init__(self, num):
        self.num = num

    # a+b
    def __add__(self, value):
        num_sum = self.num+value.num
        return num_sum

    # a-b
    def __sub__(self, value):
        num_sub = self.num - value.num
        return num_sub
    
    # a*b
    def __mul__(self, value):
        num_mul = self.num*value.num
        return num_mul

    # a/b
    def __truediv__(self, value):
        num_truediv = self.num/value.num
        return num_truediv

    # a=b
    def __eq__(self, value):
        if self.num == value.num:
            return True
        return False

    # a>b
    def __gt__(self, value):
        if self.num > value.num:
            return True
        return False

    # a<b
    def __lt__(self, value):
        if self.num < value.num:
            return True
        return False
    

a = Foo(4)
b = Foo(2)

print(a+b)
print(a-b)
print(a*b)
print(a/b)
print(a == b)
print(a>b)
print(a<b)

# 运行结果
>>> 
6
2
8
2.0
False
True
False
>>>