python-3.x - 如何扩展 Django 的身份验证用户管理器以添加我自己的方法?
问题描述
我正在使用 Python 3.9 和 Django 3.2。我想使用 Django 的 auth 模块进行基本用户管理,并将其添加到我的 settings.py 文件中
INSTALLED_APPS = [
'django.contrib.admin',
'django.contrib.auth',
'django.contrib.contenttypes',
'django.contrib.sessions',
'django.contrib.messages',
'django.contrib.staticfiles',
'cbapp',
]
但是,我想将其用户表的主键更改为 UUID,然后添加我自己的方法,所以我在 models/custom_user.py 创建了这个文件
from django.contrib.auth.models import AbstractUser
from django.db import models
class CustomeUserManager(models.Manager):
def get_active_users(self):
qset = CustomUser.objects.filter(
is_active=True
)
return list(qset)
class CustomUser(AbstractUser):
pass
uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
objects = CustomeUserManager()
我仍然想访问 Django 的原始身份验证方法,但似乎我无法再访问它们,例如“create_user”......
>>> user = CustomUser.objects.create_user('Dave', 'myemail@test.com', 'johnpassword')
Traceback (most recent call last):
File "<input>", line 1, in <module>
user = CustomUser.objects.create_user('Dave', 'myemail@test.com', 'johnpassword')
AttributeError: 'CustomeUserManager' object has no attribute 'create_user'
在添加我自己的一些方法的同时扩展 Django 的 auth 用户方法的正确方法是什么?
解决方案
这些是在UserManager
[Django-doc]中指定的,因此您应该继承自UserManager
而不是简单的Manager
:
from django.contrib.auth.models import UserManager
class CustomeUserManager(UserManager):
def get_active_users(self):
return CustomUser.objects.filter(
is_active=True
)
我强烈建议不要返回列表,而是QuerySet
为您的get_active_users
. AQuerySet
可以进一步过滤,因此不会进行数据库查询,以后应该在 Django/Python 层进行后处理。
推荐阅读
- sql - 在 MS SQL Management Studio 中的表中插入时间戳
- php - 如何打破 foreach 循环 - 数组索引
- jenkins - 这些停止 Jenkins 管道的方式有什么区别?
- php - 致命错误:未捕获的错误:在 app/code/DeviceDetector/Yaml/Spyc.php 中找不到类“Spyc”
- python - 是否可以将对象所属的所有者类作为属性引用?
- android - 适用于 android 8.0 及更高版本的 android 通话记录器
- angular - Angular 6 ngrx,如何将新项目添加到状态对象的数组中?
- python - python中的迷宫生成器
- android - Push notification icons are displaying very tiny in Marshmallow devices in Android
- typescript - vue 中的打字稿-“Vue | 类型”上不存在属性“验证” 元素 | Vue[] | 元素[]'。