Python版の iffattrs

iffattrs.py

def iffattrs(cls, *attrs):
    def _iffattrs(self, other):
        for name in attrs:
            if not hasattr(self, name) or not hasattr(other, name):
                return False
            if getattr(self, name) != getattr(other, name):
                return False
        return True

    cls.__eq__ = _iffattrs

class SampleClass(object):

    def __init__(self, name, age):
        self.name = name
        self.age = age

iffattrs(SampleClass, "name", "age")

yae = SampleClass("yae", 16)
_yae = SampleClass("yae", 16)
cath = SampleClass("cath", 16)

print(yae == _yae)
print(yae == cath)