Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #14 from SchrodingersGat/master

Browse Source 
Added function to get all parents for a tree item
This commit is contained in:
Oliver 2017-03-29 20:23:51 +11:00 committed by GitHub
commit c779476650
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
InvenTree/InvenTree/models.py
View File

@ -42,7 +42,23 @@ class InvenTreeTree(models.Model):
parent = models.ForeignKey('self', parent = models.ForeignKey('self',
on_delete=models.CASCADE, on_delete=models.CASCADE,
blank=True, blank=True,
null=True) null=True,
related_name='children')
def getUniqueParents(self, unique=None):
""" Return a flat set of all parent items that exist above this node.
If any parents are repeated (which would be very bad!), the process is halted
"""
if unique is None:
unique = set()
else:
unique.add(self.id)
if self.parent and self.parent.id not in unique:
self.parent.getUniqueParents(unique)
return unique
def getUniqueChildren(self, unique=None): def getUniqueChildren(self, unique=None):
""" Return a flat set of all child items that exist under this node. """ Return a flat set of all child items that exist under this node.
@ -60,9 +76,6 @@ class InvenTreeTree(models.Model):
# Some magic to get around the limitations of abstract models # Some magic to get around the limitations of abstract models
contents = ContentType.objects.get_for_model(type(self)) contents = ContentType.objects.get_for_model(type(self))
children = contents.get_all_objects_for_this_type(parent=self.id) children = contents.get_all_objects_for_this_type(parent=self.id)
for child in children:
child.getUniqueChildren(unique)
return unique return unique