Cách triển khai __lt__ __gt__ tùy chỉnh (overload) trong Python

Python có magic method để định nghĩa hành vi overload của toán tử. Các toán tử so sánh (<, <=, >, >=, == và !=) có thể overload bằng cách định nghĩa __lt__, __le__, __gt__, __ge__, __eq__ và __ne__. Chương trình sau overload < và > để so sánh đối tượng của lớp distance.

class distance:
    def __init__(self, x=5, y=5):
        self.ft = x
        self.inch = y
    
    def __eq__(self, other):
        if self.ft == other.ft and self.inch == other.inch:
            return "both objects are equal"
        else:
            return "both objects are not equal"
    
    def __lt__(self, other):
        in1 = self.ft * 12 + self.inch
        in2 = other.ft * 12 + other.inch
        if in1 < in2:
            return "first object smaller than other"
        else:
            return "first object not smaller than other"
    
    def __gt__(self, other):
        in1 = self.ft * 12 + self.inch
        in2 = other.ft * 12 + other.inch
        if in1 > in2:
            return "first object greater than other"
        else:
            return "first object not greater than other"

d1 = distance(5, 5)
d2 = distance()
print(d1 > d2)

d3 = distance()
d4 = distance(6, 10)
print(d1 < d2)

d5 = distance(3, 11)
d6 = distance()
print(d5 < d6)

Kết quả cho thấy triển khai magic method __lt__ và __gt__:

first object not greater than other
first object not smaller than other
first object smaller than other