source: products/ploneorg.kudobounty/trunk/ploneorg/kudobounty/setuphandlers.py @ 3158

from ploneorg.kudobounty.config import FORM_ID
from ploneorg.kudobounty.config import TOPIC_ID
from ploneorg.kudobounty.config import CONTAINER_ID
from ploneorg.kudobounty.config import CONTAINER_TITLE

from Products.CMFCore.utils import getToolByName
from Products.CMFPlone.utils import _createObjectByType
from Products.ATContentTypes.lib import constraintypes

from Products.GenericSetup.context import TarballExportContext, TarballImportContext
from Products.GenericSetup.interfaces import IFilesystemExporter, IFilesystemImporter

CREATE_SCRIPT_BODY = """
portal = context.portal_url.getPortalObject()
formProcessor = portal.restrictedTraverse('@@processBountyForm')
formProcessor()
return {}
"""

def createTopic(container, wftool, logger):
    """
    Located in: /bounty-mission/aggregator
    Item Type = ["Bounty Program Submission",]
    state = ['published',]
    sort_on = "creation"    
    """
    topic = getattr(container, TOPIC_ID, None)
    if topic is None:
        # Add criteria
        _createObjectByType('Topic', container, id=TOPIC_ID,
                            title=CONTAINER_TITLE)
        topic = getattr(container, TOPIC_ID, None)
        crit = topic.addCriterion('Type','ATPortalTypeCriterion')
        crit.setValue(["Bounty Program Submission",])
        crit = topic.addCriterion('review_state','ATSelectionCriterion')
        crit.setValue("published")
        crit = topic.addCriterion('created','ATSortCriterion') 
        topic.setLayout('folder_summary_view')
        topic.unmarkCreationFlag()

        if wftool.getInfoFor(topic, 'review_state') != 'published':
            wftool.doActionFor(topic, 'publish')
        
        logger.info("Bounty submissions aggregator successfully added")


def createPFGForm(context, container, wftool, logger):
    """
    """
    form = getattr(container, FORM_ID, None)
    if form is not None:
        container.manage_delObjects(ids = FORM_ID)
    # Create new form object
    container.invokeFactory(id=FORM_ID, type_name="FormFolder",
                           title="Bounty Submission Form")
    form = getattr(container, FORM_ID)
    # cleanup form and import data from the archive
    form.manage_delObjects(ids=form.objectIds())
    pfg_data = context.readDataFile("submissions-form.tar.gz")
    ctx = TarballImportContext(form, pfg_data)
    IFilesystemImporter(form).import_(ctx, 'structure', True)
    # Fix importing PFG via GS bug
    #   - it adds extra indentation, wchich breaks the script.
    create_bounty_script = form["create-bounty-submission"]
    create_bounty_script.setScriptBody(CREATE_SCRIPT_BODY)
    # Update and pubhish the form
    form.update(**{"actionAdapter":["create-bounty-submission",],})
    form.unmarkCreationFlag()
    form.reindexObject()
    if wftool.getInfoFor(form, 'review_state') != 'published':
        wftool.doActionFor(form, 'publish')
    logger.info("Bounty submission form successfully created") 

def createStructure(context, logger):
    site = context.getSite()
    wftool = getToolByName(site, "portal_workflow")

    # CONTAINER
    if  CONTAINER_ID not in site.objectIds():
        _createObjectByType('Folder', site, id=CONTAINER_ID,
                            title=CONTAINER_TITLE)
        folder = getattr(site, CONTAINER_ID)
        folder.setOrdering('unordered')
        folder.setConstrainTypesMode(constraintypes.ENABLED)
        #folder.setLocallyAllowedTypes(["Bounty Program Submission"])
        #folder.setImmediatelyAddableTypes(["Bounty Program Submission"])
        folder.setDefaultPage(TOPIC_ID)
        folder.unmarkCreationFlag()

        if wftool.getInfoFor(folder, 'review_state') != 'published':
            wftool.doActionFor(folder, 'publish')

        logger.info("Submissions container added")

    createTopic(folder, wftool, logger)
    createPFGForm(context, folder, wftool, logger)

def importVarious(context):
    """ Various import steps
    """
    if context.readDataFile('ploneorg_kudobounty.txt') is None:
        return
    logger = context.getLogger("ploneorg.kudobounty")
    createStructure(context, logger)