fixed typing in contrib.auth.decorators (#2564)

Author: Rusich90

django-stubs/contrib/auth/decorators.pyi

@@ -1,11 +1,11 @@
-from collections.abc import Callable, Iterable
+from collections.abc import Awaitable, Callable, Iterable
 from typing import TypeVar, overload
 
 from django.contrib.auth.base_user import _UserModel
 from django.contrib.auth.models import AnonymousUser
 from django.http.response import HttpResponseBase
 
-_VIEW = TypeVar("_VIEW", bound=Callable[..., HttpResponseBase])
+_VIEW = TypeVar("_VIEW", bound=Callable[..., HttpResponseBase | Awaitable[HttpResponseBase]])
 
 def user_passes_test(
     test_func: Callable[[_UserModel | AnonymousUser], bool],

tests/typecheck/contrib/auth/test_decorators.yml

@@ -6,6 +6,14 @@
         @login_required
         def view_func(request: HttpRequest) -> HttpResponse: ...
         reveal_type(view_func)  # N: Revealed type is "def (request: django.http.request.HttpRequest) -> django.http.response.HttpResponse"
+-   case: login_required_bare_async
+    main: |
+        from typing import Any
+        from django.contrib.auth.decorators import login_required
+        from django.http import HttpRequest, HttpResponse
+        @login_required
+        async def view_func(request: HttpRequest) -> HttpResponse: ...
+        reveal_type(view_func)  # N: Revealed type is "def (request: django.http.request.HttpRequest) -> typing.Coroutine[Any, Any, django.http.response.HttpResponse]"
 -   case: login_required_fancy
     main: |
         from django.contrib.auth.decorators import login_required
@@ -14,6 +22,14 @@
         @login_required(redirect_field_name='a', login_url='b')
         def view_func(request: WSGIRequest, arg: str) -> HttpResponse: ...
         reveal_type(view_func)  # N: Revealed type is "def (request: django.core.handlers.wsgi.WSGIRequest, arg: builtins.str) -> django.http.response.HttpResponse"
+-   case: login_required_fancy_async
+    main: |
+        from django.contrib.auth.decorators import login_required
+        from django.core.handlers.asgi import ASGIRequest
+        from django.http import HttpResponse
+        @login_required(redirect_field_name='a', login_url='b')
+        async def view_func(request: ASGIRequest, arg: str) -> HttpResponse: ...
+        reveal_type(view_func)  # N: Revealed type is "def (request: django.core.handlers.asgi.ASGIRequest, arg: builtins.str) -> typing.Coroutine[Any, Any, django.http.response.HttpResponse]"
 -   case: login_required_weird
     main: |
         from django.contrib.auth.decorators import login_required
@@ -35,6 +51,13 @@
         @user_passes_test(lambda u: u.get_username().startswith('super'))
         def view_func(request: HttpRequest) -> HttpResponse: ...
         reveal_type(view_func)  # N: Revealed type is "def (request: django.http.request.HttpRequest) -> django.http.response.HttpResponse"
+-   case: user_passes_test_async
+    main: |
+        from django.contrib.auth.decorators import user_passes_test
+        from django.http import HttpRequest, HttpResponse
+        @user_passes_test(lambda u: u.get_username().startswith('super'))
+        async def view_func(request: HttpRequest) -> HttpResponse: ...
+        reveal_type(view_func)  # N: Revealed type is "def (request: django.http.request.HttpRequest) -> typing.Coroutine[Any, Any, django.http.response.HttpResponse]"
 -   case: user_passes_test_bare_is_error
     main: |
         from django.http import HttpRequest, HttpResponse
@@ -47,3 +70,11 @@
         from django.http import HttpRequest, HttpResponse
         @permission_required('polls.can_vote')
         def view_func(request: HttpRequest) -> HttpResponse: ...
+        reveal_type(view_func)  # N: Revealed type is "def (request: django.http.request.HttpRequest) -> django.http.response.HttpResponse"
+-   case: permission_required_async
+    main: |
+        from django.contrib.auth.decorators import permission_required
+        from django.http import HttpRequest, HttpResponse
+        @permission_required('polls.can_vote')
+        async def view_func(request: HttpRequest) -> HttpResponse: ...
+        reveal_type(view_func)  # N: Revealed type is "def (request: django.http.request.HttpRequest) -> typing.Coroutine[Any, Any, django.http.response.HttpResponse]"