django 之admin 后台的Auth user表字段改造,增加自定义 oauth2 认证admin登陆

在Django中,要改造Auth user表并增加对OAuth2认证的支持,你需要定制UserAdmin并添加自定义字段,同时创建相应的模型来处理OAuth2认证信息。以下是一个简化的例子:

首先,定义一个新的模型来存储OAuth2认证信息:




from django.db import models
from django.contrib.auth.models import User
 
class OAuth2Info(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    access_token = models.CharField(max_length=255)
    refresh_token = models.CharField(max_length=255)
    # 其他OAuth2相关字段

然后,定制UserAdmin来管理额外的字段:




from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import OAuth2Info
 
class CustomUserAdmin(UserAdmin):
    list_display = ('username', 'email', 'is_staff', 'is_active', 'oauth2_info')
    list_select_related = ('oauth2info',)
 
    fieldsets = (
        (None, {'fields': ('username', 'password')}),
        ('Personal info', {'fields': ('first_name', 'last_name', 'email')}),
        ('OAuth2 Info', {'fields': ('oauth2info',)}),  # 添加OAuth2信息的字段
        ('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser',
                                       'groups', 'user_permissions')}),
        ('Important dates', {'fields': ('last_login', 'date_joined')}),
    )
 
    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('username', 'password1', 'password2', 'email', 'is_staff', 'is_active'),
        }),
    )
 
    # 注册模型关联
    inlines = (OAuth2InfoInline,)
 
admin.site.unregister(User)
admin.site.register(User, CustomUserAdmin)

最后,创建一个StackedInline来编辑用户的OAuth2信息:




from django.contrib import admin
from django.contrib.auth.models import User
from .models import OAuth2Info
 
class OAuth2InfoInline(admin.StackedInline):
    model = OAuth2Info
    can_delete = False
    verbose_name_plural = 'OAuth2 info'
 
class CustomUserAdmin(UserAdmin):
    inlines = (OAuth2InfoInline,)
 
admin.site.unregister(User)
admin.site.register(User, CustomUserAdmin)

这样,在Django Admin后台,你就可以管理用户的OAuth2认证信息了。记得在settings.py中添加你的应用,以确保上述代码能够被正确加载。

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日