From 76a12459e94a5f47cd9e98b854ae16569529a299 Mon Sep 17 00:00:00 2001
From: migodoo <jordan>
Date: Thu, 5 May 2022 12:38:18 +0200
Subject: [PATCH] [fix] migration script

---
 migrations/13.0.1.0.0/post-migration.py | 31 +++++++++++++++++++------
 migrations/13.0.1.0.0/pre-migration.py  |  2 +-
 2 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/migrations/13.0.1.0.0/post-migration.py b/migrations/13.0.1.0.0/post-migration.py
index 8b38397..51b0ebb 100644
--- a/migrations/13.0.1.0.0/post-migration.py
+++ b/migrations/13.0.1.0.0/post-migration.py
@@ -2,20 +2,37 @@
 # 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 scop_account_move_aura_map_values(env):
+    env.cr.execute(
+        sql.SQL(
+            """
+                UPDATE account_move am
+                SET
+                    cotisation_aura_id = ai.cotisation_aura_id,
+                    wage_cg_connu = ai.wage_cg_connu,
+                    wage_cg_retenu = ai.wage_cg_retenu,
+                    amount_aura_calculated = ai.amount_aura_calculated,
+                    nb_quarter_aura = cast(ai.nb_quarter_aura as varchar),
+                    amount_aura_prorata = ai.amount_aura_prorata
+                FROM account_invoice ai
+                WHERE ai.id = am.old_invoice_id;
+            """
+        ),
+    )
+
     openupgrade.map_values(
         env.cr,
-        openupgrade.get_legacy_name("nb_quarter_aura"),
-        "nb_quarter_aura",
+        openupgrade.get_legacy_name("year"),
+        "year",
         [
-            (1, "1"),
-            (2, "2"),
-            (3, "3"),
-            (4, "4"),
+            (2020, "2020"),
+            (2021, "2021"),
+            (2022, "2022"),
         ],
-        table="account_move",
+        table="scop_cotisation_aura",
     )
 
 
diff --git a/migrations/13.0.1.0.0/pre-migration.py b/migrations/13.0.1.0.0/pre-migration.py
index 25570fa..b4ea5b8 100644
--- a/migrations/13.0.1.0.0/pre-migration.py
+++ b/migrations/13.0.1.0.0/pre-migration.py
@@ -4,7 +4,7 @@
 from openupgradelib import openupgrade
 
 column_renames = {
-    "account_move": [("nb_quarter_aura", None)],
+    "scop_cotisation_aura": [("year", None)],
 }
 
 
-- 
GitLab