From 80fa40694cf9b321d4a8419a82da5486753b2435 Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Mon, 13 Apr 2020 20:36:51 +1000 Subject: [PATCH] Unit testing fixes - For some reason the builds were passing locally? --- InvenTree/company/api.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/InvenTree/company/api.py b/InvenTree/company/api.py index db92baafad..19ae54af32 100644 --- a/InvenTree/company/api.py +++ b/InvenTree/company/api.py @@ -101,9 +101,20 @@ class SupplierPartList(generics.ListCreateAPIView): def get_serializer(self, *args, **kwargs): # Do we wish to include extra detail? - kwargs['part_detail'] = str2bool(self.request.query_params.get('part_detail', None)) - kwargs['supplier_detail'] = str2bool(self.request.query_params.get('supplier_detail', None)) - kwargs['manufacturer_detail'] = str2bool(self.request.query_params.get('manufacturer_detail', None)) + try: + kwargs['part_detail'] = str2bool(self.request.query_params.get('part_detail', None)) + except AttributeError: + pass + + try: + kwargs['supplier_detail'] = str2bool(self.request.query_params.get('supplier_detail', None)) + except AttributeError: + pass + + try: + kwargs['manufacturer_detail'] = str2bool(self.request.query_params.get('manufacturer_detail', None)) + except AttributeError: + pass kwargs['context'] = self.get_serializer_context()