appypod-rattail/gen/plone25/mixins/PodTemplateMixin.py

# ------------------------------------------------------------------------------
import os, os.path, time
from appy.shared import mimeTypes
from appy.gen.plone25.mixins import AbstractMixin
from StringIO import StringIO

# ------------------------------------------------------------------------------
class PodError(Exception): pass

# ------------------------------------------------------------------------------
def getOsTempFolder():
    tmp = '/tmp'
    if os.path.exists(tmp) and os.path.isdir(tmp):
        res = tmp
    elif os.environ.has_key('TMP'):
        res = os.environ['TMP']
    elif os.environ.has_key('TEMP'):
        res = os.environ['TEMP']
    else:
        raise "Sorry, I can't find a temp folder on your machine."
    return res

# Error-related constants ------------------------------------------------------
POD_ERROR = 'An error occurred while generating the document. Please check ' \
            'the following things if you wanted to generate the document in ' \
            'PDF, DOC or RTF: (1) OpenOffice is started in server mode on ' \
            'the port you should have specified in the PloneMeeting ' \
            'configuration (go to Site setup-> PloneMeeting configuration); ' \
            '(2) if the Python interpreter running Zope and ' \
            'Plone is not able to discuss with OpenOffice (it does not have ' \
            '"uno" installed - check it by typing "import uno" at the Python ' \
            'prompt) please specify, in the PloneMeeting configuration, ' \
            'the path to a UNO-enabled Python interpreter (ie, the Python ' \
            'interpreter included in the OpenOffice distribution, or, if ' \
            'your server runs Ubuntu, the standard Python interpreter ' \
            'installed in /usr/bin/python). Here is the error as reported ' \
            'by the appy.pod library:\n\n %s'
DELETE_TEMP_DOC_ERROR = 'A temporary document could not be removed. %s.'

# ------------------------------------------------------------------------------
class PodTemplateMixin(AbstractMixin):
    _appy_meta_type = 'podtemplate'
    def generateDocument(self, obj):
        '''Generates a document from this template, for object p_obj.'''
        appySelf = self._appy_getWrapper(force=True)
        appName = self.getProductConfig().PROJECTNAME
        appModule = getattr(self.getProductConfig(), appName)
        # Temporary file where to generate the result
        tempFileName = '%s/%s_%f.%s' % (
            getOsTempFolder(), obj.UID(), time.time(), self.getPodFormat())
        # Define parameters to pass to the appy.pod renderer
        currentUser = self.portal_membership.getAuthenticatedMember()
        podContext = {'self': obj._appy_getWrapper(force=True),
                      'user': currentUser,
                      'podTemplate': appySelf,
                      'now': self.getProductConfig().DateTime(),
                      'projectFolder': os.path.dirname(appModule.__file__)
                      }
        rendererParams = {'template': StringIO(appySelf.podTemplate.content),
                          'context': podContext,
                          'result': tempFileName }
        if appySelf.tool.unoEnabledPython:
            rendererParams['pythonWithUnoPath'] = appySelf.tool.unoEnabledPython
        if appySelf.tool.openOfficePort:
            rendererParams['ooPort'] = appySelf.tool.openOfficePort
        # Launch the renderer
        import appy.pod
        try:
            renderer = appy.pod.renderer.Renderer(**rendererParams)
            renderer.run()
        except appy.pod.PodError, pe:
            if not os.path.exists(tempFileName):
                # In some (most?) cases, when OO returns an error, the result is
                # nevertheless generated.
                raise PodError(POD_ERROR % str(pe))
        # Open the temp file on the filesystem
        f = file(tempFileName, 'rb')
        forBrowser = True
        if forBrowser:
            # Create a OFS.Image.File object that will manage correclty HTTP
            # headers, etc.
            theFile = self.getProductConfig().File('dummyId', 'dummyTitle', f,
                           content_type=mimeTypes[appySelf.podFormat])
            res = theFile.index_html(self.REQUEST, self.REQUEST.RESPONSE)
        else:
            # I must return the raw document content.
            res = f.read()
        f.close()
        # Returns the doc and removes the temp file
        try:
            os.remove(tempFileName)
        except OSError, oe:
            self.getProductConfig().logger.warn(DELETE_TEMP_DOC_ERROR % str(oe))
        except IOError, ie:
            self.getProductConfig().logger.warn(DELETE_TEMP_DOC_ERROR % str(ie))
        return res
# ------------------------------------------------------------------------------