Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-04-30 19:44:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle image_formats parameter on ImageFeature

Browse source
This commit is contained in:
Matt Westcott 2017-12-14 16:00:59 +00:00 committed by Thibaud Colas
parent f647c9f270
commit b0f8cd12b2
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
wagtail/images/rich_text.py
View file

@ -6,7 +6,7 @@ from wagtail.admin.rich_text.converters.contentstate_models import Entity
from wagtail.admin.rich_text.converters.html_to_contentstate import AtomicBlockEntityElementHandler
from wagtail.admin.rich_text.editors.draftail.features import EntityFeature
from wagtail.images import get_image_model
from wagtail.images.formats import get_image_format
from wagtail.images.formats import get_image_format, get_image_formats
# Front-end conversion
@ -91,12 +91,17 @@ class ImageFeature(EntityFeature):
Special case of EntityFeature so that we can easily define features that
replicate the default 'image' feature with a custom list of image formats
"""
def __init__(self, image_formats='__all__'):
def __init__(self, image_formats=None):
if image_formats is None:
format_defs = get_image_formats()
else:
format_defs = [get_image_format(f) for f in image_formats]
super().__init__({
'label': 'Image',
'type': ENTITY_TYPES.IMAGE,
'icon': 'icon-image',
'imageFormats': image_formats,
'imageFormats': [{'label': str(f.label), 'value': f.name} for f in format_defs],
'source': 'ImageSource',
'decorator': 'Image',
})