当前位置：极客教程 > Django > Django 问答 > Django Django REST Framework 创建自定义用户

Django Django REST Framework 创建自定义用户

Django Django REST Framework 创建自定义用户

在本文中，我们将介绍使用Django和Django REST Framework创建自定义用户的步骤和技巧。在许多应用程序中，使用默认的Django用户模型可能无法满足需求。为了适应特定的应用需求，我们可以创建自己的用户模型，并通过Django REST Framework进行序列化和验证。

阅读更多：Django 教程

1. 创建自定义用户模型

在开始之前，我们需要先创建一个新的自定义用户模型。通常，一个用户模型将包含标识用户的一些基本信息，例如用户名、电子邮件地址和密码等。为了创建自定义用户模型，我们需要执行以下步骤：

  1. 在Django项目的models.py文件中导入AbstractBaseUser和BaseUserManager。
  2. 创建一个名为CustomUserManager的类，继承自BaseUserManager，并添加必要的方法来创建和保存用户。
  3. 创建一个名为CustomUser的类，继承自AbstractBaseUser，并定义自定义用户模型的字段和属性。
  4. 在CustomUser类中设置USERNAME_FIELD和REQUIRED_FIELDS变量，分别指定用户登录所使用的字段和其他必填字段。
  5. 通过在settings.py文件中修改AUTH_USER_MODEL设置来告诉Django使用自定义用户模型。

下面是一个示例：

from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
from django.db import models

class CustomUserManager(BaseUserManager):
    def create_user(self, email, username, password=None):
        if not email:
            raise ValueError("Email address is required.")
        if not username:
            raise ValueError("Username is required.")

        email = self.normalize_email(email)
        user = self.model(username=username, email=email)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, email, username, password=None):
        user = self.create_user(email, username, password)
        user.is_admin = True
        user.save(using=self._db)
        return user

class CustomUser(AbstractBaseUser):
    email = models.EmailField(verbose_name="Email", max_length=255, unique=True)
    username = models.CharField(verbose_name="Username", max_length=30, unique=True)
    is_active = models.BooleanField(verbose_name="Active", default=True)
    is_admin = models.BooleanField(verbose_name="Admin", default=False)

    objects = CustomUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['username']

    def __str__(self):
        return self.email

    def has_perm(self, perm, obj=None):
        return True

    def has_module_perms(self, app_label):
        return True

    @property
    def is_staff(self):
        return self.is_admin
Python

2. 配置Django REST Framework

完成自定义用户模型的创建后，我们需要配置Django REST Framework以便使用该模型进行序列化和验证。首先，安装Django REST Framework：

pip install djangorestframework
Bash

然后，在项目的settings.py文件中进行如下配置：

INSTALLED_APPS = [
    ...
    'rest_framework',
    ...
]

REST_FRAMEWORK = {
    'DEFAULT_AUTHENTICATION_CLASSES': (
        'rest_framework.authentication.TokenAuthentication',
    ),
    'DEFAULT_PERMISSION_CLASSES': (
        'rest_framework.permissions.IsAuthenticated',
    ),
}
Python

上述配置中包含了Django REST Framework的认证和权限设置，我们可以根据项目的需要进行自定义。

3. 创建序列化器和视图

接下来，我们需要创建一个序列化器和相关的视图类来处理自定义用户模型的数据。序列化器将定义模型的序列化和反序列化规则，而视图将处理与用户数据相关的HTTP请求。

首先，创建一个名为CustomUserSerializer的序列化器类，并添加必要的字段和验证规则：

from rest_framework import serializers
from .models import CustomUser

class CustomUserSerializer(serializers.ModelSerializer):
    class Meta:
        model = CustomUser
        fields = ('id', 'email', 'username', 'password')
        extra_kwargs = {'password': {'write_only': True}}

    def create(self, validated_data):
        user = CustomUser.objects.create_user(
            email=validated_data['email'],
            username=validated_data['username'],
            password=validated_data['password']
        )
        return user
