Как избежать дублирования кода с помощью аналогичных методов модели Django?

Question

Следующая модель содержит две почти идентичные функции list_ancestors и list_descendants. Что было бы хорошим способом написать этот код только один раз?

class Node(models.Model): 
    name = models.CharField(max_length=120, blank=True, null=True) 
    parents = models.ManyToManyField('self', blank=True, symmetrical=False) 

    def list_parents(self): 
     return self.parents.all() 

    def list_children(self): 
     return Node.objects.filter(parents=self.id) 

    def list_ancestors(self): 
     parents = self.list_parents() 
     ancestors = set(parents) 
     for p in parents: 
      ancestors |= set(p.list_ancestors()) # set union 
     return list(ancestors) 

    def list_descendants(self): 
     children = self.list_children() 
     descendants = set(children) 
     for c in children: 
      descendants |= set(c.list_descendants()) # set union 
     return list(descendants) 

    def __str__(self): 
     return self.name 

EDIT: раствор, полученный из приведенных ниже ответов:

def list_withindirect(self, arg): 
    direct = getattr(self, arg)() 
    withindirect = set(direct) 
    for d in direct: 
     withindirect |= set(d.list_withindirect(arg)) 
    return list(withindirect) 

def list_ancestors(self): 
    return self.list_withindirect('list_parents') 

def list_descendants(self): 
    return self.list_withindirect('list_children') 

Answer 1

Используйте строку и вызвать GetAttr на объект, чтобы получить вызываемую функцию.

def list_withindirect(self, fn1): 
    direct = getattr(self, fn1)() 
    withindirect = set(direct) 
    for d in direct: 
     withindirect |= set(d.list_withindirect(fn1)) 

    return list(withindirect) 

def list_ancestors(self): 
    return self.list_withindirect('list_parents') 

Answer 2

Это очень похоже на проблемы в связанной и несвязанной задачи методы.

Когда вы изначально проходите self.list_parents до , все в порядке.

Но когда вы рекурсивно проходите то же самое! self.list_parents - d.list_withindirect (т. Е. Потомкам), вы случайно заполняете переменную direct родителями самого верхнего объекта-вызывающего, а не d.

Например, она может быть решена с помощью getattr, как он ответил на 2ps(UPD: ошибка в его исходном коде был найден в комментариях там).