python 重载_Python中的运算符重载

运算符大家在写程序时不知不觉都已经用烂了，所谓运算符就是对操作数的操作，比如算数运算符（+，-，*，/）, 关系运算符（>,<,==）等。不过大家都发现了这些运算符只能对特定的类型去操作。比如+，操作两个整型对象相加，或者操作两个字符串对象拼接， 这里你应该注意到了原来+运算符的“实现”不止针对数字啊，还可以针对字符串，其实这就是运算符重载，和函数重载一样，它体现了多态性。第一次知道运算符重载是在c++里，python同样也可以实现。

其实当我们使用一些运算符时，python本质调用了的一些magic method，比如+，对应的是__add__。比如==，对应的是__eq__。 我们可以在自己的类中去重写它们。

class User:
    def __init__(self, name, pwd):
        self.name = name
        self.pwd = pwd


u1 = User("jack", "123456")
u2 = User("jack", "123456")
print(u1 == u2)

运行结果肯定是False，因为u1和u2指向了不同的对象，可是产品里，希望只要两个对象字段name和pwd一样，他们就是同一个用户，于是你可以重写__eq__, 重写==的实现。

class User:
    def __init__(self, name, pwd):
        self.name = name
        self.pwd = pwd

    def __eq__(self, other):
        if self.name == other.name and self.pwd == other.pwd:
            return True
        else:
            return False


u1 = User("jack", "123456")
u2 = User("jack", "123456")
print(u1 == u2)

运行结果

True

又比如上个例子中两个对象如果相加，就返回两个对象name字段的拼接，那需要重写__add__

class User:
    def __init__(self, name, pwd):
        self.name = name
        self.pwd = pwd

    def __eq__(self, other):
        if self.name == other.name and self.pwd == other.pwd:
            return True
        else:
            return False

    def __add__(self, other):
        return self.name + other.name


u1 = User("jack", "123456")
u2 = User("jack", "123456")
print(u1 == u2)
print(u1 + u2)

运行结果

True
jackjack