合并企业版代码(未测试，先提交到测试分支)

web_studio/models/ir_model_data.py

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+
+class IrModelData(models.Model):
+    _inherit = 'ir.model.data'
+
+    studio = fields.Boolean(help='Checked if it has been edited with Studio.')
+
+    @api.model_create_multi
+    def create(self, vals_list):
+        if self._context.get('studio'):
+            for vals in vals_list:
+                vals['studio'] = True
+        return super().create(vals_list)
+
+    def write(self, vals):
+        """ When editing an ir.model.data with Studio, we put it in noupdate to
+                avoid the customizations to be dropped when upgrading the module.
+        """
+        if self._context.get('studio'):
+            vals['noupdate'] = True
+            vals['studio'] = True
+        return super(IrModelData, self).write(vals)
+
+    def _build_update_xmlids_query(self, sub_rows, update):
+        '''Override of the base method to include the `studio` attribute for studio module imports.'''
+        if self._context.get('studio'):
+            rowf = "(%s, %s, %s, %s, %s, 't')"
+            return """
+                INSERT INTO ir_model_data (module, name, model, res_id, noupdate, studio)
+                VALUES {rows}
+                ON CONFLICT (module, name)
+                DO UPDATE SET (model, res_id, write_date, noupdate) =
+                    (EXCLUDED.model, EXCLUDED.res_id, now() at time zone 'UTC', 't')
+                    {where}
+            """.format(
+                rows=", ".join([rowf] * len(sub_rows)),
+                where="WHERE NOT ir_model_data.noupdate" if update else "",
+            )
+        else:
+            return super(IrModelData, self)._build_update_xmlids_query(sub_rows, update)