add type alias _UserModel = type[_User]

Author: terencehonles

django-stubs/contrib/auth/__init__.pyi

@@ -1,7 +1,7 @@
 from typing import Any
 
 from django.contrib.auth.backends import BaseBackend
-from django.contrib.auth.models import _AnyUser, _User
+from django.contrib.auth.models import _AnyUser, _User, _UserModel
 from django.db.models.options import Options
 from django.http.request import HttpRequest
 from django.test.client import Client
@@ -23,7 +23,7 @@ def login(request: HttpRequest, user: _User | None, backend: type[BaseBackend] |
 async def alogin(request: HttpRequest, user: _User | None, backend: type[BaseBackend] | str | None = ...) -> None: ...
 def logout(request: HttpRequest) -> None: ...
 async def alogout(request: HttpRequest) -> None: ...
-def get_user_model() -> type[_User]: ...
+def get_user_model() -> _UserModel: ...
 def get_user(request: HttpRequest | Client) -> _AnyUser: ...
 async def aget_user(request: HttpRequest | Client) -> _AnyUser: ...
 def get_permission_codename(action: str, opts: Options) -> str: ...

django-stubs/contrib/auth/backends.pyi

@@ -1,11 +1,11 @@
-from typing import Any, TypeAlias
+from typing import Any
 
-from django.contrib.auth.models import Permission, _AnyUser, _User, _UserType
+from django.contrib.auth.models import Permission, _AnyUser, _User, _UserModel, _UserType
 from django.db.models import QuerySet
 from django.db.models.base import Model
 from django.http.request import HttpRequest
 
-UserModel: TypeAlias = type[_User]
+UserModel = _UserModel
 
 class BaseBackend:
     def authenticate(self, request: HttpRequest | None, **kwargs: Any) -> _User | None: ...

django-stubs/contrib/auth/forms.pyi

@@ -1,9 +1,9 @@
 from collections.abc import Iterable
 from logging import Logger
-from typing import Any, Generic, TypeAlias
+from typing import Any, Generic
 
 from django import forms
-from django.contrib.auth.models import _User, _UserType
+from django.contrib.auth.models import _User, _UserModel, _UserType
 from django.contrib.auth.tokens import PasswordResetTokenGenerator
 from django.core.exceptions import ValidationError
 from django.db import models
@@ -14,8 +14,7 @@ from django.http.request import HttpRequest
 from django.utils.functional import _StrOrPromise
 
 logger: Logger
-
-UserModel: TypeAlias = type[_User]
+UserModel = _UserModel
 
 class ReadOnlyPasswordHashWidget(forms.Widget):
     template_name: str

django-stubs/contrib/auth/models.pyi

@@ -17,6 +17,8 @@ from typing_extensions import Self
 # `HttpRequest.user`
 _User: TypeAlias = AbstractBaseUser
 
+_UserModel: TypeAlias = type[_User]
+
 _AnyUser: TypeAlias = _User | AnonymousUser
 
 # These are only needed for generic classes in order to bind to a specific implementation
@@ -25,7 +27,7 @@ _AnyUserType = TypeVar("_AnyUserType", bound=_AnyUser)  # noqa: PYI018
 # do not use the alias `_User` so the bound remains at `AbstractUser`
 _UserType = TypeVar("_UserType", bound=AbstractUser)
 
-def update_last_login(sender: type[_User], user: _User, **kwargs: Any) -> None: ...
+def update_last_login(sender: _UserModel, user: _User, **kwargs: Any) -> None: ...
 
 class PermissionManager(models.Manager[Permission]):
     def get_by_natural_key(self, codename: str, app_label: str, model: str) -> Permission: ...

django-stubs/contrib/auth/views.pyi

@@ -1,13 +1,13 @@
-from typing import Any, TypeAlias
+from typing import Any
 
 from django.contrib.auth.forms import AuthenticationForm
-from django.contrib.auth.models import _User
+from django.contrib.auth.models import _User, _UserModel
 from django.http.request import HttpRequest
 from django.http.response import HttpResponse, HttpResponseRedirect
 from django.views.generic.base import TemplateView
 from django.views.generic.edit import FormView
 
-UserModel: TypeAlias = type[_User]
+UserModel = _UserModel
 
 class RedirectURLMixin:
     next_page: str | None