Fix generation of API documentation

- Can't use get_object_or_404 in API views!
This commit is contained in:
Oliver 2021-10-17 22:32:03 +11:00
parent 7c5c1b6696
commit 6ebb956c8d
2 changed files with 21 additions and 6 deletions

View File

@ -5,7 +5,6 @@ JSON API for the Build app
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.shortcuts import get_object_or_404
from django.conf.urls import url, include
from rest_framework import filters, generics
@ -203,7 +202,12 @@ class BuildUnallocate(generics.CreateAPIView):
def get_serializer_context(self):
ctx = super().get_serializer_context()
ctx['build'] = get_object_or_404(Build, pk=self.kwargs.get('pk', None))
try:
ctx['build'] = Build.objects.get(pk=self.kwargs.get('pk', None))
except:
pass
ctx['request'] = self.request
return ctx
@ -222,7 +226,11 @@ class BuildComplete(generics.CreateAPIView):
ctx = super().get_serializer_context()
ctx['request'] = self.request
ctx['build'] = get_object_or_404(Build, pk=self.kwargs.get('pk', None))
try:
ctx['build'] = Build.objects.get(pk=self.kwargs.get('pk', None))
except:
pass
return ctx
@ -250,7 +258,11 @@ class BuildAllocate(generics.CreateAPIView):
context = super().get_serializer_context()
context['build'] = get_object_or_404(Build, pk=self.kwargs.get('pk', None))
try:
context['build'] = Build.objects.get(pk=self.kwargs.get('pk', None))
except:
pass
context['request'] = self.request
return context

View File

@ -7,7 +7,6 @@ from __future__ import unicode_literals
from django.conf.urls import url, include
from django.db.models import Q, F
from django.shortcuts import get_object_or_404
from django_filters import rest_framework as rest_filters
from rest_framework import generics
@ -235,7 +234,11 @@ class POReceive(generics.CreateAPIView):
context = super().get_serializer_context()
# Pass the purchase order through to the serializer for validation
context['order'] = get_object_or_404(PurchaseOrder, pk=self.kwargs.get('pk', None))
try:
context['order'] = PurchaseOrder.objects.get(pk=self.kwargs.get('pk', None))
except:
pass
context['request'] = self.request
return context