From 2f644936ce828c34ac57cd8c2b1261beb5ff1cca Mon Sep 17 00:00:00 2001 From: AdrienLSH Date: Wed, 7 Feb 2024 21:53:36 +0100 Subject: [PATCH] fix: 2MB size limit on profile picture upload --- profiles/viewsets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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()