Bundle trimed down alternative to photologue

photologue/management/commands/plflush.py

@@ -0,0 +1,34 @@
+# 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 = 'Clears the Photologue cache for the given sizes.'
+
+    def add_arguments(self, parser):
+        parser.add_argument('sizes',
+                            nargs='*',
+                            type=str,
+                            help='Name of the photosize.')
+
+    def handle(self, *args, **options):
+        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('Flushing cache...')
+
+        for cls in ImageModel.__subclasses__():
+            for photosize in photosizes:
+                print('Flushing %s size images' % photosize.name)
+                for obj in cls.objects.all():
+                    obj.remove_size(photosize)