Added upgrade step to added approved field

Author: chris-adam

src/collective/iconifiedcategory/profiles/default/metadata.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0"?>
 <metadata>
-  <version>2106</version>
+  <version>2107</version>
   <dependencies>
     <dependency>profile-collective.documentviewer:default</dependency>
     <dependency>profile-collective.fontawesome:default</dependency>

src/collective/iconifiedcategory/upgrades.py

@@ -212,3 +212,31 @@ def upgrade_to_2105(context):
 def upgrade_to_2106(context):
     load_type_from_package('ContentCategoryConfiguration', 'profile-collective.iconifiedcategory:default')
     remove_gs_step('collective.iconifiedcategory:default-postInstall')
+
+
+def upgrade_to_2107(context):
+    ''' '''
+    portal_types = _portal_types_using_behavior()
+    catalog = api.portal.get_tool('portal_catalog')
+    brains = catalog(portal_type=portal_types)
+
+    logger.info('Querying elements to update among "{0}" objects of portal_type "{1}"'.format(
+        len(brains), ', '.join(portal_types)))
+    parents_to_update = []
+    for brain in brains:
+        obj = brain.getObject()
+        if not(base_hasattr(obj, 'approved')):
+            setattr(obj, 'approved', False)
+
+        parent = obj.aq_parent
+        if parent not in parents_to_update:
+            parents_to_update.append(parent)
+
+    # finally update parents that contains categorized elements
+    nb_of_parents_to_update = len(parents_to_update)
+    i = 1
+    for parent_to_update in parents_to_update:
+        logger.info('Running update_all_categorized_elements for element {0}/{1} ({2})'.format(
+            i, nb_of_parents_to_update, '/'.join(parent_to_update.getPhysicalPath())))
+        i = i + 1
+        update_all_categorized_elements(parent_to_update)

src/collective/iconifiedcategory/upgrades.zcml

@@ -61,4 +61,12 @@
         handler="collective.iconifiedcategory.upgrades.upgrade_to_2106"
         profile="collective.iconifiedcategory:default" />
 
+    <genericsetup:upgradeStep
+        title="Updated ContentCategoryConfiguration type"
+        description=""
+        source="2106"
+        destination="2107"
+        handler="collective.iconifiedcategory.upgrades.upgrade_to_2107"
+        profile="collective.iconifiedcategory:default" />
+
 </configure>