# Based on https://github.com/richardbarran/django-photologue/
# by Richard Barran, BSD-3 licensed

from django.core.management.base import BaseCommand, CommandError
from photologue.models import ImageModel, PhotoSize


class Command(BaseCommand):
    help = "Manages Photologue cache file for the given sizes."

    def add_arguments(self, parser):
        parser.add_argument("sizes", nargs="*", type=str, help="Name of the photosize.")
        parser.add_argument(
            "--reset",
            action="store_true",
            default=False,
            dest="reset",
            help="Reset photo cache before generating.",
        )

    def handle(self, *args, **options):
        reset = options["reset"]
        sizes = options["sizes"]

        if not sizes:
            photosizes = PhotoSize.objects.all()
        else:
            photosizes = PhotoSize.objects.filter(name__in=sizes)

        if not len(photosizes):
            raise CommandError("No photo sizes were found.")

        print("Caching photos, this may take a while...")

        for cls in ImageModel.__subclasses__():
            for photosize in photosizes:
                print("Cacheing %s size images" % photosize.name)
                for obj in cls.objects.all():
                    if reset:
                        obj.remove_size(photosize)
                    obj.create_size(photosize)