From ebed54a6f04921b8d047f8f0008691b42f02989d Mon Sep 17 00:00:00 2001
From: Migodoo <jordan@le-filament.com>
Date: Thu, 5 May 2022 11:10:51 +0200
Subject: [PATCH] [fix] migration script

---
 migrations/13.0.1.0.0/post-migration.py | 39 +++++++++++++++++++++++++
 migrations/13.0.1.0.0/pre-migration.py  | 33 ++++++---------------
 2 files changed, 48 insertions(+), 24 deletions(-)
 create mode 100644 migrations/13.0.1.0.0/post-migration.py

diff --git a/migrations/13.0.1.0.0/post-migration.py b/migrations/13.0.1.0.0/post-migration.py
new file mode 100644
index 0000000..d07c700
--- /dev/null
+++ b/migrations/13.0.1.0.0/post-migration.py
@@ -0,0 +1,39 @@
+# © 2022 Le Filament (<http://www.le-filament.com>)
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+from openupgradelib import openupgrade  # pylint: disable=W7936
+from psycopg2 import sql
+
+
+def migrate_type_lm(env):
+
+    type_id = openupgrade.get_legacy_name("type_id")
+
+    env.cr.execute(
+        sql.SQL(
+            """
+                UPDATE scop_liste_ministere lm
+                SET type_lm = 'subscribe'
+                WHERE %(type_id)s = 1;
+            """ % {
+                    "type_id": type_id,
+                },
+        ),
+    )
+
+    env.cr.execute(
+        sql.SQL(
+            """
+                UPDATE scop_liste_ministere lm
+                SET type_lm = 'renew'
+                WHERE %(type_id)s = 2;
+            """ % {
+                    "type_id": type_id,
+                },
+        ),
+    )
+
+
+@openupgrade.migrate()
+def migrate(env, version):
+    migrate_type_lm(env)
diff --git a/migrations/13.0.1.0.0/pre-migration.py b/migrations/13.0.1.0.0/pre-migration.py
index 0cb342f..4177f84 100644
--- a/migrations/13.0.1.0.0/pre-migration.py
+++ b/migrations/13.0.1.0.0/pre-migration.py
@@ -14,7 +14,8 @@ _table_renames = [
     ("liste_ministere_prestataires", "scop_liste_ministere_prestataires"),
     ("liste_ministere_domaine_scic", "scop_liste_ministere_domaine_scic"),
     ("liste_ministere_politique_action", "scop_liste_ministere_politique_action"),
-    ("liste_ministere_cae", "scop_liste_ministere_cae"),
+    ("liste_ministere_inscription_wizard", "scop_liste_ministere_inscription_wizard"),
+    ("liste_ministere_renouvellement_wizard", "scop_liste_ministere_renouvellement_wizard"),
 ]
 _model_renames = [
     ("liste.ministere", "scop.liste.ministere"),
@@ -26,35 +27,19 @@ _model_renames = [
     ("liste.ministere.prestataires", "scop.liste.ministere.prestataires"),
     ("liste.ministere.domaine.scic", "scop.liste.ministere.domaine.scic"),
     ("liste.ministere.politique.action", "scop.liste.ministere.politique.action"),
-    ("liste.ministere.cae", "scop.liste.ministere.cae"),
+    ("liste.ministere.inscription.wizard", "scop.liste.ministere.inscription.wizard"),
+    ("liste.ministere.renouvellement.wizard", "scop.liste.ministere.renouvellement.wizard"),
 ]
 
 
-def migrate_type_lm(env):
-    openupgrade.logged_query(
-        env.cr,
-        sql.SQL(
-            """
-                UPDATE scop_liste_ministere lm
-                SET type_lm = "subscribe"
-                WHERE type_id = 1;
-            """,
-        ).format(sql.Identifier(openupgrade.get_legacy_name("mass_editing_id"))),
-    )
-    openupgrade.logged_query(
-        env.cr,
-        sql.SQL(
-            """
-                UPDATE scop_liste_ministere lm
-                SET type_lm = "renew"
-                WHERE type_id = 2;
-            """,
-        ).format(sql.Identifier(openupgrade.get_legacy_name("mass_editing_id"))),
-    )
+column_renames = {
+    "scop_liste_ministere": [("type_id", None)],
+}
 
 
 @openupgrade.migrate()
 def migrate(env, version):
     openupgrade.rename_models(env.cr, _model_renames)
     openupgrade.rename_tables(env.cr, _table_renames)
-    migrate_type_lm(env)
+    openupgrade.rename_columns(env.cr, column_renames)
+    # migrate_type_lm(env)
-- 
GitLab