30 lines
1.1 KiB
Python
30 lines
1.1 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 collection.models import Collection
|
|
from utils.methods import image_url_valid, get_image_meta_by_url
|
|
|
|
|
|
class Command(BaseCommand):
|
|
SORL_THUMBNAIL_ALIAS = 'collection_image'
|
|
|
|
def handle(self, *args, **options):
|
|
with transaction.atomic():
|
|
for collection in Collection.objects.all():
|
|
if not image_url_valid(collection.image_url):
|
|
continue
|
|
|
|
_, width, height = get_image_meta_by_url(collection.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:
|
|
collection.image_url = get_thumbnail(
|
|
file_=collection.image_url,
|
|
**settings.SORL_THUMBNAIL_ALIASES[self.SORL_THUMBNAIL_ALIAS]
|
|
).url
|
|
|
|
collection.save()
|