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


class Student(models.Model):
    user = models.ForeignKey(User)
    grade = models.IntegerField()

from rest_framework import serializers

from users.models import Student


class StudentSerializer(serializers.ModelSerializer):
    class Meta:
        model = Student
        fields = ('username', 'password', 'grade')

    def create(self, validated_data):
        user = validated_data.get('user')
        grade = validated_data.get('grade')
        student = Student.objects.create(self, user=user, grade=grade)
        return student

class CreateUser(APIView):
    def post(self, request, format=None):
        serializer = StudentSerializer(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

class StudentSerializer(serializers.ModelSerializer):
    password = serializers.CharField(write_only=True)
    class Meta:
        model = Student
        fields = ('username', 'password', 'grade')

    def create(self, validated_data):
        user = validated_data.get('user')
        grade = validated_data.get('grade')
        student = Student.objects.create(user=user, grade=grade)
        student.set_password(validated_data.get('password')
        return student