debian-django-ckeditor/ckeditor/management/commands/generateckeditorthumbnails.py

39 lines
1.3 KiB
Python

import os
from django.conf import settings
from django.core.management.base import NoArgsCommand
from ckeditor.views import get_image_files
from ckeditor.utils import get_thumb_filename
from ckeditor.image_processing import get_backend
class Command(NoArgsCommand):
"""
Creates thumbnail files for the CKEditor file image browser.
Useful if starting to use django-ckeditor with existing images.
"""
def handle_noargs(self, **options):
if getattr(settings, 'CKEDITOR_IMAGE_BACKEND', None):
backend = get_backend()
for image in get_image_files():
if not self._thumbnail_exists(image):
self.stdout.write("Creating thumbnail for %s" % image)
try:
backend.create_thumbnail(image)
except Exception as e:
self.stdout.write("Couldn't create thumbnail for %s: %s" % (image, e))
self.stdout.write("Finished")
else:
self.stdout.write("No thumbnail backend is enabled")
def _thumbnail_exists(self, image_path):
thumb_path = self._to_absolute_path(
get_thumb_filename(image_path)
)
return os.path.isfile(thumb_path)
@staticmethod
def _to_absolute_path(image_path):
return os.path.join(settings.MEDIA_ROOT, image_path)