diff --git a/profiles/viewsets.py b/profiles/viewsets.py index 425cceb..1d902b5 100644 --- a/profiles/viewsets.py +++ b/profiles/viewsets.py @@ -56,11 +56,11 @@ class MyProfileViewSet(viewsets.ModelViewSet): def perform_update(self, serializer, pk=None): profile: ProfileModel = self.get_object() avatar = self.request.data.get("file", None) - if (avatar is not None): + if (avatar is not None and avatar.size <= 2 * 1024 * 1024): if (profile.avatar_url.name != "./profiles/static/avatars/default.avif"): profile.avatar_url.storage.delete(profile.avatar_url.name) profile.avatar_url = avatar - profile.save() + profile.save() def retrieve(self, request: HttpRequest, pk=None): instance: ProfileModel = self.get_object()