From a53cd50accd89de3128073ee214fa2cc01c322dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Messiant?= <cedricmessiant@ecreall.com>
Date: Thu, 30 May 2013 11:49:04 +0200
Subject: [PATCH] Add upgrade step to migrate reception dates to datetimes

---
 src/collective/dms/mailcontent/configure.zcml    |  1 +
 .../dms/mailcontent/upgrades/__init__.py         |  0
 .../dms/mailcontent/upgrades/configure.zcml      | 16 ++++++++++++++++
 .../dms/mailcontent/upgrades/upgrades.py         | 15 +++++++++++++++
 4 files changed, 32 insertions(+)
 create mode 100644 src/collective/dms/mailcontent/upgrades/__init__.py
 create mode 100644 src/collective/dms/mailcontent/upgrades/configure.zcml
 create mode 100644 src/collective/dms/mailcontent/upgrades/upgrades.py

diff --git a/src/collective/dms/mailcontent/configure.zcml b/src/collective/dms/mailcontent/configure.zcml
index 1961690..42391c2 100644
--- a/src/collective/dms/mailcontent/configure.zcml
+++ b/src/collective/dms/mailcontent/configure.zcml
@@ -9,6 +9,7 @@
   <include file="permissions.zcml" />
 
   <include package="five.grok" />
+  <include package=".upgrades" />
 
   <grok:grok package="." />
 
diff --git a/src/collective/dms/mailcontent/upgrades/__init__.py b/src/collective/dms/mailcontent/upgrades/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/collective/dms/mailcontent/upgrades/configure.zcml b/src/collective/dms/mailcontent/upgrades/configure.zcml
new file mode 100644
index 0000000..f6965d3
--- /dev/null
+++ b/src/collective/dms/mailcontent/upgrades/configure.zcml
@@ -0,0 +1,16 @@
+<configure
+    xmlns="http://namespaces.zope.org/zope"
+    xmlns:genericsetup="http://namespaces.zope.org/genericsetup"
+    i18n_domain="collective.dms.mailcontent">
+
+  <include package="ecreall.helpers.upgrade" />
+
+  <genericsetup:upgradeStep
+      title="Migration profile for collective.dms.mailcontent to 2"
+      description="Upgrade from 1 to 2"
+      source="1"
+      destination="2"
+      handler=".upgrades.v2"
+      profile="collective.dms.mailcontent:default" />
+
+</configure>
\ No newline at end of file
diff --git a/src/collective/dms/mailcontent/upgrades/upgrades.py b/src/collective/dms/mailcontent/upgrades/upgrades.py
new file mode 100644
index 0000000..aaa655e
--- /dev/null
+++ b/src/collective/dms/mailcontent/upgrades/upgrades.py
@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+from datetime import date, datetime, time
+
+from plone import api
+
+
+def v2(context):
+    default_time = time(18, 0)
+    catalog = api.portal.get_tool('portal_catalog')
+    brains = catalog.searchResults({'portal_type': 'dmsincomingmail'})
+    for brain in brains:
+        obj = brain.getObject()
+        reception_date = obj.reception_date
+        if isinstance(reception_date, date):
+            obj.reception_date = datetime.combine(reception_date, default_time)