Merge pull request #233 from antaliadom-team/feature/preview_recreate

Feature/preview recreate

catalog/admin.py

@@ -1,5 +1,6 @@
 from django.conf import settings
 from django.contrib import admin
+from django.core.management import call_command
 from django.db import models
 from django.utils.html import format_html
 
@@ -197,6 +198,7 @@ class RealEstateAdmin(admin.ModelAdmin):
     list_filter = ('status', 'location', 'property_type')  # 'owner'
     readonly_fields = ('date_added',)
     inlines = (ImageInline,)
+    actions = ['regenerate_previews']
 
     class Media:
         js = ('admin/js/admin_period_visibility.js',)
@@ -212,6 +214,11 @@ def price_with_currency(self, obj):
             return f'{obj.price}{obj.currency}'
         return f'{obj.price}{obj.currency} в {obj.period.lower()}'
 
+    def regenerate_previews(self, request, queryset):
+        call_command('recreate_previews')
+
+    regenerate_previews.short_description = 'Пересоздать превью фотографий'
+
 
 @admin.register(Favorite)
 class FavoriteAdmin(admin.ModelAdmin):

catalog/models.py

@@ -235,7 +235,8 @@ class Meta:
         verbose_name_plural = 'Фотографии'
         ordering = ('-id',)
 
-    def thumbnail_generator(self, infile, outfile, image_size):
+    @staticmethod
+    def thumbnail_generator(infile, outfile, image_size):
         """Генерирует изображения в требуемых размерах."""
         with PillowImage.open(infile) as im:
             im.thumbnail(image_size)
@@ -248,7 +249,8 @@ def thumbnail_generator(self, infile, outfile, image_size):
                     im, image_size, PillowImage.Resampling.LANCZOS, 0.5
                 ).save(outfile, quality=95)
 
-    def filename_generator(self, filepath, size):
+    @staticmethod
+    def filename_generator(filepath, size):
         """Генерирует имя для каждого размера изображения."""
         width, height = size
         name, extension = os.path.splitext(filepath)

core/management/commands/recreate_previews.py

@@ -0,0 +1,45 @@
+from django.conf import settings
+from django.core.management import BaseCommand
+
+from catalog.models import Image
+
+
+class Command(BaseCommand):
+    help = 'Regenerate previews for all real estate pictures.'
+
+    def handle(self, *args, **kwargs):
+        """Regenerate previews for all real estate pictures."""
+
+        all_images = Image.objects.all()
+        self.stdout.write('Deleting existing previews...')
+        for image in all_images:
+            try:
+                image.delete_thumbnails()
+            except FileNotFoundError:
+                all_images.pop(image)
+                continue
+        self.stdout.write('Generating new previews...')
+        for image in all_images:
+            for size in settings.PREVIEW_SIZES:
+                try:
+                    Image.thumbnail_generator(
+                        infile=image.image.path,
+                        outfile=image.filename_generator(
+                            image.image.path, size
+                        ),
+                        image_size=size,
+                    )
+                except FileNotFoundError:
+                    self.stdout.write(
+                        self.style.WARNING(
+                            f'Photo of the object #id:{image.real_estate.id} '
+                            'found in the DB but file '
+                            f'...{image.image.path[-30:]} not found... '
+                            'skipping'
+                        )
+                    )
+                except Exception as error:
+                    self.stdout.write(self.style.ERROR(error))
+        self.stdout.write(
+            self.style.SUCCESS('Previews regenerated successfully.')
+        )