Django - 如何使用 get_object_or_404 获取自定义用户模型

Question

我想检索我的 User 类的一个实例，它扩展了 AbstractBaseUser 。我还想使用 django get_object_or_404 快捷方式按主键过滤我的用户，但这行不通:

from django.shortcuts import get_object_or_404
from django.conf import settings

user = get_object_or_404(settings.AUTH_USER_MODEL, pk=pk)

引发了以下异常:

"ValueError at /v1/users/24c47b4a-920e-47c7-902c-80c64c0dd657/
First argument to get_object_or_404() must be a Model, Manager, or QuerySet, not 'str'.

在我的 settings.py 中，这是我的 AUTH_USER_MODEL 的样子:

AUTH_USER_MODEL = 'users.User'

有什么办法可以实现吗？

Answer 1

你可以试试这个get_user_model;

from django.contrib.auth import get_user_model
from django.shortcuts import get_object_or_404
from django.conf import settings

user = get_object_or_404(get_user_model(), pk=pk)