mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
130 lines
3.1 KiB
Python
130 lines
3.1 KiB
Python
from django_filters.rest_framework import FilterSet, DjangoFilterBackend
|
|
from django_filters import NumberFilter
|
|
|
|
from rest_framework import generics, permissions, response
|
|
|
|
# from InvenTree.models import FilterChildren
|
|
from .models import StockLocation, StockItem
|
|
from .serializers import StockItemSerializer, StockQuantitySerializer
|
|
from .serializers import LocationSerializer
|
|
|
|
|
|
class StockDetail(generics.RetrieveUpdateDestroyAPIView):
|
|
"""
|
|
|
|
get:
|
|
Return a single StockItem object
|
|
|
|
post:
|
|
Update a StockItem
|
|
|
|
delete:
|
|
Remove a StockItem
|
|
"""
|
|
|
|
queryset = StockItem.objects.all()
|
|
serializer_class = StockItemSerializer
|
|
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
|
|
|
|
|
class StockFilter(FilterSet):
|
|
min_stock = NumberFilter(name='quantity', lookup_expr='gte')
|
|
max_stock = NumberFilter(name='quantity', lookup_expr='lte')
|
|
|
|
class Meta:
|
|
model = StockItem
|
|
fields = ['quantity', 'part', 'location']
|
|
|
|
|
|
class StockList(generics.ListCreateAPIView):
|
|
"""
|
|
|
|
get:
|
|
Return a list of all StockItem objects
|
|
(with optional query filters)
|
|
|
|
post:
|
|
Create a new StockItem
|
|
"""
|
|
|
|
queryset = StockItem.objects.all()
|
|
serializer_class = StockItemSerializer
|
|
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
|
filter_backends = (DjangoFilterBackend,)
|
|
filter_class = StockFilter
|
|
|
|
|
|
class StockStocktakeEndpoint(generics.UpdateAPIView):
|
|
|
|
queryset = StockItem.objects.all()
|
|
serializer_class = StockQuantitySerializer
|
|
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
|
|
|
def update(self, request, *args, **kwargs):
|
|
object = self.get_object()
|
|
object.stocktake(request.data['quantity'], request.user)
|
|
|
|
serializer = self.get_serializer(object)
|
|
|
|
return response.Response(serializer.data)
|
|
|
|
|
|
class AddStockEndpoint(generics.UpdateAPIView):
|
|
|
|
queryset = StockItem.objects.all()
|
|
serializer_class = StockQuantitySerializer
|
|
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
|
|
|
def update(self, request, *args, **kwargs):
|
|
object = self.get_object()
|
|
object.add_stock(request.data['quantity'])
|
|
|
|
serializer = self.get_serializer(object)
|
|
|
|
return response.Response(serializer.data)
|
|
|
|
|
|
class LocationDetail(generics.RetrieveUpdateDestroyAPIView):
|
|
"""
|
|
|
|
get:
|
|
Return a single StockLocation object
|
|
|
|
post:
|
|
Update a StockLocation object
|
|
|
|
delete:
|
|
Remove a StockLocation object
|
|
|
|
"""
|
|
|
|
queryset = StockLocation.objects.all()
|
|
serializer_class = LocationSerializer
|
|
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
|
|
|
|
|
class StockLocationFilter(FilterSet):
|
|
|
|
class Meta:
|
|
model = StockLocation
|
|
fields = ['parent']
|
|
|
|
|
|
class LocationList(generics.ListCreateAPIView):
|
|
"""
|
|
|
|
get:
|
|
Return a list of all StockLocation objects
|
|
(with optional query filter)
|
|
|
|
post:
|
|
Create a new StockLocation
|
|
|
|
"""
|
|
|
|
queryset = StockLocation.objects.all()
|
|
serializer_class = LocationSerializer
|
|
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
|
filter_backends = (DjangoFilterBackend,)
|
|
filter_class = StockLocationFilter
|