68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
from plone.app.discussion.interfaces import IDiscussionSettings
|
|
from plone.registry.interfaces import IRegistry
|
|
from Products.CMFCore.utils import getToolByName
|
|
from zope.component import getUtility
|
|
|
|
import logging
|
|
|
|
|
|
default_profile = 'profile-plone.app.discussion:default'
|
|
logger = logging.getLogger('plone.app.discussion')
|
|
|
|
|
|
def update_registry(context):
|
|
registry = getUtility(IRegistry)
|
|
registry.registerInterface(IDiscussionSettings)
|
|
|
|
|
|
def update_rolemap(context):
|
|
context.runImportStepFromProfile(default_profile, 'rolemap')
|
|
|
|
|
|
def upgrade_comment_workflows(context):
|
|
# If the current comment workflow is the one_state_workflow, running our
|
|
# import step will change it to comment_one_state_workflow. This is good.
|
|
# If it was anything else, we should restore this. So get the original
|
|
# chain.
|
|
portal_type = 'Discussion Item'
|
|
wf_tool = getToolByName(context, 'portal_workflow')
|
|
orig_chain = list(wf_tool.getChainFor(portal_type))
|
|
|
|
# Run the workflow step. This sets the chain to
|
|
# comment_one_state_workflow.
|
|
context.runImportStepFromProfile(default_profile, 'workflow')
|
|
|
|
# Restore original workflow chain if needed.
|
|
old_workflow = 'one_state_workflow'
|
|
if old_workflow not in orig_chain:
|
|
# Restore the chain. Probably comment_review_workflow.
|
|
wf_tool.setChainForPortalTypes([portal_type], orig_chain)
|
|
elif len(orig_chain) > 1:
|
|
# This is strange, but I guess it could happen.
|
|
if old_workflow in orig_chain:
|
|
# Replace with new one.
|
|
idx = orig_chain.index(old_workflow)
|
|
orig_chain[idx] = 'comment_one_state_workflow'
|
|
# Restore the chain.
|
|
wf_tool.setChainForPortalTypes([portal_type], orig_chain)
|
|
new_chain = list(wf_tool.getChainFor(portal_type))
|
|
workflows = [wf_tool.getWorkflowById(wf_id)
|
|
for wf_id in new_chain]
|
|
|
|
# Now go over the comments, update their role mappings, and reindex the
|
|
# allowedRolesAndUsers index.
|
|
catalog = getToolByName(context, 'portal_catalog')
|
|
for brain in catalog.unrestrictedSearchResults(portal_type=portal_type):
|
|
try:
|
|
comment = brain.getObject()
|
|
for wf in workflows:
|
|
wf.updateRoleMappingsFor(comment)
|
|
comment.reindexObjectSecurity()
|
|
except (AttributeError, KeyError):
|
|
logger.info('Could not reindex comment %s' % brain.getURL())
|
|
|
|
|
|
def update_catalog(context):
|
|
context.runImportStepFromProfile(default_profile, 'catalog')
|