29 lines
1.2 KiB
Python
29 lines
1.2 KiB
Python
from django.conf import settings
|
|
from django.core.management.base import BaseCommand
|
|
from django.db import transaction
|
|
from sorl.thumbnail import get_thumbnail
|
|
|
|
from establishment.models import Establishment
|
|
from utils.methods import image_url_valid, get_image_meta_by_url
|
|
|
|
|
|
class Command(BaseCommand):
|
|
SORL_THUMBNAIL_ALIAS = 'establishment_collection_image'
|
|
|
|
def handle(self, *args, **options):
|
|
with transaction.atomic():
|
|
for establishment in Establishment.objects.all():
|
|
if not image_url_valid(establishment.preview_image_url):
|
|
continue
|
|
|
|
_, width, height = get_image_meta_by_url(establishment.preview_image_url)
|
|
sorl_settings = settings.SORL_THUMBNAIL_ALIASES[self.SORL_THUMBNAIL_ALIAS]
|
|
sorl_width_height = sorl_settings['geometry_string'].split('x')
|
|
|
|
if int(sorl_width_height[0]) > width or int(sorl_width_height[1]) > height:
|
|
establishment.preview_image_url = get_thumbnail(
|
|
file_=establishment.preview_image_url,
|
|
**sorl_settings
|
|
)
|
|
establishment.save()
|