Python

在上述示例中，我们定义了CustomUserSerializer类，并指定了需要序列化和反序列化的字段。注意，我们将密码字段的write_only属性设置为True，以便在反序列化时忽略密码字段。

接下来，创建一个名为CustomUserView的视图类，用于处理与用户数据相关的HTTP请求：

from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status

class CustomUserView(APIView):
    def post(self, request):
        serializer = CustomUserSerializer(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
Python

在上述示例中，我们定义了CustomUserView类，并在其中实现了一个post方法来处理POST请求。它将接收用户数据，将其传递给序列化器进行验证和保存。

4. 配置URL

最后一步是配置URL，以便将请求映射到视图类。在项目的urls.py文件中添加以下代码：

from django.urls import path
from .views import CustomUserView

urlpatterns = [
    path('api/register/', CustomUserView.as_view()),
]
Python

上述代码将POST请求映射到名为/api/register/的URL，并使用CustomUserView处理该请求。

总结

在本文中，我们介绍了使用Django和Django REST Framework创建自定义用户的步骤和技巧。我们学习了如何创建自定义用户模型，并通过Django REST Framework进行序列化和验证。我们还创建了一个序列化器和视图来处理用户数据，并配置了URL以将请求映射到相应的视图类。

通过使用自定义用户模型和Django REST Framework，我们可以更好地满足特定应用程序的用户需求，提供更灵活和可定制的用户管理功能。

Python教程

Python 教程

Python 教程
Tkinter 教程

Tkinter 教程
Pandas 教程

Pandas 教程
NumPy 教程

NumPy 教程
Flask 教程

Flask 教程
Django 教程

Django 教程
PySpark 教程

PySpark 教程
wxPython 教程

wxPython 教程
SymPy 教程

SymPy 教程
Seaborn 教程

Seaborn 教程
SciPy 教程

SciPy 教程
RxPY 教程

RxPY 教程
Pycharm 教程

Pycharm 教程
Pygame 教程

Pygame 教程
PyGTK 教程

PyGTK 教程
PyQt 教程

PyQt 教程
PyQt5 教程

PyQt5 教程
PyTorch 教程

PyTorch 教程
Matplotlib 教程

Matplotlib 教程
Web2py 教程

Web2py 教程
BeautifulSoup 教程

BeautifulSoup 教程

Java教程

Java 教程

Java 教程

Web教程

HTML 教程

HTML 教程
CSS 教程

CSS 教程
CSS3 教程

CSS3 教程
jQuery 教程

jQuery 教程
Ajax 教程

Ajax 教程
AngularJS 教程

AngularJS 教程
TypeScript 教程

TypeScript 教程
WordPress 教程

WordPress 教程
Laravel 教程

Laravel 教程
Next.js 教程

Next.js 教程
PhantomJS 教程

PhantomJS 教程
Three.js 教程

Three.js 教程
Underscore.JS 教程

Underscore.JS 教程
WebGL 教程

WebGL 教程
WebRTC 教程

WebRTC 教程
VueJS 教程

VueJS 教程

数据库教程

SQL 教程

SQL 教程
MySQL 教程

MySQL 教程
MongoDB 教程

MongoDB 教程
PostgreSQL 教程

PostgreSQL 教程
SQLite 教程

SQLite 教程
Redis 教程

Redis 教程
MariaDB 教程

MariaDB 教程

图形图像教程

Vulkan 教程

Vulkan 教程
OpenCV 教程

OpenCV 教程

大数据教程

R语言 教程

R语言 教程

开发工具教程

Git 教程

Git 教程
VSCode 教程

VSCode 教程
Docker 教程

Docker 教程
Gerrit 教程

Gerrit 教程
Excel 教程

Excel 教程

计算机教程

Go语言 教程

Go语言 教程
C++ 教程

C++ 教程

Django 精品教程

登录

找回密码？

注册