Override some ImportContent method for migration

Author: boulch

base.cfg

@@ -36,6 +36,7 @@ eggs =
     Plone
     Pillow
     imio.smartweb.core
+    collective.exportimport
 zcml =
     imio.smartweb.core
     imio.smartweb.core-overrides
@@ -68,4 +69,4 @@ imio.smartweb.core =
 
 [remotes]
 imio = https://github.com/IMIO
-imio_push = git@github.com:IMIO
+imio_push = git@github.com:IMIO

src/imio/smartweb/core/browser/configure.zcml

@@ -22,6 +22,7 @@
   <include package=".herobanner" />
   <include package=".ia" />
   <include package=".instancebehaviors" />
+  <include package=".migration" />
   <include package=".minisite" />
   <include package=".subsite" />
   <include package=".search" />

src/imio/smartweb/core/browser/migration/__init__.py

@@ -0,0 +1,21 @@
+<configure
+    xmlns="http://namespaces.zope.org/zope"
+    xmlns:browser="http://namespaces.zope.org/browser">
+
+  <browser:page
+    name="import_content"
+    for="zope.interface.Interface"
+    class=".import_content.CustomImportContent"
+    layer="imio.smartweb.core.interfaces.IImioSmartwebCoreLayer"
+    permission="cmf.ManagePortal"
+    />
+
+  <browser:page
+      name="create_contents"
+      for="*"
+      class=".create.CreateContentsForMigration"
+      permission="cmf.AddPortalContent"
+      layer="imio.smartweb.core.interfaces.IImioSmartwebCoreLayer"
+      />
+
+</configure>

src/imio/smartweb/core/browser/migration/configure.zcml

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+
+from imio.smartweb.core.behaviors.subsite import IImioSmartwebSubsite
+from imio.smartweb.core.contents import IFolder
+from imio.smartweb.locales import SmartwebMessageFactory as _
+from plone import api
+from plone.app.layout.navigation.interfaces import INavigationRoot
+from Products.CMFPlone.interfaces.constrains import DISABLED
+from Products.CMFPlone.interfaces.constrains import ISelectableConstrainTypes
+from Products.Five.browser import BrowserView
+from plone.protect.interfaces import IDisableCSRFProtection
+from zope.interface import alsoProvides
+
+
+class CreateContentsForMigration(BrowserView):
+
+    def __call__(self):
+        # Faut lancer la migration depuis /Plone
+        alsoProvides(self.request, IDisableCSRFProtection)
+        self.create_content("actualites", "Actualités")
+        self.create_content("images", "Images")
+
+    def create_content(self, id, title):
+        fr = [obj for obj in self.context.items() if obj[0] == "fr"][0][1]
+        if id not in [i[0] for i in fr.items()]:
+            api.content.create(
+                id=id, title=title, container=fr, type="imio.smartweb.Folder"
+            )

src/imio/smartweb/core/browser/migration/create.py

@@ -0,0 +1,110 @@
+from collective.exportimport.import_content import ImportContent
+from imio.smartweb.core.behaviors.minisite import IImioSmartwebMinisite
+from imio.smartweb.core.behaviors.minisite import IImioSmartwebMinisiteSettings
+from imio.smartweb.core.contents.pages.pages import IDefaultPages
+from plone.dexterity.interfaces import IDexterityFTI
+from six.moves.urllib.parse import unquote
+from six.moves.urllib.parse import urlparse
+from zope.component import getUtility
+from zope.interface import alsoProvides
+from plone import api
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class CustomImportContent(ImportContent):
+
+    def create_container(self, item):
+        """Create container for item.
+
+        See remarks in get_parent_as_container for some corner cases.
+        """
+        # if (
+        #     item["@id"]
+        #     == "http://localhost:8080/Plone/fr/je-suis/commercant/section-gallery/banner.jpg"
+        # ):
+        #     import pdb
+
+        #     pdb.set_trace()
+        folder = self.context
+        parent_url = unquote(item["parent"]["@id"])
+        parent_url_parsed = urlparse(parent_url)
+        # Get the path part, split it, remove the always empty first element.
+        parent_path = parent_url_parsed.path.split("/")[1:]
+        if (
+            len(parent_url_parsed.netloc.split(":")) > 1
+            or parent_url_parsed.netloc == "nohost"
+        ):
+            # For example localhost:8080, or nohost when running tests.
+            # First element will then be a Plone Site id.
+            # Get rid of it.
+            parent_path = parent_path[1:]
+
+        # Handle folderish Documents provided by plone.volto
+        fti = getUtility(IDexterityFTI, name="Document")
+        # BOULCH
+        # if "fr/la-boverie/album/section-gallery/logo-laboverie.jpg" in item["@id"]:
+        #     import pdb
+
+        #     pdb.set_trace()
+        parent_type = (
+            "Document"
+            if fti.klass.endswith("FolderishDocument")
+            else item["parent"]["@type"]
+        )
+        # parent_type = (
+        #    "Document" if fti.klass.endswith("FolderishDocument") else "Folder"
+        # )
+        # create original structure for imported content
+        for element in parent_path:
+            try:
+                if element not in folder:
+                    folder = api.content.create(
+                        container=folder,
+                        type=parent_type,
+                        id=element,
+                        title=element,
+                    )
+                    logger.info(
+                        "Created container %s to hold %s",
+                        folder.absolute_url(),
+                        item["@id"],
+                    )
+                else:
+                    folder = folder[element]
+            except:
+                print(f"ERREUR SUR :{element} -- {folder} -- {folder.absolute_url()}")
+
+        return folder
+
+    def global_obj_hook(self, obj, item):
+        """Inspect the content item before serialization data."""
+
+        if "cpskin.minisite.interfaces.IMinisiteRoot" in item.get(
+            "_cpskin_interfaces", []
+        ):
+            alsoProvides(obj, IImioSmartwebMinisite)
+            alsoProvides(obj, IImioSmartwebMinisiteSettings)
+            logger.info(f"Set minisite to {obj.absolute_url()} ")
+
+        if "imio.smartweb.core.contents.pages.pages.IDefaultPages" in item.get(
+            "_smartweb_interfaces", []
+        ):
+            alsoProvides(obj, IDefaultPages)
+            logger.info(f"Set default page to {obj.absolute_url()} ")
+        layout = item.get("layout")
+        if layout:
+            try:
+                obj.setLayout(layout)
+                logger.info(f"Set layout : {layout} to {obj.absolute_url()} ")
+                # print(f"setLayout {layout} TO {obj.absolute_url()}")
+            except AttributeError:
+                pass
+        obj.reindexObject()
+        return obj
+
+    # !!! by default, this method remove item layout
+    def global_dict_hook(self, item):
+        return item