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)