45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
# -*- 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)
|