Changeset 890 for Products.ecmigration/trunk/Products/ecmigration/migrate.py

Timestamp:

05/22/12 15:51:08 (12 months ago)

Author:

brent

Message:

Added support for theme properties and images.

File:

• Products.ecmigration/trunk/Products/ecmigration/migrate.py (modified) (7 diffs)

Products.ecmigration/trunk/Products/ecmigration/migrate.py

@@ -177 +177 @@
     def getFile(self, fn):
         fname = os.path.join(self.base, fn)
-        f = open(fname, 'rb')
+        try:
+            f = open(fname, 'rb')
+        except IOError:
+            return None
         data = f.read()
         f.close()
@@ -232 +235 @@
     def readFile(self, info):
         f = self.tar_archive.extractfile(info)
-        data = f.read()
-        f.close()
-        # Circumvent caching for large file handling
-        self.tar_archive.members = []
-        return data
+        if f:
+            data = f.read()
+            f.close()
+            # Circumvent caching for large file handling
+            self.tar_archive.members = []
+            return data
+        return None
 
     def getFile(self, info):
@@ -331 +336 @@
                 data = self._getMetadata(x, archive)
                 self.importSettings(data)
+            elif 'ecmigration_theme' == ext:
+                self.logger.info('Getting eduCommons theme settings: %s' %x)
+                data = self._getMetadata(x, archive)
+                self.importThemeSettings(data)
+                self.importThemeFiles(archive)
             elif 'ecmigration_userinfo' == ext:
                 self.logger.info('Getting eduCommons user info: %s' %x)
@@ -345 +355 @@
                     obj = self.importObject(data)
             else:
-                #f = archvie.getFileOpen(x)
-                #data = archive.getFile(x)
                 self.importObjectData(x, archive)
                 self.imp += 1
@@ -362 +370 @@
         transaction.savepoint(optimistic=True)
 
+    def importThemeSettings(self, data):
+        """ Import theme data """
+        portal = self.context.portal_url.getPortalObject()
+        props = portal.portal_skins.custom
+        bp = props.base_properties
+        bp.manage_changeProperties(
+            logoName=data['logoName'],
+            faviconName=data['faviconName'],
+            bodyBackgroundColor=data['bodyBackgroundColor'],
+            bodyBackgroundImage=data['bodyBackgroundImage'],
+            portalHeaderBackgroundImage=data['portalHeaderBackgroundImage'],
+            portalHeaderBackgroundColor=data['portalHeaderBackgroundColor'],
+            portalHeaderFontColor=data['portalHeaderFontColor'],
+            portalHeaderLinkColor=data['portalHeaderLinkColor'],
+            portalHeaderActiveColor=data['portalHeaderActiveColor'],
+            portalTopNavBackgroundImage=data['portalTopNavBackgroundImage'],
+            portalTopNavBackgroundColor=data['portalTopNavBackgroundColor'],
+            portalTopNavFontColor=data['portalTopNavFontColor'],
+            portalTopNavLinkColor=data['portalTopNavLinkColor'],
+            portalTopNavActiveColor=data['portalTopNavActiveColor'],
+            portalColumnOneBackgroundImage=data['portalColumnOneBackgroundImage'],
+            portalColumnOneBackgroundColor=data['portalColumnOneBackgroundColor'],
+            portalColumnOneFontColor=data['portalColumnOneFontColor'],
+            portalColumnOneLinkColor=data['portalColumnOneLinkColor'],
+            portalColumnOneActiveColor=data['portalColumnOneActiveColor'],
+            backgroundColor=data['backgroundColor'],
+            fontColor=data['fontColor'],
+            linkColor=data['linkColor'],
+            activeColor=data['activeColor'],
+            )
+        transaction.savepoint(optimistic=True)
+
+    def importThemeFiles(self, archive):
+        portal = self.context.portal_url.getPortalObject()
+        props = portal.portal_skins.custom
+        bp = props.base_properties
+        images = [
+            bp.getProperty('logoName'),
+            bp.getProperty('faviconName'),
+            bp.getProperty('bodyBackgroundImage'), 
+            bp.getProperty('portalHeaderBackgroundImage'),
+            bp.getProperty('portalTopNavBackgroundImage'),
+            bp.getProperty('portalColumnOneBackgroundImage'),
+            ]
+        path = '%s/portal_skins/custom/' %self.context.getId()
+        for x in images:
+            data = archive.getFile(path+x)
+            if data:
+                if x in props.objectIds():
+                    props.manage_delObjects(ids=[x])
+                props.manage_addImage(x, data)
+                self.logger.info('Added Theme File: %s' %path+x)
+        transaction.savepoint(optimistic=True)
+                
     def importUsers(self, data):
         """ Import Users """
