source: collective.imstransport/trunk/collective/imstransport/utilities/imsutility.py @ 91

from zope.interface import implements
from zope.component import getUtility, getUtilitiesFor
from collective.imstransport.utilities.interfaces import IIMSTransportUtility, IIMSObjectCreator, IIMSManifestReader, IIMSManifestWriter
from Products.PageTemplates.PageTemplateFile import PageTemplateFile
#from collective.imstransport.config import WWW_DIR
from OFS.SimpleItem import SimpleItem
from zipfile import ZipFile, ZIP_DEFLATED
from StringIO import StringIO


class IMSTransportUtility(SimpleItem):

    implements(IIMSTransportUtility)
#    tocpage = PageTemplateFile('tableofcontents',WWW_DIR)

    def importPackage(self, context, package, packagetype):
        reader = getUtility(IIMSManifestReader, packagetype)
        manifestDictionary = reader.readManifest(package)
        oc = getUtility(IIMSObjectCreator)
        oc.createObject(context, manifestDictionary, package)
        
    def exportPackage(self, package, filename, packagetype = "eduCommons Common Cartridge", wtype='Common Cartridge'):  
        writer = getUtility(IIMSManifestWriter, packagetype)
#        writer.setDestination(ZipfileWriter(filename, package.getId()))
        return writer.createPackage(package, filename)
        
    def getWriters(self):
        writers = getUtilitiesFor(IIMSManifestWriter)
        values = []
        for writer in writers:
            values.append((writer[1].getPackageName(), writer[0]))
        return values

    def getReaders(self):
        utils = getUtilitiesFor(IIMSManifestReader)
        values = []
        for x in utils:
            values.append((x[1].getPackageName(), x[0]))
        return values


class ZipfileWriter:
    """ Write a zip file which contains all the IMS Content packaging stuff. """

    def __init__(self, archive_name, package_name):
        self.fullpath = '%s/' %package_name
        self.archive_name = archive_name
        self.package_name = package_name
        self.archive = StringIO()
        self.zipfile = ZipFile(self.archive, 'w', ZIP_DEFLATED)


    def writeFile(self, path, data):
        """ Write a file to the zip archive. """
        fpath = '%s%s' %(self.fullpath, path)
        self.zipfile.writestr(fpath, data)

    def getOutput(self):
        """ Close the zip file and get the binary archive. """
        self.zipfile.close()
        self.archive.seek(0)
        return self.archive.read(), self.archive_name