image_file.file

ita-social-projects · Oct 8, 2023 · 4ac70bf · 4ac70bf
1 parent e8057a2
commit 4ac70bf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/profiles/tests/test_crud_profile.py b/profiles/tests/test_crud_profile.py
@@ -28,11 +28,12 @@ def _generate_image(ext, size):
         image.save(file, formatext)
         file.name = f"test.{formatext}"
         file.seek(0)
+        # print(file.name, sys.getsizeof(file), sys.getsizeof(image), type(image))
         return file
 
     def setUp(self) -> None:
         # self.right_image = self._generate_image("jpeg", (10, 10))
-        self.right_image = self._generate_image("png", (1, 1))
+        self.right_image = self._generate_image("png", (100, 100))
         self.wrong_image = self._generate_image("png", (5, 5))
         self.user = UserFactory(email="[email protected]")
         self.profile = ProfileStartupFactory.create(

diff --git a/validation/validate_image.py b/validation/validate_image.py
@@ -12,10 +12,12 @@ def validate_image_format(image: Image):
         )
 
 
-def validate_image_size(image: Image):
+def validate_image_size(image_file):
     import sys
 
-    max_size = sys.getsizeof(image)
-    print(f"{max_size=}")
-    if max_size > 50:
+    max_size = sys.getsizeof(image_file.file)
+    # print(image_file, f"{max_size=}", f"{type(image_file)}")
+    # print(sys.getsizeof(image_file.file))
+    # print(image_file.__dict__)
+    if max_size > 50000:
         raise ValidationError("Image size exceeds the maximum allowed (50MB).")