From 054c100cbdc3f1802da8d38784d1b0b8054fe878 Mon Sep 17 00:00:00 2001
From: "Pedro M. Baeza" <pedro.baeza@tecnativa.com>
Date: Fri, 3 Sep 2021 20:55:37 +0200
Subject: [PATCH] [OU-ADD] rma: Migration scripts

---
 rma/migrations/13.0.1.0.0/post-migration.py | 39 +++++++++++++++++++++
 rma/migrations/13.0.1.0.0/pre-migration.py  | 11 ++++++
 2 files changed, 50 insertions(+)
 create mode 100644 rma/migrations/13.0.1.0.0/post-migration.py
 create mode 100644 rma/migrations/13.0.1.0.0/pre-migration.py

diff --git a/rma/migrations/13.0.1.0.0/post-migration.py b/rma/migrations/13.0.1.0.0/post-migration.py
new file mode 100644
index 00000000..64a62c70
--- /dev/null
+++ b/rma/migrations/13.0.1.0.0/post-migration.py
@@ -0,0 +1,39 @@
+# Copyright 2021 Tecnativa - Pedro M. Baeza
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+from openupgradelib import openupgrade  # pylint: disable=W7936
+from psycopg2 import sql
+
+
+@openupgrade.migrate()
+def migrate(env, version):
+    openupgrade.logged_query(
+        env.cr,
+        """
+        UPDATE account_move_line aml
+        SET rma_id = ail.rma_id
+        FROM account_invoice_line ail
+        WHERE ail.id = aml.old_invoice_line_id""",
+    )
+    openupgrade.logged_query(
+        env.cr,
+        sql.SQL(
+            """
+            UPDATE rma
+            SET refund_id = am.id
+            FROM account_move am
+            WHERE am.old_invoice_id = {}
+            """
+        ).format(sql.Identifier(openupgrade.get_legacy_name("refund_id"))),
+    )
+    openupgrade.logged_query(
+        env.cr,
+        sql.SQL(
+            """
+            UPDATE rma
+            SET refund_line_id = aml.id
+            FROM account_move_line aml
+            WHERE aml.old_invoice_line_id = {}
+            """
+        ).format(sql.Identifier(openupgrade.get_legacy_name("refund_id"))),
+    )
diff --git a/rma/migrations/13.0.1.0.0/pre-migration.py b/rma/migrations/13.0.1.0.0/pre-migration.py
new file mode 100644
index 00000000..66b1fc82
--- /dev/null
+++ b/rma/migrations/13.0.1.0.0/pre-migration.py
@@ -0,0 +1,11 @@
+# Copyright 2021 Tecnativa - Pedro M. Baeza
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+from openupgradelib import openupgrade  # pylint: disable=W7936
+
+
+@openupgrade.migrate()
+def migrate(env, version):
+    openupgrade.rename_columns(
+        env.cr, {"rma": [("refund_id", None), ("refund_line_id", None)]}
+    )