From 44db5af9a819017075e0493f86db48d52b296a84 Mon Sep 17 00:00:00 2001 From: Klaus-Uwe Mitterer Date: Mon, 12 Oct 2020 16:54:45 +0200 Subject: [PATCH] Catch missing API key field --- core/views/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/views/__init__.py b/core/views/__init__.py index 57ed959..d28750b 100644 --- a/core/views/__init__.py +++ b/core/views/__init__.py @@ -26,7 +26,7 @@ class APIView(View): def dispatch(self, request, *args, **kwargs): try: APIKey.objects.get(key=request.headers["apikey"]) - except APIKey.DoesNotExist: + except (APIKey.DoesNotExist, KeyError): return JsonResponse({"error": "No valid API key provided"}) return super().dispatch(request, *args, **kwargs)