feat: add custom User model with role-based permissions

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
TianyangZhang 2026-03-24 17:11:40 +08:00
parent 4cee7e9569
commit c3466f4866
25 changed files with 274 additions and 0 deletions

View File

View File

View File

@ -0,0 +1,7 @@
from django.apps import AppConfig
class AccountsConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'apps.accounts'
verbose_name = '账户管理'

View File

@ -0,0 +1,48 @@
# Generated by Django 4.2.20 on 2026-03-24 09:09
import django.contrib.auth.models
import django.contrib.auth.validators
from django.db import migrations, models
import django.db.models.deletion
import django.utils.timezone
class Migration(migrations.Migration):
initial = True
dependencies = [
('auth', '0012_alter_user_first_name_max_length'),
('organizations', '__first__'),
]
operations = [
migrations.CreateModel(
name='User',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('password', models.CharField(max_length=128, verbose_name='password')),
('last_login', models.DateTimeField(blank=True, null=True, verbose_name='last login')),
('is_superuser', models.BooleanField(default=False, help_text='Designates that this user has all permissions without explicitly assigning them.', verbose_name='superuser status')),
('username', models.CharField(error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username')),
('first_name', models.CharField(blank=True, max_length=150, verbose_name='first name')),
('last_name', models.CharField(blank=True, max_length=150, verbose_name='last name')),
('email', models.EmailField(blank=True, max_length=254, verbose_name='email address')),
('is_staff', models.BooleanField(default=False, help_text='Designates whether the user can log into this admin site.', verbose_name='staff status')),
('is_active', models.BooleanField(default=True, help_text='Designates whether this user should be treated as active. Unselect this instead of deleting accounts.', verbose_name='active')),
('date_joined', models.DateTimeField(default=django.utils.timezone.now, verbose_name='date joined')),
('role', models.CharField(choices=[('superadmin', '超级管理员'), ('admin', '公司管理员'), ('seeker', '求职者')], default='seeker', max_length=20)),
('phone', models.CharField(blank=True, max_length=20)),
('groups', models.ManyToManyField(blank=True, help_text='The groups this user belongs to. A user will get all permissions granted to each of their groups.', related_name='user_set', related_query_name='user', to='auth.group', verbose_name='groups')),
('organization', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='admins', to='organizations.organization')),
('user_permissions', models.ManyToManyField(blank=True, help_text='Specific permissions for this user.', related_name='user_set', related_query_name='user', to='auth.permission', verbose_name='user permissions')),
],
options={
'verbose_name': '用户',
'verbose_name_plural': '用户',
},
managers=[
('objects', django.contrib.auth.models.UserManager()),
],
),
]

View File

@ -0,0 +1,34 @@
from django.contrib.auth.models import AbstractUser
from django.db import models
class User(AbstractUser):
ROLE_CHOICES = [
('superadmin', '超级管理员'),
('admin', '公司管理员'),
('seeker', '求职者'),
]
role = models.CharField(max_length=20, choices=ROLE_CHOICES, default='seeker')
phone = models.CharField(max_length=20, blank=True)
organization = models.ForeignKey(
'organizations.Organization',
null=True, blank=True,
on_delete=models.SET_NULL,
related_name='admins'
)
@property
def is_superadmin(self):
return self.role == 'superadmin'
@property
def is_admin(self):
return self.role == 'admin'
@property
def is_seeker(self):
return self.role == 'seeker'
class Meta:
verbose_name = '用户'
verbose_name_plural = '用户'

View File

@ -0,0 +1,23 @@
from rest_framework.permissions import BasePermission
class IsSuperAdmin(BasePermission):
def has_permission(self, request, view):
return request.user.is_authenticated and request.user.is_superadmin
class IsCompanyAdmin(BasePermission):
def has_permission(self, request, view):
return request.user.is_authenticated and request.user.is_admin
class IsAdminOrSuperAdmin(BasePermission):
def has_permission(self, request, view):
return request.user.is_authenticated and (
request.user.is_admin or request.user.is_superadmin
)
class IsSeeker(BasePermission):
def has_permission(self, request, view):
return request.user.is_authenticated and request.user.is_seeker

View File

