Python рекурсивно __getattribute__ - PullRequest
2 голосов
/ 22 ноября 2010

Мне нужно реализовать такое поведение:

obj.attr1.attr2.attr3 --> obj.attr1__attr2__attr3

Похоже, мне нужно переопределить класс obj __getattribute__, а также каким-то образом использовать дескрипторы python.

UPDATE:

У меня есть проект Django.

obj - это экземпляр SearchResult от django-haystack, он содержит много ненормализованных данных (user__name, user__address) из модели django, и мне нужно получить к нему доступ result.user.name для совместимости.

ОБНОВЛЕНИЕ для ответа THC4k:

Что если у меня есть:

class Target(object):
    attr1 = 1
    attr1__attr2__attr3 = 5

>>> proxy.attr1
1
>>> proxy.attr1.attr2.attr3
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'int' object has no attribute 'attr2'

Любая помощь будет высоко ценится.

Ответы [ 3 ]

3 голосов
/ 22 ноября 2010

Надеюсь, вы знаете, что делаете, и это не просто схема , чтобы избежать исправления существующего кода .

Я думаю, что есть законные причины для этого, после всего, что я сделал в Lua нечто подобное, чтобы реализовать обертку вокруг некоторого кода на Си без необходимости фактически писать код для каждой представленной функции.

Но вы должны хотя бы отделить реальный класс от прокси:

# the proxy maps attribute access to another object
class GetattrProxy(object):
    def __init__(self, proxied, prefix=None):
        self.proxied = proxied
        self.prefix = prefix

    def __getattr__(self, key):
        attr = (key if self.prefix is None else self.prefix + '__' + key)
        try:
            # if the proxied object has the attr return it
            return getattr(self.proxied, attr)
        except AttributeError:
            # else just return another proxy
            return GetattrProxy(self.proxied, attr)


# the thing you want to wrap
class Target(object):
    attr1__attr2__attr3 = 5


t = Target()
proxy = GetattrProxy(t)

print proxy.attr1.attr2.attr3

@ katrielalex предложение:

class GetattrProxy2(GetattrProxy):
    def __getattr__(self, key):
            attr = (key if self.prefix is None else self.prefix + '__' + key)
            proxy = GetattrProxy2(self.proxied, attr)

            # store val only if the proxied object has the attribute, 
            # this way we still get AttributeErrors on nonexisting items 
            if hasattr(self.proxied, attr):
                proxy.val = getattr(self.proxied, attr)
            return proxy

proxy = GetattrProxy2(t)
proxy.attr1.val # 1
proxy.attr1.attr2.attr3.val # 5
proxy.attr1.attr2.val # raise AttributeError
1 голос
/ 21 мая 2017

Для случаев, когда у вас есть список имен атрибутов, вы можете использовать функцию itertools() (в python-3.x functools.reduce()) и встроенную функцию getattr():

Вотпример:

In [1]: class A:
   ...:     def __init__(self):
   ...:         self.a1 = B()
   ...:         

In [2]: class B:
   ...:     def __init__(self):
   ...:         self.b1 = C()
   ...:         

In [3]: class C:
   ...:     def __init__(self):
   ...:         self.c1 = 7
   ...:         

In [4]: from functools import reduce 
In [5]: reduce(getattr, [A(), 'a1', 'b1', 'c1'])
Out[5]: 7
0 голосов
/ 22 ноября 2010
class A:
    def __init__(self, a):
        self.a = a

# create objects with many a  atributes..        
a = A(A(A(A('a'))))


x = a
# as long as a has atribute a continue...
while hasattr(x, 'a'):
    print x
    x = x.a
...