From 161d296871aab85943dde89ebb5027360601eaf4 Mon Sep 17 00:00:00 2001
From: Oleg Nykolyn <juravel2@gmail.com>
Date: Sat, 26 Jan 2019 16:41:41 +0200
Subject: [PATCH] Fix get_object_or_404() klass parameter annotation.

---
 django-stubs/shortcuts.pyi | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/django-stubs/shortcuts.pyi b/django-stubs/shortcuts.pyi
index a158af8..fc1b900 100644
--- a/django-stubs/shortcuts.pyi
+++ b/django-stubs/shortcuts.pyi
@@ -1,6 +1,6 @@
 from typing import Any, Callable, Dict, List, Optional, Type, Union
 
-from django.db.models.base import Model
+from django.db.models.base import Model, Manager, QuerySet
 from django.http.response import HttpResponse, HttpResponseRedirect
 
 from django.http import HttpRequest
@@ -21,6 +21,6 @@ def render(
     using: Optional[str] = ...,
 ) -> HttpResponse: ...
 def redirect(to: Union[Callable, str], *args: Any, permanent: bool = ..., **kwargs: Any) -> HttpResponseRedirect: ...
-def get_object_or_404(klass: Union[Type[Model], str], *args: Any, **kwargs: Any) -> Model: ...
+def get_object_or_404(klass: Union[Type[Model], Type[Manager], QuerySet], *args: Any, **kwargs: Any) -> Model: ...
 def get_list_or_404(klass: List[Type[Model]], *args: Any, **kwargs: Any) -> List[Model]: ...
 def resolve_url(to: Union[Callable, Model, str], *args: Any, **kwargs: Any) -> str: ...