119 lines
4.5 KiB
Python
119 lines
4.5 KiB
Python
from django.db import transaction
|
|
from django.urls import path
|
|
from rest_framework import routers, viewsets, status
|
|
from rest_framework.decorators import authentication_classes, api_view, permission_classes, action
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
|
|
from authentication.models import ToolshedUser, KnownIdentity
|
|
from authentication.signature_auth import SignatureAuthentication
|
|
from toolshed.models import InventoryItem, StorageLocation, WorkflowInstance
|
|
from toolshed.serializers import InventoryItemSerializer, StorageLocationSerializer, WorkflowInstanceSerializer
|
|
|
|
router = routers.SimpleRouter()
|
|
|
|
|
|
def inventory_items(identity):
|
|
try:
|
|
user = identity.user.get()
|
|
if user:
|
|
for item in user.inventory_items.all():
|
|
yield item
|
|
except ToolshedUser.DoesNotExist:
|
|
pass
|
|
for friend in identity.friends.all():
|
|
if friend_user := friend.user.get():
|
|
for item in friend_user.inventory_items.all():
|
|
if item.availability_policy != 'private':
|
|
yield item
|
|
|
|
|
|
class InventoryItemViewSet(viewsets.ModelViewSet):
|
|
serializer_class = InventoryItemSerializer
|
|
authentication_classes = [SignatureAuthentication]
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
if type(self.request.user) == KnownIdentity and self.request.user.user.exists():
|
|
return InventoryItem.objects.filter(owner=self.request.user.user.get())
|
|
return InventoryItem.objects.none()
|
|
|
|
def perform_create(self, serializer):
|
|
with transaction.atomic():
|
|
serializer.save(owner=self.request.user.user.get()).clean()
|
|
|
|
def perform_update(self, serializer):
|
|
with transaction.atomic():
|
|
if serializer.instance.owner == self.request.user.user.get():
|
|
serializer.save().clean()
|
|
|
|
def perform_destroy(self, instance):
|
|
if instance.owner == self.request.user.user.get():
|
|
instance.delete()
|
|
|
|
|
|
@api_view(['GET'])
|
|
@authentication_classes([SignatureAuthentication])
|
|
@permission_classes([IsAuthenticated])
|
|
def search_inventory_items(request):
|
|
query = request.query_params.get('query')
|
|
if query:
|
|
return Response(InventoryItemSerializer(inventory_items(request.user), many=True).data)
|
|
return Response({'error': 'No query provided.'}, status=400)
|
|
|
|
|
|
class StorageLocationViewSet(viewsets.ModelViewSet):
|
|
serializer_class = StorageLocationSerializer
|
|
authentication_classes = [SignatureAuthentication]
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
if type(self.request.user) == KnownIdentity and self.request.user.user.exists():
|
|
return StorageLocation.objects.filter(owner=self.request.user.user.get())
|
|
return StorageLocation.objects.none()
|
|
|
|
def perform_create(self, serializer):
|
|
with transaction.atomic():
|
|
serializer.save(owner=self.request.user.user.get())
|
|
|
|
def perform_update(self, serializer):
|
|
with transaction.atomic():
|
|
if serializer.instance.owner == self.request.user.user.get():
|
|
serializer.save()
|
|
|
|
def perform_destroy(self, instance):
|
|
if instance.owner == self.request.user.user.get():
|
|
instance.delete()
|
|
|
|
|
|
class WorkflowInstanceViewSet(viewsets.ModelViewSet):
|
|
serializer_class = WorkflowInstanceSerializer
|
|
authentication_classes = [SignatureAuthentication]
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
if type(self.request.user) == KnownIdentity and self.request.user.user.exists():
|
|
return WorkflowInstance.objects.filter(owner=self.request.user.user.get())
|
|
return WorkflowInstance.objects.none()
|
|
|
|
def perform_create(self, serializer):
|
|
with transaction.atomic():
|
|
serializer.save(owner=self.request.user.user.get())
|
|
|
|
def perform_update(self, serializer):
|
|
with transaction.atomic():
|
|
if serializer.instance.owner == self.request.user.user.get():
|
|
serializer.save()
|
|
|
|
def perform_destroy(self, instance):
|
|
if instance.owner == self.request.user.user.get():
|
|
instance.delete()
|
|
|
|
|
|
router.register(r'inventory_items', InventoryItemViewSet, basename='inventory_items')
|
|
router.register(r'storage_locations', StorageLocationViewSet, basename='storage_locations')
|
|
router.register(r'workflows', WorkflowInstanceViewSet, basename='workflows')
|
|
|
|
urlpatterns = router.urls + [
|
|
path('search/', search_inventory_items, name='search_inventory_items'),
|
|
]
|