quality calculate

apps/news/management/commands/news_optimize_images.py

@@ -2,12 +2,14 @@
 from django.core.management.base import BaseCommand
 from django.conf import settings
 
-from utils.methods import get_url_images_in_text
+from utils.methods import get_url_images_in_text, get_page_size_by_url
 from news.models import News
 from sorl.thumbnail import get_thumbnail
 
+
 class Command(BaseCommand):
     SORL_THUMBNAIL_ALIASES = 'news_description'
+    IMAGE_MAX_SIZE_IN_BYTES = 1048576  # ~ 1mb
 
     def add_arguments(self, parser):
         parser.add_argument(
@@ -16,24 +18,51 @@ class Command(BaseCommand):
             default=self.SORL_THUMBNAIL_ALIASES,
             help='Ключ для параметров оптимизации',
         )
+        parser.add_argument(
+            '-s',
+            '--size',
+            default=self.IMAGE_MAX_SIZE_IN_BYTES,
+            help='Максимальный размер файла в байтах',
+            type=int
+        )
 
-    def optimize(self, text, alias):
+    def optimize(self, text, alias, max_size):
+        """optimize news images"""
         for image in get_url_images_in_text(text):
+            size = get_page_size_by_url(image)
+
+            if size < max_size:
+                continue
+
+            quality = round(max_size / (size * 0.01))
             optimized_image = get_thumbnail(
                 file_=image,
-                **settings.SORL_THUMBNAIL_ALIASES[alias]
+                **settings.SORL_THUMBNAIL_ALIASES[alias],
+                quality=100,
+                upscale=False
             ).url
             text = text.replace(image, optimized_image)
-            self.stdout.write(self.style.SUCCESS(f'Optimized {image} -> {optimized_image}...\n'))
+            self.stdout.write(self.style.SUCCESS(f'Optimized {image} -> {optimized_image}\n'
+                                                 f'Quality [{quality}%]\n'))
 
         return text
 
     def handle(self, *args, **options):
+        alias = options['alias']
+        size = options['size']
+
+        if alias not in settings.SORL_THUMBNAIL_ALIASES:
+            self.stdout.write(self.style.ERROR(f'{alias} not found in alias'))
+            return
+
+        self.optimize('https://upload.wikimedia.org/wikipedia/commons/b/b9/Pizigani_1367_Chart_1MB.jpg',
+                      'news_description', size)
+
         for news in News.objects.all():
             if not isinstance(news.description, dict):
                 continue
             news.description = {
-                locale: self.optimize(text, options['alias'])
-                    for locale, text in news.description.items()
+                locale: self.optimize(text, alias, size)
+                for locale, text in news.description.items()
             }
-            news.save()
+            news.save()

apps/utils/methods.py

@@ -173,4 +173,9 @@ def section_name_into_index_name(section_name: str):
 
 def get_url_images_in_text(text):
     """Find images urls in text"""
-    return re.findall(r'(?:http:|https:)?//.*\.(?:png|jpg|svg)', text)
+    return re.findall(r'(?:http:|https:)?//.*\.(?:png|jpg|svg)', text)
+
+def get_page_size_by_url(url):
+    """Get page size by Content-Length header"""
+    response = requests.head(url)
+    return int(response.headers.get('content-length'))

project/settings/base.py

@@ -385,7 +385,7 @@ THUMBNAIL_QUALITY = 85
 THUMBNAIL_DEBUG = False
 SORL_THUMBNAIL_ALIASES = {
     'news_preview': {'geometry_string': '300x260', 'crop': 'center'},
-    'news_description': {'geometry_string': '100x100', 'quality': 50},
+    'news_description': {'geometry_string': '100x100'},
     'news_promo_horizontal_web': {'geometry_string': '1900x600', 'crop': 'center'},
     'news_promo_horizontal_mobile': {'geometry_string': '375x260', 'crop': 'center'},
     'news_tile_horizontal_web': {'geometry_string': '300x275', 'crop': 'center'},