Просмотр исходного кода

Fixes #2334: TypeError raised when WritableNestedSerializer receives a non-integer value

Jeremy Stretch 7 лет назад
Родитель
Сommit
a2ff21fab9
1 измененных файлов с 3 добавлено и 1 удалено
  1. 3 1
      netbox/utilities/api.py

+ 3 - 1
netbox/utilities/api.py

@@ -170,7 +170,9 @@ class WritableNestedSerializer(ModelSerializer):
         if data is None:
             return None
         try:
-            return self.Meta.model.objects.get(pk=data)
+            return self.Meta.model.objects.get(pk=int(data))
+        except (TypeError, ValueError):
+            raise ValidationError("Primary key must be an integer")
         except ObjectDoesNotExist:
             raise ValidationError("Invalid ID")