@ -0,0 +1,38 @@
from rest_framework import serializers
from django.contrib.auth import get_user_model
User = get_user_model()
class RegisterSerializer(serializers.ModelSerializer):
password = serializers.CharField(write_only=True, min_length=6)
class Meta:
model = User
fields = ['username', 'email', 'phone', 'password']
def create(self, validated_data):
return User.objects.create_user(**validated_data, role='seeker')
class UserSerializer(serializers.ModelSerializer):
class Meta:
model = User
fields = ['id', 'username', 'email', 'phone', 'role', 'organization']
read_only_fields = ['role']
class AdminUserSerializer(serializers.ModelSerializer):
"""超管用于创建/管理公司管理员账号"""
password = serializers.CharField(write_only=True, min_length=6)
class Meta:
model = User
fields = ['id', 'username', 'email', 'phone', 'role', 'organization', 'password', 'is_active']
def create(self, validated_data):
password = validated_data.pop('password')
user = User(**validated_data)
user.set_password(password)
user.save()
return user

View File

@ -0,0 +1,27 @@
import pytest
from django.contrib.auth import get_user_model
User = get_user_model()
@pytest.mark.django_db
class TestUserModel:
def test_create_seeker(self):
user = User.objects.create_user(
username='seeker1', password='pass123', role='seeker'
)
assert user.role == 'seeker'
assert user.is_seeker is True
assert user.is_admin is False
def test_create_admin(self):
user = User.objects.create_user(
username='admin1', password='pass123', role='admin'
)
assert user.is_admin is True
def test_create_superadmin(self):
user = User.objects.create_user(
username='super1', password='pass123', role='superadmin'
)
assert user.is_superadmin is True

View File

@ -0,0 +1,12 @@
from django.urls import path
from rest_framework_simplejwt.views import TokenObtainPairView, TokenRefreshView
from .views import RegisterView, MeView, UserManageViewSet, UserDetailView
urlpatterns = [
path('register/', RegisterView.as_view()),
path('login/', TokenObtainPairView.as_view()),
path('token/refresh/', TokenRefreshView.as_view()),
path('me/', MeView.as_view()),
path('users/', UserManageViewSet.as_view()),
path('users/<int:pk>/', UserDetailView.as_view()),
]

View File

@ -0,0 +1,40 @@
from rest_framework import generics
from rest_framework.permissions import AllowAny, IsAuthenticated
from rest_framework.response import Response
from rest_framework.views import APIView
from django.contrib.auth import get_user_model
from .serializers import RegisterSerializer, UserSerializer, AdminUserSerializer
from .permissions import IsSuperAdmin
User = get_user_model()
class RegisterView(generics.CreateAPIView):
serializer_class = RegisterSerializer
permission_classes = [AllowAny]
class MeView(APIView):
permission_classes = [IsAuthenticated]
def get(self, request):
return Response(UserSerializer(request.user).data)
def patch(self, request):
serializer = UserSerializer(request.user, data=request.data, partial=True)
serializer.is_valid(raise_exception=True)
serializer.save()
return Response(serializer.data)
class UserManageViewSet(generics.ListCreateAPIView):
"""超管:管理所有用户"""
serializer_class = AdminUserSerializer
permission_classes = [IsSuperAdmin]
queryset = User.objects.all()
class UserDetailView(generics.RetrieveUpdateDestroyAPIView):
serializer_class = AdminUserSerializer
permission_classes = [IsSuperAdmin]
queryset = User.objects.all()

View File

@ -0,0 +1,5 @@
from django.apps import AppConfig
class ApplicationsConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'apps.applications'

View File

@ -0,0 +1,3 @@
from django.urls import path
urlpatterns = []

View File

View File

@ -0,0 +1,5 @@
from django.apps import AppConfig
class JobsConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'apps.jobs'

View File

@ -0,0 +1,3 @@
from django.urls import path
urlpatterns = []

View File

@ -0,0 +1,5 @@
from django.apps import AppConfig
class OrganizationsConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'apps.organizations'

View File

@ -0,0 +1,13 @@
from django.db import models
class Organization(models.Model):
name = models.CharField(max_length=255)
class Meta:
app_label = 'organizations'
verbose_name = '组织'
verbose_name_plural = '组织'
def __str__(self):
return self.name

View File

@ -0,0 +1,3 @@
from django.urls import path
urlpatterns = []

View File

View File

@ -0,0 +1,5 @@
from django.apps import AppConfig
class ResumesConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'apps.resumes'

View File

@ -0,0 +1,3 @@
from django.urls import path
urlpatterns = []