# 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 models, fields


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)