source: products/quintagroup.transmogrifier.simpleblog2quills/trunk/quintagroup/transmogrifier/simpleblog2quills/activator.py @ 612

from zope.interface import classProvides, implements
from zope import event

from collective.transmogrifier.interfaces import ISectionBlueprint
from collective.transmogrifier.interfaces import ISection
from collective.transmogrifier.utils import defaultMatcher

from Products.CMFCore import utils
from Products.CMFCore.WorkflowCore import WorkflowException

try:
    from zope.interface import alsoProvides
    from quills.core.interfaces.enabled import IWeblogEnhanced, IPossibleWeblog
    from Products.QuillsEnabled.activation import WeblogActivationEvent
except ImportError:
    # this try: ... except: ... clause is needed for this package to work on 
    # plone 2.1, because zcml:condition attribute in zcml doesn't work
    pass

from quintagroup.transmogrifier.simpleblog2quills.adapters import IMAGE_FOLDER

class BlogActivatorSection(object):
    classProvides(ISectionBlueprint)
    implements(ISection)

    def __init__(self, transmogrifier, name, options, previous):
        self.transmogrifier = transmogrifier
        self.context = transmogrifier.context
        self.wftool = utils.getToolByName(self.context, 'portal_workflow')

        self.pathkey = defaultMatcher(options, 'path-key', name, 'path')
        self.flagkey = options.get('flag-key', '_old_type').strip()
        self.previous = previous

    def __iter__(self):
        for item in self.previous:
            pathkey = self.pathkey(*item.keys())[0]

            if not pathkey or self.flagkey not in item:
                yield item; continue

            if item[self.flagkey] != 'Blog':
                yield item; continue

            path = item[pathkey]
            obj = self.context.unrestrictedTraverse(path, None)
            if obj is None:         # path doesn't exist
                yield item; continue

            # pulish 'images' subfolder
            images = getattr(obj, IMAGE_FOLDER, None)
            if images is not None:
                try:
                    self.wftool.doActionFor(images, 'publish')
                except WorkflowException:
                    pass

            if not IWeblogEnhanced.providedBy(obj) and \
                IPossibleWeblog.providedBy(obj):
                alsoProvides(obj, IWeblogEnhanced)
                event.notify(WeblogActivationEvent(obj))

            yield item