Add files via upload

xtnctx · Nov 8, 2022 · 64ffa52 · 64ffa52
1 parent 3af1d71
commit 64ffa52
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/bfrbsysAPI/apis/serializers.py b/bfrbsysAPI/apis/serializers.py
@@ -1,5 +1,7 @@
 from rest_framework import serializers
 from .models import Item, TrainedModel
+from django.contrib.auth.models import User
+from rest_framework.validators import UniqueTogetherValidator
 
 class ItemSerializer(serializers.ModelSerializer):
     class Meta:
@@ -9,4 +11,27 @@ class Meta:
 class TrainedModelSerializer(serializers.ModelSerializer):
     class Meta:
         model = TrainedModel
-        fields = '__all__'
+        fields = '__all__'
+
+
+
+class UserSerializer(serializers.ModelSerializer):
+    def create(self, validated_data):
+        user = User.objects.create_user(**validated_data)
+        return user
+
+    class Meta:
+        model = User
+        fields = (
+            'username',
+            'first_name',
+            'last_name',
+            'email',
+            'password',
+        )
+        validators = [
+            UniqueTogetherValidator(
+                queryset=User.objects.all(),
+                fields=['username', 'email']
+            )
+        ]