From 9ea7c250853051aa63fc782b71f831fcf25b5862 Mon Sep 17 00:00:00 2001
From: "a.gorbunov" <a.gorbunov@id-east.ru>
Date: Thu, 16 Jan 2020 07:56:34 +0000
Subject: [PATCH] logs and save extension for preview

---
 .../commands/establishment_optimize_preview_image.py     | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/apps/establishment/management/commands/establishment_optimize_preview_image.py b/apps/establishment/management/commands/establishment_optimize_preview_image.py
index 4cacea1a..cbb02b37 100644
--- a/apps/establishment/management/commands/establishment_optimize_preview_image.py
+++ b/apps/establishment/management/commands/establishment_optimize_preview_image.py
@@ -1,3 +1,5 @@
+import os
+
 from django.conf import settings
 from django.core.management.base import BaseCommand
 from django.db import transaction
@@ -21,8 +23,13 @@ class Command(BaseCommand):
                 sorl_width_height = sorl_settings['geometry_string'].split('x')
 
                 if int(sorl_width_height[0]) > width or int(sorl_width_height[1]) > height:
+                    _, file_ext = os.path.splitext(establishment.preview_image_url)
+
+                    self.stdout.write(self.style.SUCCESS(f'Optimize: {establishment.preview_image_url}, extension: {file_ext}'))
+
                     establishment.preview_image_url = get_thumbnail(
                         file_=establishment.preview_image_url,
-                        **sorl_settings
+                        **sorl_settings,
+                        format='PNG' if file_ext[1:] == 'png' else 'JPEG'
                     ).url
                     establishment.save()