validate img size

Author: rockstarr-programmerr

account/serializers.py

@@ -1,4 +1,6 @@
+from django.conf import settings
 from django.contrib.auth import get_user_model
+from django.utils.translation import gettext as _
 from rest_framework import serializers
 
 User = get_user_model()
@@ -15,6 +17,7 @@ class Meta:
             'avatar': {'allow_null': True},
             'avatar_thumbnail': {'read_only': True},
             'user_type': {'read_only': True},
+            'email': {'read_only': True}
         }
 
     def validate(self, attrs):
@@ -24,7 +27,13 @@ def validate(self, attrs):
         return attrs
 
     def validate_avatar(self, img):
-        return img  # TODO
+        size = img.size / 1e6  # bytes to megabytes
+        if size > settings.MAX_UPLOAD_SIZE_MEGABYTES:
+            raise serializers.ValidationError(
+                _('File size must not exceed %dMB.') % settings.MAX_UPLOAD_SIZE_MEGABYTES,
+                code='exceed_max_upload_size'
+            )
+        return img
 
 
 class RegisterTeacherSerializer(UserSerializer):

account/views.py

@@ -1,4 +1,5 @@
 from django.contrib.auth import get_user_model
+from django.db.models import Q
 from django.utils.decorators import classonlymethod, method_decorator
 from django.views.decorators.debug import sensitive_post_parameters
 from rest_framework import mixins
@@ -9,6 +10,7 @@
 from rest_framework.viewsets import GenericViewSet
 
 from account import business
+from account.managers import UserTypes
 from account.serializers import (MeSerializer, RegisterTeacherSerializer,
                                  UserSerializer)
 
@@ -34,7 +36,23 @@ class UserViewSet(mixins.ListModelMixin,
     serializer_class = UserSerializer
 
     def get_queryset(self):
-        return User.objects.all()  # TODO: if teacher: return students, if student: return teacher and classmates
+        user = self.request.user
+
+        if user.is_teacher():
+            classrooms = user.classrooms_teaching.all()
+        else:
+            classrooms = user.classrooms_studying.all()
+
+        classrooms = classrooms.select_related('teacher').prefetch_related('students')
+
+        users_pk = []
+        for classroom in classrooms:
+            teacher = classroom.teacher
+            students = classroom.students.all()
+            users_pk.append(teacher.pk)
+            users_pk.extend(student.pk for student in students)
+
+        return User.objects.filter(pk__in=users_pk)
 
     @action(
         methods=['POST'], detail=False, url_path='register-teacher',

classroom/serializers/exercise.py

@@ -1,3 +1,4 @@
+from django.conf import settings
 from django.utils.translation import gettext as _
 from rest_framework import serializers
 
@@ -65,4 +66,10 @@ class ReadingExerciseUploadImgSerializer(serializers.Serializer):
     image_url = serializers.URLField(read_only=True)
 
     def validate_image(self, image):
-        return image  # TODO
+        size = image.size / 1e6  # bytes to megabytes
+        if size > settings.MAX_UPLOAD_SIZE_MEGABYTES:
+            raise serializers.ValidationError(
+                _('File size must not exceed %dMB.') % settings.MAX_UPLOAD_SIZE_MEGABYTES,
+                code='exceed_max_upload_size'
+            )
+        return image

keep_learning/settings.py

@@ -281,3 +281,5 @@
         del REST_FRAMEWORK['DEFAULT_THROTTLE_RATES']
 
 WEB_LOGIN_URL = env('WEB_LOGIN_URL')
+
+MAX_UPLOAD_SIZE_MEGABYTES = 10