@@ -540 +602 @@
         self.logger.info('Starting Export')
         self.exportSettings(archive)
+        self.exportThemeSettings(archive)
         self.exportUsers(archive)
         self.exportObjects(brains, archive, 99999)
@@ -561 +624 @@
         archive.addFile(fn, data)
         self.logger.info('Added settings file: %s' %fn)
+
+    def exportThemeSettings(self, archive):
+        tprops = self.context.portal_skins.custom
+        bp = tprops.base_properties
+        fn = '%s/theme.ecmigration_theme' %self.context.getId()
+        objstore = {
+            'logoName':bp.getProperty('logoName'),
+            'faviconName':bp.getProperty('faviconName'),
+            'bodyBackgroundColor':bp.getProperty('bodyBackgroundColor'),
+            'bodyBackgroundImage':bp.getProperty('bodyBackgroundImage'),
+            'portalHeaderBackgroundImage':bp.getProperty('portalHeaderBackgroundImage'),
+            'portalHeaderBackgroundColor':bp.getProperty('portalHeaderBackgroundColor'),
+            'portalHeaderFontColor':bp.getProperty('portalHeaderFontColor'),
+            'portalHeaderLinkColor':bp.getProperty('portalHeaderLinkColor'),
+            'portalHeaderActiveColor':bp.getProperty('portalHeaderActiveColor'),
+            'portalTopNavBackgroundImage':bp.getProperty('portalTopNavBackgroundImage'),
+            'portalTopNavBackgroundColor':bp.getProperty('portalTopNavBackgroundColor'),
+            'portalTopNavFontColor':bp.getProperty('portalTopNavFontColor'),
+            'portalTopNavLinkColor':bp.getProperty('portalTopNavLinkColor'),
+            'portalTopNavActiveColor':bp.getProperty('portalTopNavActiveColor'),
+            'portalColumnOneBackgroundImage':bp.getProperty('portalColumnOneBackgroundImage'),
+            'portalColumnOneBackgroundColor':bp.getProperty('portalColumnOneBackgroundColor'),
+            'portalColumnOneFontColor':bp.getProperty('portalColumnOneFontColor'),
+            'portalColumnOneLinkColor':bp.getProperty('portalColumnOneLinkColor'),
+            'portalColumnOneActiveColor':bp.getProperty('portalColumnOneActiveColor'),
+            'backgroundColor':bp.getProperty('backgroundColor'),
+            'fontColor':bp.getProperty('fontColor'),
+            'linkColor':bp.getProperty('linkColor'),
+            'activeColor':bp.getProperty('activeColor'),
+            }
+        data = Pickle.dumps(objstore)
+        self.files.append(fn)
+        archive.addFile(fn, data)
+        self.logger.info('Added theme settings: %s' %fn)
+        images = [
+            objstore['logoName'],
+            objstore['faviconName'],
+            objstore['bodyBackgroundImage'], 
+            objstore['portalHeaderBackgroundImage'],
+            objstore['portalTopNavBackgroundImage'],
+            objstore['portalColumnOneBackgroundImage'],
+            ]
+        path = '%s/portal_skins/custom/' %self.context.getId()
+        for x in images:
+            if x in tprops.objectIds():
+                img = getattr(tprops, x, None)
+                if img:
+                    archive.addFile(path + x, img.data)
+                    self.logger.info('Added theme image file: %s' %path+x)
 
     def exportUsers(self, archive):