request.user 返回抽象用户（django-stubs）

Question

我正在尝试输入以下内容：

user: Final[CustomUser] = self.request.user

CustomUser 是我在settings.py 中定义的自定义用户模型。我的 IDE 抱怨 Expected type 'CustomUser', got 'AbstractBaseUser' instead。查看文档，它说self.request.user 返回一个AUTH_USER_MODEL 的实例，但显然这没有发生。

也试过了：

 def get(self, request: MyRequest, *args, **kwargs) -> HttpResponse:
    user: Final[CustomUser] = self.request.user

在哪里

from django.http import HttpRequest
from my_user_app.models import CustomUser

class MyRequest(HttpRequest):
    user: CustomUser

Answer 1

这里有一个关于此错误的未解决问题https://github.com/typeddjango/django-stubs/issues/309

根据 django-stubs 的文档，您必须在使用自定义用户模型时提供自己的请求类型。在 PyPi 的页面底部How can I use HttpRequest with custom user model?

from django.http import HttpRequest
from my_user_app.models import CustomUser

class MyRequest(HttpRequest):
    user: CustomUser

那么你将不得不使用它来注释self.request

 def get(self, request: MyRequest, *args, **kwargs) -> HttpResponse:
    user: Final[CustomUser] = request.user