From 96bcb7f885255ed8f8171c403af5ac4bf5dc31b6 Mon Sep 17 00:00:00 2001
From: Alin Voinea <alin@eaudeweb.ro>
Date: Thu, 23 Apr 2015 17:37:38 +0300
Subject: [PATCH] Refs #24901 - Add upgrade step to remove duplicate fallback
 images

---
 docs/HISTORY.txt                          |  2 ++
 eea/exhibit/profiles/default/metadata.xml |  2 +-
 eea/exhibit/upgrades/configure.zcml       | 12 +++++++
 eea/exhibit/upgrades/evolve80.py          | 43 +++++++++++++++++++++++
 setup.py                                  |  2 +-
 5 files changed, 59 insertions(+), 2 deletions(-)
 create mode 100644 eea/exhibit/upgrades/evolve80.py

diff --git a/docs/HISTORY.txt b/docs/HISTORY.txt
index 9ddcc67..a77cce4 100644
--- a/docs/HISTORY.txt
+++ b/docs/HISTORY.txt
@@ -3,6 +3,8 @@ Changelog
 
 8.0.dev0 - (unreleased)
 -----------------------
+* Upgrade step: Within "Plone > Site setup > Add-ons" click on
+  upgrade button available for eea.exhibit
 * Cleanup: Removed fallback images from ZODB
   [voineali refs #24901]
 
diff --git a/eea/exhibit/profiles/default/metadata.xml b/eea/exhibit/profiles/default/metadata.xml
index 8fd1ece..64cfef5 100644
--- a/eea/exhibit/profiles/default/metadata.xml
+++ b/eea/exhibit/profiles/default/metadata.xml
@@ -1,7 +1,7 @@
 <?xml version="1.0"?>
 <metadata>
  <description>EEA Exhibit profile</description>
- <version>6.2</version>
+ <version>8.0</version>
  <dependencies>
   <dependency>profile-eea.app.visualization:default</dependency>
  </dependencies>
diff --git a/eea/exhibit/upgrades/configure.zcml b/eea/exhibit/upgrades/configure.zcml
index 0ccfb55..179cf6d 100644
--- a/eea/exhibit/upgrades/configure.zcml
+++ b/eea/exhibit/upgrades/configure.zcml
@@ -3,6 +3,18 @@
   xmlns:genericsetup="http://namespaces.zope.org/genericsetup"
   i18n_domain="eea">
 
+  <genericsetup:upgradeSteps
+    source="6.2"
+    destination="8.0"
+    profile="eea.exhibit:default">
+
+    <genericsetup:upgradeStep
+      title="Cleanup fallback images from ZODB"
+      handler="eea.exhibit.upgrades.evolve80.cleanup_fallback_images"
+      />
+
+  </genericsetup:upgradeSteps>
+
   <!-- 6.0 => 6.2 -->
   <genericsetup:upgradeSteps
     source="6.0"
diff --git a/eea/exhibit/upgrades/evolve80.py b/eea/exhibit/upgrades/evolve80.py
new file mode 100644
index 0000000..af2a5a0
--- /dev/null
+++ b/eea/exhibit/upgrades/evolve80.py
@@ -0,0 +1,43 @@
+""" Cleanup
+"""
+import logging
+from zope.component.interface import interfaceToName
+from Products.CMFCore.utils import getToolByName
+from eea.app.visualization.interfaces import IVisualizationEnabled
+logger = logging.getLogger('eea.exhibit')
+
+CLEANUP = [
+    'daviz.map.preview.png',
+    'daviz.tabular.preview.png',
+    'daviz.tile.preview.png',
+    'daviz.timeline.preview.png'
+]
+
+def cleanup_fallback_images(context):
+    """ Migrate exhibit image charts"""
+    ctool = getToolByName (context, 'portal_catalog')
+    iface = interfaceToName(context, IVisualizationEnabled)
+    brains = ctool(
+        object_provides=iface,
+        show_inactive=True, Language='all'
+    )
+
+
+    count = 0
+    for brain in brains:
+        doc = brain.getObject()
+        clean = []
+        for item in CLEANUP:
+            if item in doc.objectIds():
+                clean.append(item)
+
+        if not clean:
+            continue
+
+        length = len(clean)
+        logger.info('Removing %s exhibit fallback images from %s',
+                    length, doc.absolute_url())
+        doc.manage_delObjects(clean)
+        count += length
+
+    logger.info('Removed %s exhibit fallback images', count)
diff --git a/setup.py b/setup.py
index 08e9742..e59602d 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@
       install_requires=[
           'setuptools',
           'rdflib',
-          'eea.app.visualization > 7.0',
+          'eea.app.visualization > 9.3',
           # -*- Extra requirements: -*-
       ],
       extras_require={