Andrej Spielmann
28222d634d
- Add role field to UserProfile (superadmin/admin/trainer) - Add role-based permission classes - Create UserManagementViewSet with CRUD and password change - Add API types and components for user management - Create users management page in settings - Only superadmins can manage users
113 lines
3.8 KiB
Python
113 lines
3.8 KiB
Python
from rest_framework import serializers
|
|
from django.contrib.auth.models import User
|
|
from .models import UserPreferences, UserProfile
|
|
|
|
|
|
class UserSerializer(serializers.ModelSerializer):
|
|
club_id = serializers.SerializerMethodField()
|
|
club_name = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['id', 'username', 'email', 'first_name', 'last_name', 'club_id', 'club_name']
|
|
read_only_fields = ['id']
|
|
|
|
def get_club_id(self, obj):
|
|
if hasattr(obj, 'profile') and obj.profile and obj.profile.club:
|
|
return obj.profile.club.id
|
|
return None
|
|
|
|
def get_club_name(self, obj):
|
|
if hasattr(obj, 'profile') and obj.profile and obj.profile.club:
|
|
return obj.profile.club.name
|
|
return None
|
|
|
|
|
|
class LoginSerializer(serializers.Serializer):
|
|
username = serializers.CharField()
|
|
password = serializers.CharField(write_only=True)
|
|
|
|
|
|
class RegisterSerializer(serializers.ModelSerializer):
|
|
password = serializers.CharField(write_only=True, min_length=8)
|
|
password_confirm = serializers.CharField(write_only=True)
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['username', 'email', 'password', 'password_confirm', 'first_name', 'last_name']
|
|
|
|
def validate_email(self, value):
|
|
if User.objects.filter(email=value).exists():
|
|
raise serializers.ValidationError('A user with this email already exists')
|
|
return value
|
|
|
|
def validate(self, attrs):
|
|
if attrs['password'] != attrs['password_confirm']:
|
|
raise serializers.ValidationError({'password_confirm': 'Passwords do not match'})
|
|
return attrs
|
|
|
|
def create(self, validated_data):
|
|
validated_data.pop('password_confirm')
|
|
user = User.objects.create_user(
|
|
username=validated_data['username'],
|
|
email=validated_data.get('email', ''),
|
|
password=validated_data['password'],
|
|
first_name=validated_data.get('first_name', ''),
|
|
last_name=validated_data.get('last_name', ''),
|
|
)
|
|
return user
|
|
|
|
|
|
class UserPreferencesSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = UserPreferences
|
|
fields = '__all__'
|
|
|
|
|
|
class UserListSerializer(serializers.ModelSerializer):
|
|
role = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['id', 'username', 'email', 'first_name', 'last_name', 'is_active', 'role', 'date_joined']
|
|
|
|
def get_role(self, obj):
|
|
if hasattr(obj, 'profile'):
|
|
return obj.profile.role
|
|
return 'trainer'
|
|
|
|
|
|
class UserCreateSerializer(serializers.ModelSerializer):
|
|
password = serializers.CharField(write_only=True)
|
|
role = serializers.ChoiceField(choices=UserProfile.ROLE_CHOICES, default='trainer')
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['id', 'username', 'email', 'first_name', 'last_name', 'password', 'role']
|
|
|
|
def create(self, validated_data):
|
|
role = validated_data.pop('role', 'trainer')
|
|
user = User.objects.create_user(**validated_data)
|
|
UserProfile.objects.create(user=user, role=role)
|
|
return user
|
|
|
|
|
|
class UserUpdateSerializer(serializers.ModelSerializer):
|
|
role = serializers.ChoiceField(choices=UserProfile.ROLE_CHOICES, required=False)
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['id', 'username', 'email', 'first_name', 'last_name', 'is_active', 'role']
|
|
|
|
def update(self, instance, validated_data):
|
|
role = validated_data.pop('role', None)
|
|
user = super().update(instance, validated_data)
|
|
if role and hasattr(user, 'profile'):
|
|
user.profile.role = role
|
|
user.profile.save()
|
|
return user
|
|
|
|
|
|
class PasswordChangeSerializer(serializers.Serializer):
|
|
password = serializers.CharField(write_only=True, required=True)
|