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

Split tree generation off into a separate function

Browse Source
This commit is contained in:
Oliver Walters 2019-06-17 23:39:43 +10:00
parent c6b1a80703
commit b519a1981d
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
InvenTree/InvenTree/views.py
View File

@ -59,19 +59,24 @@ class TreeSerializer(views.APIView):
return data
def get(self, request, *args, **kwargs):
def get_items(self):
top_items = self.model.objects.filter(parent=None).order_by('name')
return self.model.objects.all()
def generate_tree(self, items):
nodes = []
# Construct the top-level items
top_items = [i for i in items if i.parent is None]
top_count = 0
for item in top_items:
nodes.append(self.itemToJson(item))
top_count += item.item_count
top = {
self.tree = {
'pk': None,
'text': self.title,
'href': self.root_url,
@ -79,8 +84,15 @@ class TreeSerializer(views.APIView):
'tags': [top_count],
}
def get(self, request, *args, **kwargs):
""" Respond to a GET request for the Tree """
items = self.model.objects.all()
self.generate_tree(items)
response = {
'tree': [top]
'tree': [self.tree]
}
return JsonResponse(response, safe=False)