Django 2.2.5，使用Django的内置身份验证系统支持多种用户类型

from django.contrib.auth.models import AbstractUser
from django.db import models

class UserType(models.TextChoices):
    ADMIN = 'ADMIN', '管理员'
    USER = 'USER', '普通用户'
    GUEST = 'GUEST', '访客'

class CustomUser(AbstractUser):
    user_type = models.CharField(max_length=10, choices=UserType.choices, default=UserType.USER)

    def is_admin(self):
        return self.user_type == UserType.ADMIN

    def is_guest(self):
        return self.user_type == UserType.GUEST

from django.contrib.auth.decorators import user_passes_test

def admin_required(view_func):
    decorated_view_func = user_passes_test(
        lambda u: u.is_authenticated and u.is_admin(),
        login_url='admin_login'
    )
    return decorated_view_func(view_func)

# 在视图中使用装饰器
@admin_required
def admin_only_view(request):
    # 只有管理员可以访问的视图逻辑
    pass

from django import forms
from .models import CustomUser

class UserTypeForm(forms.ModelForm):
    class Meta:
        model = CustomUser
        fields = ['username', 'password', 'user_type']

# 在视图中处理表单提交
def register(request):
    if request.method == 'POST':
        form = UserTypeForm(request.POST)
        if form.is_valid():
            user = form.save(commit=False)
            user.set_password(form.cleaned_data['password'])
            user.save()
            # 登录用户等后续操作
    else:
        form = UserTypeForm()
    return render(request, 'register.html', {'form': form})