inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-07-09 07:00:56 +00:00
Code Activity

add stuff to mark starting endpoint

Browse Source
This commit is contained in:
Matthias
2021-12-07 23:54:45 +01:00
parent 2cf4e20d9a
commit 948270bbcb
3 changed files with 33 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
InvenTree/part/api.py
View File

@ -255,14 +255,28 @@ class CategoryTree(generics.ListAPIView):
cat_id = params.get('id', None)
if cat_id in (None, '', '#', ):
queryset = queryset.filter(parent=None)
filter_kwargs = {}
# get whole tree
if 'ref_type' in self.kwargs and 'ref' in self.kwargs:
if 'part' in self.kwargs['ref_type']:
filter_kwargs['parts__id'] = self.kwargs['ref']
elif 'category' in self.kwargs['ref_type']:
filter_kwargs['id'] = self.kwargs['ref']
else:
filter_kwargs['id'] = None
branch = queryset.model.objects.filter(**filter_kwargs).get_ancestors(include_self=True)
queryset = queryset.filter(parent=None) | branch
self.kwargs['ancestors'] = branch
else:
queryset = queryset.filter(parent_id=cat_id)
return queryset
def get(self, request, *args, **kwargs):
return super().get(request, *args, **kwargs)
def get_serializer_context(self):
context = super().get_serializer_context()
if 'ancestors' in self.kwargs:
context['ancestors'] = self.kwargs['ancestors']
return context
class PartSalePriceList(generics.ListCreateAPIView):
@ -1541,6 +1555,7 @@ part_api_urls = [
# Base URL for PartCategory API endpoints
url(r'^category/', include([
url(r'^tree/(?P<ref_type>[-\w]+)/(?P<ref>\d+)/', CategoryTree.as_view(), name='api-part-category-root-tree'),
url(r'^tree/', CategoryTree.as_view(), name='api-part-category-tree'),
url(r'^parameters/', CategoryParameterList.as_view(), name='api-part-category-parameter-list'),