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()