migration: add collector_is_creator flag migrator

git-svn-id: http://svn.parisson.org/svn/crem@92 3bf09e05-f825-4182-b9bc-eedd7160adf0

diff --git a/trunk/import/migration/tasks/collections.py b/trunk/import/migration/tasks/collections.py
index 2d3978203aa886d5efeb19f1456af8c51494d6d2..999ccb5c08fe6ca9fb3fa44d802a55c1d155ce8a 100644 (file)
--- a/trunk/import/migration/tasks/collections.py
+++ b/trunk/import/migration/tasks/collections.py
@@ -453,3 +453,17 @@ class CollectionsYearConverter(DataMigrator):
             self.target_cursor.execute("UPDATE media_collections SET recorded_from_year = %s, "
                                        "recorded_to_year = %s WHERE old_code = %s", (from_year, to_year, old_code))
 
+
+class CollectionsCollectorNormalizer(DataMigrator):
+    """Migrate collector_is_creator flag"""
+
+    implements(IDataMigrator)
+
+    def get_name(self):
+        return "collections:collector"
+
+    def process(self):
+        self.target_cursor.execute("UPDATE media_collections SET collector_is_creator = 0")
+        self.target_cursor.execute("UPDATE media_collections SET collector_is_creator = (collector IN ('=', '=+N', 'ID' 'ID.' 'IDE.' 'IDEM'))")
+        self.stats['flagged'] = self.target_cursor.rowcount
+