Python декораторы и методы - PullRequest
Новая
       28

Python декораторы и методы

7 голосов
/ 31 марта 2009

Новое здесь. Также я (очень) новичок в python и пытаюсь понять следующее поведение. Может кто-нибудь объяснить мне, почему два метода в этом примере имеют разные выходные данные? 

def map_children(method):
    def wrapper(self,*args,**kwargs):
        res = method(self,*args,**kwargs)
        for child in self._children:
            method(child,*args,**kwargs)            
        return res
    return wrapper

class Node(object):

    def __init__(self,name,parent=None):
        self._namestring = name
        if parent:
            self._parent = parent

        self._children = []

    @map_children
    def decorated(self):
        if hasattr(self,'_parent'):
            print '%s (child of %s)'%(self._namestring,self._parent._namestring)
        else:
            print '%s'% self._namestring

    def undecorated(self):
        if hasattr(self,'_parent'):
            print '%s (child of %s)'%(self._namestring,self._parent._namestring)
        else:
            print '%s'% self._namestring

        for child in self._children:
            child.undecorated()


def runme():
    parent = Node('parent')

    child1 = Node('child1',parent)
    child2 = Node('child2',parent)
    grandchild = Node('grandchild',child1)
    child1._children.append(grandchild)
    parent._children.append(child1)
    parent._children.append(child2)

    print '**********result from decorator**********'
    parent.decorated()

    print '**********result by hand**********'
    parent.undecorated()

Вот вывод на моей системе: 

In[]:testcase.runme()
**********result from decorator**********
parent
child1 (child of parent)
child2 (child of parent)
**********result by hand**********
parent
child1 (child of parent)
grandchild (child of child1)
child2 (child of parent)

Так почему же декорированный вызов никогда не спускается в узел внука? Я явно что-то упускаю из-за синтаксиса ...

1 Ответ

7 голосов
/ 31 марта 2009

В декораторе вы перебираете дочерние узлы и вызываете оригинал , не рекурсивный method для них 

method(child, *args, **kwargs)

так что вы пройдете только один уровень. Попробуйте заменить эту строку на 

map_children(method)(child, *args, **kwargs)

и вы получите тот же вывод, что и для рекурсивной версии вручную.

...