mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
e768ada83b
- Consolidated "in_stock" filter to single code location - Improve 'limit_choices_to' for BuildItem and SalesOrderAllocation - Various template improvements etc
153 lines
3.6 KiB
Python
153 lines
3.6 KiB
Python
"""
|
|
JSON API for the Build app
|
|
"""
|
|
|
|
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from rest_framework import filters
|
|
from rest_framework import generics, permissions
|
|
|
|
from django.conf.urls import url, include
|
|
|
|
from InvenTree.helpers import str2bool
|
|
|
|
from .models import Build, BuildItem
|
|
from .serializers import BuildSerializer, BuildItemSerializer
|
|
|
|
|
|
class BuildList(generics.ListCreateAPIView):
|
|
""" API endpoint for accessing a list of Build objects.
|
|
|
|
- GET: Return list of objects (with filters)
|
|
- POST: Create a new Build object
|
|
"""
|
|
|
|
queryset = Build.objects.all()
|
|
serializer_class = BuildSerializer
|
|
|
|
permission_classes = [
|
|
permissions.IsAuthenticated,
|
|
]
|
|
|
|
filter_backends = [
|
|
DjangoFilterBackend,
|
|
filters.SearchFilter,
|
|
filters.OrderingFilter,
|
|
]
|
|
|
|
filter_fields = [
|
|
'sales_order',
|
|
]
|
|
|
|
def get_queryset(self):
|
|
"""
|
|
Override the queryset filtering,
|
|
as some of the fields don't natively play nicely with DRF
|
|
"""
|
|
|
|
queryset = super().get_queryset().prefetch_related('part')
|
|
|
|
return queryset
|
|
|
|
def filter_queryset(self, queryset):
|
|
|
|
queryset = super().filter_queryset(queryset)
|
|
|
|
# Filter by build status?
|
|
status = self.request.query_params.get('status', None)
|
|
|
|
if status is not None:
|
|
queryset = queryset.filter(status=status)
|
|
|
|
# Filter by associated part?
|
|
part = self.request.query_params.get('part', None)
|
|
|
|
if part is not None:
|
|
queryset = queryset.filter(part=part)
|
|
|
|
return queryset
|
|
|
|
def get_serializer(self, *args, **kwargs):
|
|
|
|
try:
|
|
part_detail = str2bool(self.request.GET.get('part_detail', None))
|
|
except AttributeError:
|
|
part_detail = None
|
|
|
|
kwargs['part_detail'] = part_detail
|
|
|
|
return self.serializer_class(*args, **kwargs)
|
|
|
|
|
|
class BuildDetail(generics.RetrieveUpdateAPIView):
|
|
""" API endpoint for detail view of a Build object """
|
|
|
|
queryset = Build.objects.all()
|
|
serializer_class = BuildSerializer
|
|
|
|
permission_classes = [
|
|
permissions.IsAuthenticated,
|
|
]
|
|
|
|
|
|
class BuildItemList(generics.ListCreateAPIView):
|
|
""" API endpoint for accessing a list of BuildItem objects
|
|
|
|
- GET: Return list of objects
|
|
- POST: Create a new BuildItem object
|
|
"""
|
|
|
|
serializer_class = BuildItemSerializer
|
|
|
|
def get_queryset(self):
|
|
""" Override the queryset method,
|
|
to allow filtering by stock_item.part
|
|
"""
|
|
|
|
query = BuildItem.objects.all()
|
|
|
|
query = query.select_related('stock_item')
|
|
query = query.prefetch_related('stock_item__part')
|
|
query = query.prefetch_related('stock_item__part__category')
|
|
|
|
return query
|
|
|
|
def filter_queryset(self, queryset):
|
|
queryset = super().filter_queryset(queryset)
|
|
|
|
# Does the user wish to filter by part?
|
|
part_pk = self.request.query_params.get('part', None)
|
|
|
|
if part_pk:
|
|
queryset = queryset.filter(stock_item__part=part_pk)
|
|
|
|
return queryset
|
|
|
|
permission_classes = [
|
|
permissions.IsAuthenticated,
|
|
]
|
|
|
|
filter_backends = [
|
|
DjangoFilterBackend,
|
|
]
|
|
|
|
filter_fields = [
|
|
'build',
|
|
'stock_item'
|
|
]
|
|
|
|
|
|
build_item_api_urls = [
|
|
url('^.*$', BuildItemList.as_view(), name='api-build-item-list'),
|
|
]
|
|
|
|
build_api_urls = [
|
|
url(r'^item/', include(build_item_api_urls)),
|
|
|
|
url(r'^(?P<pk>\d+)/', BuildDetail.as_view(), name='api-build-detail'),
|
|
|
|
url(r'^.*$', BuildList.as_view(), name='api-build-list'),
|
|
]
|