Migration script

golem_member_minor/migrations/10.0.1.1.3/post-migrate.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+
+#    Copyright 2018 Fabien Bourgeois <fabien@yaltik.com>
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as
+#    published by the Free Software Foundation, either version 3 of the
+#    License, or (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+""" Post-migration script """
+
+from openupgradelib import openupgrade
+
+@openupgrade.migrate(use_env=False)
+def migrate(cursor, version):
+    """ Recover old legal_gardian_ids and delete their database table  """
+    cursor.execute('SELECT golem_member_id, res_partner_id FROM golem_member_res_partner_rel')
+    for member_id, legal_guardian_id in cursor.fetchall():
+        cursor.execute('''INSERT INTO golem_legal_guardian(member_id, legal_guardian_id) VALUES
+                           (%s, %s)''' % (member_id, legal_guardian_id))
+    cursor.execute('DROP TABLE golem_member_res_partner_rel')