Create a gist now

Instantly share code, notes, and snippets.

What would you like to do?
fix image scales from archetypes scales to dx scales, old scale names to new scale names
# -*- coding: utf-8 -*-
from Products.CMFCore.utils import getToolByName
from plone.app.textfield.value import RichTextValue
from zope.component.hooks import getSite
import logging
log = logging.getLogger(__name__)
# Old scale name to new scale name
IMAGE_SCALE_MAP = {
'icon': 'icon',
'large': 'large',
'listing': 'listing',
'mini': 'mini',
'preview': 'preview',
'thumb': 'thumb',
'tile': 'tile',
# BBB
'article': 'preview',
'artikel': 'preview',
'carousel': 'preview',
'company_index': 'thumb',
'content': 'preview',
'leadimage': 'tile',
'portlet-fullpage': 'large',
'portlet-halfpage': 'large',
'portlet-links': 'thumb',
'portlet': 'thumb',
'staff_crop': 'thumb',
'staff_index': 'thumb',
}
def image_scale_fixer(text):
if text:
for old, new in IMAGE_SCALE_MAP.items():
# replace plone.app.imaging old scale names with new ones
text = text.replace(
'@@images/image/{0}'.format(old),
'@@images/image/{0}'.format(new)
)
# replace AT traversing scales
text = text.replace(
'image_{0}'.format(old),
'@@images/image/{0}'.format(new)
)
return text
def fix_at_image_scales(context):
portal = getSite()
catalog = getToolByName(portal, 'portal_catalog')
query = {}
query['object_provides'] = 'plone.app.contenttypes.behaviors.richtext.IRichText' # noqa
results = catalog(**query)
log.info('There are {0} in total, stating migration...'.format(
len(results)))
for result in results:
try:
obj = result.getObject()
except:
log.warning(
'Not possible to fetch object from catalog result for '
'item: {0}.'.format(result.getPath()))
continue
text = getattr(obj, 'text', None)
if text:
clean_text = image_scale_fixer(text.raw)
if clean_text != text.raw:
obj.text = RichTextValue(
raw=clean_text,
mimeType=text.mimeType,
outputMimeType=text.outputMimeType,
encoding=text.encoding
)
obj.reindexObject(idxs=('SearchableText', ))
log.info('Text cleanup for {0}'.format(
'/'.join(obj.getPhysicalPath())
))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment