serializers.py 810 B

1234567891011121314151617181920
  1. from rest_framework import serializers
  2. from .models import MyUser
  3. class LoginViewSerializer(serializers.Serializer):
  4. username = serializers.CharField(help_text="账户", required=True)
  5. password = serializers.CharField(help_text="密码", write_only=True)
  6. class RegisterViewSerializer(serializers.Serializer):
  7. username = serializers.CharField(help_text="注册账户名", required=True)
  8. password = serializers.CharField(help_text="注册密码", write_only=True)
  9. user_modules = serializers.IntegerField(help_text="用户所属模块", required=True)
  10. def create(self, validated_data):
  11. try:
  12. user = MyUser.objects.create_user(**validated_data)
  13. except Exception as e:
  14. raise ValueError("用户创建失败,已存在用户名")
  15. return user