Я хочу переопределить метод __getattr__
для класса, чтобы сделать что-то необычное, но я не хочу нарушать поведение по умолчанию.
Каков правильный способ сделать это?
Я хочу переопределить метод __getattr__
для класса, чтобы сделать что-то необычное, но я не хочу нарушать поведение по умолчанию.
Каков правильный способ сделать это?
Переопределение __getattr__
должно быть точным - __getattr__
вызывается только как последнее средство, то есть если в экземпляре нет атрибутов, соответствующих имени. Например, если вы обращаетесь к foo.bar
, то __getattr__
вызывается только в том случае, если foo
не имеет атрибута bar
. Если атрибутом является тот, который вы не хотите обрабатывать, поднимите AttributeError
:
class Foo(object):
def __getattr__(self, name):
if some_predicate(name):
# ...
else:
# Default behaviour
raise AttributeError
Однако, в отличие от __getattr__
, __getattribute__
будет вызываться первым (работает только для новых классов стиля, то есть тех, которые наследуются от объекта). В этом случае вы можете сохранить поведение по умолчанию следующим образом:
class Foo(object):
def __getattribute__(self, name):
if some_predicate(name):
# ...
else:
# Default behaviour
return object.__getattribute__(self, name)
Смотрите документы Python для более.
class A(object):
def __init__(self):
self.a = 42
def __getattr__(self, attr):
if attr in ["b", "c"]:
return 42
raise AttributeError("%r object has no attribute %r" %
(self.__class__, attr))
# exception text copied from Python2.6
>>> a = A()
>>> a.a
42
>>> a.b
42
>>> a.missing
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 8, in __getattr__
AttributeError: 'A' object has no attribute 'missing'
>>> hasattr(a, "b")
True
>>> hasattr(a, "missing")
False
Чтобы продлить действие Майка, если вы хотите поддерживать поведение по умолчанию с помощью __getattr__
, вы можете сделать это следующим образом:
class Foo(object):
def __getattr__(self, name):
if name == 'something':
return 42
# Default behaviour
return self.__getattribute__(name)
Теперь сообщение об исключении более наглядное:
>>> foo.something
42
>>> foo.error
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 5, in __getattr__
AttributeError: 'Foo' object has no attribute 'error'