[FIX] l10n_ar_withholding: Fix migration scripts

Author: nicomacr

l10n_ar_withholding/17.0.0.0/post-migration.py

@@ -3,17 +3,16 @@
 _logger = logging.getLogger(__name__)
 
 
-def migrate_taxes_config(env):
+def migrate_taxes_config(cr):
     query = """update account_tax set
         type_tax_use = 'none',
         l10n_ar_withholding_payment_type = type_tax_use
     where
         type_tax_use in ('customer', 'supplier')
     """
-    openupgrade.logged_query(env.cr, query)
+    openupgrade.logged_query(cr, query)
 
 
-@openupgrade.migrate()
-def migrate(env, version):
+def migrate(cr, version):
     _logger.debug('Running migrate script for l10n_ar_withholding')
-    migrate_taxes_config(env)
+    migrate_taxes_config(cr)

l10n_ar_withholding/17.0.0.0/pre-migration.py

@@ -1,4 +1,5 @@
 from openupgradelib import openupgrade
+from odoo import SUPERUSER_ID, api
 
 
 _column_copy = {
@@ -16,8 +17,8 @@
 ]
 
 
-@openupgrade.migrate()
-def migrate(env, version):
+def migrate(cr, version):
     # backup de columnas que nos interesan antes de que se borren
-    openupgrade.copy_columns(env.cr, _column_copy)
+    env = api.Environment(cr, SUPERUSER_ID, {})
+    openupgrade.copy_columns(cr, _column_copy)
     openupgrade.rename_fields(env, _field_renames)