[16.1] Add lock date migration and settings UI

odoo-bringout-oca-ocb-account/account/migrations/16.1.0.0/post-migration.py

@@ -0,0 +1,63 @@
+from odoo import api, SUPERUSER_ID
+
+
+def _table_has_column(cr, table, column):
+    cr.execute(
+        """
+        SELECT 1
+          FROM information_schema.columns
+         WHERE table_name = %s AND column_name = %s
+        """,
+        (table, column),
+    )
+    return bool(cr.fetchone())
+
+
+def migrate(cr, version):
+    if not version:
+        return
+
+    env = api.Environment(cr, SUPERUSER_ID, {})
+
+    # Initialise new company lock fields using existing lock dates.
+    companies = env['res.company'].with_context(active_test=False).search([])
+    for company in companies:
+        vals = {}
+        if not company.sale_lock_date and company.period_lock_date:
+            vals['sale_lock_date'] = company.period_lock_date
+        if not company.purchase_lock_date and company.period_lock_date:
+            vals['purchase_lock_date'] = company.period_lock_date
+        if not company.hard_lock_date and company.fiscalyear_lock_date:
+            vals['hard_lock_date'] = company.fiscalyear_lock_date
+        if vals:
+            company.sudo().write(vals)
+
+    # If account_move_export is installed, migrate configuration defaults.
+    if env.registry.get('account.move.export.config'):
+        # Ensure new boolean fields are not left NULL and join_char has a value.
+        cr.execute(
+            """
+            UPDATE account_move_export_config
+               SET group_lines = COALESCE(group_lines, FALSE),
+                   lock_tax = COALESCE(lock_tax, FALSE),
+                   lock_sale = COALESCE(lock_sale, FALSE),
+                   lock_purchase = COALESCE(lock_purchase, FALSE),
+                   lock_fiscalyear = COALESCE(lock_fiscalyear, FALSE),
+                   lock_hard = COALESCE(lock_hard, FALSE),
+                   join_char = COALESCE(NULLIF(join_char, ''), '-')
+            """
+        )
+
+        if _table_has_column(cr, 'account_move_export_config', 'lock'):
+            cr.execute(
+                """
+                UPDATE account_move_export_config
+                   SET lock_tax = lock_tax OR lock IN ('tax', 'period', 'fiscalyear'),
+                       lock_sale = lock_sale OR lock IN ('period', 'fiscalyear'),
+                       lock_purchase = lock_purchase OR lock IN ('period', 'fiscalyear'),
+                       lock_fiscalyear = lock_fiscalyear OR lock = 'fiscalyear',
+                       lock_hard = lock_hard OR lock = 'fiscalyear'
+                """
+            )
+            # Optionally drop the legacy column to avoid confusion.
+            cr.execute("ALTER TABLE account_move_export_config DROP COLUMN lock")