Xavier Morel
555386706f
When modules get uninstalled, first the uninstall process will drop all the fields (removing all the columns) then it drops all the models (removing the tables). When uninstalling mail, this means the various (res_)model(_id) fields don't exist anymore by the time we're deleting models, so the queries blow up. Skip this step if we're unlinking the mail models, it means the tables have already been dropped, so there's nothing to delete anymore. This should not use `ondelete` because we *do* want to delete records from those tables when deleting modules which depend on mail, and thus have mail stuff associated with their own models which we're deleting. X-original-commit: e43155f940c1f0ba30378d110fc371012d791e32 Part-of: odoo/odoo#121522
128 lines
5.6 KiB
Python
128 lines
5.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
from odoo import _, api, fields, models
|
|
from odoo.exceptions import UserError
|
|
|
|
|
|
class IrModel(models.Model):
|
|
_inherit = 'ir.model'
|
|
_order = 'is_mail_thread DESC, name ASC'
|
|
|
|
is_mail_thread = fields.Boolean(
|
|
string="Has Mail Thread", default=False,
|
|
)
|
|
is_mail_activity = fields.Boolean(
|
|
string="Has Mail Activity", default=False,
|
|
)
|
|
is_mail_blacklist = fields.Boolean(
|
|
string="Has Mail Blacklist", default=False,
|
|
)
|
|
|
|
def unlink(self):
|
|
""" Delete mail data (followers, messages, activities) associated with
|
|
the models being deleted.
|
|
"""
|
|
if not self:
|
|
return True
|
|
|
|
# Delete followers, messages and attachments for models that will be unlinked.
|
|
mail_models = self.search([
|
|
('model', 'in', ('mail.activity', 'mail.activity.type', 'mail.followers', 'mail.message'))
|
|
], order='id')
|
|
|
|
if not (self & mail_models):
|
|
models = tuple(self.mapped('model'))
|
|
model_ids = tuple(self.ids)
|
|
|
|
query = "DELETE FROM mail_activity WHERE res_model_id IN %s"
|
|
self.env.cr.execute(query, [model_ids])
|
|
|
|
query = "DELETE FROM mail_activity_type WHERE res_model IN %s"
|
|
self.env.cr.execute(query, [models])
|
|
|
|
query = "DELETE FROM mail_followers WHERE res_model IN %s"
|
|
self.env.cr.execute(query, [models])
|
|
|
|
query = "DELETE FROM mail_message WHERE model in %s"
|
|
self.env.cr.execute(query, [models])
|
|
|
|
# Get files attached solely to the models being deleted (and none other)
|
|
models = tuple(self.mapped('model'))
|
|
query = """
|
|
SELECT DISTINCT store_fname
|
|
FROM ir_attachment
|
|
WHERE res_model IN %s
|
|
EXCEPT
|
|
SELECT store_fname
|
|
FROM ir_attachment
|
|
WHERE res_model not IN %s;
|
|
"""
|
|
self.env.cr.execute(query, [models, models])
|
|
fnames = self.env.cr.fetchall()
|
|
|
|
query = """DELETE FROM ir_attachment WHERE res_model in %s"""
|
|
self.env.cr.execute(query, [models])
|
|
|
|
for (fname,) in fnames:
|
|
self.env['ir.attachment']._file_delete(fname)
|
|
|
|
return super(IrModel, self).unlink()
|
|
|
|
def write(self, vals):
|
|
if self and ('is_mail_thread' in vals or 'is_mail_activity' in vals or 'is_mail_blacklist' in vals):
|
|
if any(rec.state != 'manual' for rec in self):
|
|
raise UserError(_('Only custom models can be modified.'))
|
|
if 'is_mail_thread' in vals and any(rec.is_mail_thread > vals['is_mail_thread'] for rec in self):
|
|
raise UserError(_('Field "Mail Thread" cannot be changed to "False".'))
|
|
if 'is_mail_activity' in vals and any(rec.is_mail_activity > vals['is_mail_activity'] for rec in self):
|
|
raise UserError(_('Field "Mail Activity" cannot be changed to "False".'))
|
|
if 'is_mail_blacklist' in vals and any(rec.is_mail_blacklist > vals['is_mail_blacklist'] for rec in self):
|
|
raise UserError(_('Field "Mail Blacklist" cannot be changed to "False".'))
|
|
res = super(IrModel, self).write(vals)
|
|
self.env.flush_all()
|
|
# setup models; this reloads custom models in registry
|
|
self.pool.setup_models(self._cr)
|
|
# update database schema of models
|
|
models = self.pool.descendants(self.mapped('model'), '_inherits')
|
|
self.pool.init_models(self._cr, models, dict(self._context, update_custom_fields=True))
|
|
else:
|
|
res = super(IrModel, self).write(vals)
|
|
return res
|
|
|
|
def _reflect_model_params(self, model):
|
|
vals = super(IrModel, self)._reflect_model_params(model)
|
|
vals['is_mail_thread'] = issubclass(type(model), self.pool['mail.thread'])
|
|
vals['is_mail_activity'] = issubclass(type(model), self.pool['mail.activity.mixin'])
|
|
vals['is_mail_blacklist'] = issubclass(type(model), self.pool['mail.thread.blacklist'])
|
|
return vals
|
|
|
|
@api.model
|
|
def _instanciate(self, model_data):
|
|
model_class = super(IrModel, self)._instanciate(model_data)
|
|
if model_data.get('is_mail_blacklist') and model_class._name != 'mail.thread.blacklist':
|
|
parents = model_class._inherit or []
|
|
parents = [parents] if isinstance(parents, str) else parents
|
|
model_class._inherit = parents + ['mail.thread.blacklist']
|
|
if model_class._custom:
|
|
model_class._primary_email = 'x_email'
|
|
elif model_data.get('is_mail_thread') and model_class._name != 'mail.thread':
|
|
parents = model_class._inherit or []
|
|
parents = [parents] if isinstance(parents, str) else parents
|
|
model_class._inherit = parents + ['mail.thread']
|
|
if model_data.get('is_mail_activity') and model_class._name != 'mail.activity.mixin':
|
|
parents = model_class._inherit or []
|
|
parents = [parents] if isinstance(parents, str) else parents
|
|
model_class._inherit = parents + ['mail.activity.mixin']
|
|
return model_class
|
|
|
|
def _get_model_definitions(self, model_names_to_fetch):
|
|
fields_by_model_names = super()._get_model_definitions(model_names_to_fetch)
|
|
for model_name, field_by_fname in fields_by_model_names.items():
|
|
model = self.env[model_name]
|
|
tracked_field_names = model._track_get_fields() if 'mail.thread' in model._inherit else []
|
|
for fname, field in field_by_fname.items():
|
|
if fname in tracked_field_names:
|
|
field['tracking'] = True
|
|
return fields_by_model_names
|