开发者

Show a django relationship in a template

开发者 https://www.devze.com 2023-03-02 20:53 出处:网络
I have a django model as follows: class Person(models.Model): name = models.CharField(max_length=255) class Relationship(models.Model):

I have a django model as follows:

class Person(models.Model):
    name = models.CharField(max_length=255)

class Relationship(models.Model):
    parent = models.ForeignKey(Person)
    child = models.ForeignKey(Person)
    description = models.TextField(blank=True)

In 开发者_C百科my view, I pass a certain person, and the relationships in which he/she is parent:

person = Person.objects.filter(name ='some name')
descendant_relationships = Relationship.objects.filter(parent = person)

An I want to show this person's descendants in a list in a template:

<ul>
{% for item in descendant_relationships%}
    <li> {{item.child.name}}  -  {{item.description}} </li>
{% endfor %}
</ul>

But this template code will not show the children of children (i.e. grandchildren, great-grandchildren etc.). How can I get these lower level descendants to show up? I imagine recursion is necessary somewhere, but where?


First, set a related name for your relationship ForeignKeys:

parent = models.ForeignKey(Person, related_name='child_relationships')
child = models.ForeignKey(Person, related_name='parent_relationships')

Then add something like the following to your Person models.py:

def get_descendants(self):
    descendants = []
    children = [relationship.child for relationship in self.child_relationships.all()]
    for child in children:
        descendants += child.get_descendants()
    return descendants

Instead of a method to return the children, we can adapt this to return the relationships:

def get_descendant_relationships(self):
    relationships = []
    for child in self.child_relationships.all():
        relationships += child.get_descendant_relationships()
    return descendants


If you only need a one to many rel between parent and children, you can drop the extra table:

from django.db import models
class Person(models.Model):
    name = models.CharField(max_length=255)
    parent = models.ForeignKey('self', null=True, blank=True, related_name='a_name')

    def genealogical_tree(self, type_of_members, generations):
        """
        @generations: the number of generations to be traversed in the tree(
            controls recursion)
        """
        formated_members = {}
        if generations:
            if type_of_members == 'parent':
                members = [self.parent]
            elif type_of_members == 'children':
                members = self._default_manager.filter(parent=self)
            for member in members:
                formated_members[member.name] = member.genealogical_tree(
                    type_of_members, generations-1)
        return formated_members

Examples:

from models import Person
f=Person(name='f')
s=Person(name='s')
gs=Person(name='granson')
gs.save()
s.save()
f.save()

s.parent=f
s.save()
gs.parent=s
gs.save()

f.genealogical_tree('children',1)
>>> {u's': {}}
f.genealogical_tree('children', 2)
>>> {u's': {u'granson': {}}}
gs.genealogical_tree('parent', 2)
>>> {'s': {'f': {}}}
gs.genealogical_tree('parent', 1)
>>> {'s': {}}
0

精彩评论

暂无评论...
验证码 换一张
取 消

关注公众号