mail-addons/mail_multi_website/models/mail_message.py

# Copyright 2018 Ivan Yelizariev <https://it-projects.info/team/yelizariev>
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).
from odoo import fields, models


class Message(models.Model):
    _inherit = "mail.message"

    def _default_mail_server_id(self):
        website = self.env.context.get("website_id")
        if not website:
            return
        website = self.env["website"].sudo().browse(website)
        return website.mail_server_id.id

    mail_server_id = fields.Many2one(default=_default_mail_server_id)