自定义用户表，get_user_model 的使用

最新推荐文章于 2025-08-02 16:12:35 发布

原创最新推荐文章于 2025-08-02 16:12:35 发布 · 532 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#django #python #后端

django&drf相关专栏收录该内容

19 篇文章

订阅专栏

本文介绍了如何在自定义用户模型的情况下正确使用 Django 的 get_user_model 方法来获取用户表。此外，还展示了如何在一个 API 视图中利用此方法来获取系统主页的相关数据，包括在线用户数量、总用户数及资产等。

get_user_model的使用

# 若自定义了用户模型，那么必须要用get_user_model方法导入 用户表
rom django.contrib.auth import get_user_model
from django_redis import get_redis_connection
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView

from cmdb.models import Assets
from monitor.models import OnlineUsers

Users = get_user_model()
class HomeAPIView(APIView):
    """
    get:
    系统主页--数据显示

    获取系统主页数据, status: 200(成功), return: 系统主页数据
    """

    def get(self, request):
        data = dict()
        conn = get_redis_connection('user_info')
        data['visits'] = int(conn.get('visits').decode()) if conn.get('visits') else 0
        data['users'] = Users.objects.all().count()
        data['online_users'] = OnlineUsers.objects.all().count()
        data['assets'] = Assets.objects.all().count()
        return Response(data=data, status=status.HTTP_200_OK)