from rest_framework import serializers
from .models import MyUser


class RegisterViewSerializer(serializers.Serializer):
    username = serializers.CharField(help_text="注册账户名", required=True)
    password = serializers.CharField(help_text="注册密码", write_only=True)
    user_modules = serializers.IntegerField(help_text="用户所属模块", required=True)

    def create(self, validated_data):
        try:
            user = MyUser.objects.create_user(**validated_data)
        except Exception as e:
            raise ValueError("用户创建失败,已存在用户名")
        return user