server-tools/fetchmail_attach_from_folder/wizard/attach_mail_manually.py

# -*- coding: utf-8 -*-
# Copyright - 2013-2018 Therp BV <https://therp.nl>.
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
import logging

from odoo import api, fields, models


_logger = logging.getLogger(__name__)


class AttachMailManually(models.TransientModel):
    _name = 'fetchmail.attach.mail.manually'

    folder_id = fields.Many2one(
        'fetchmail.server.folder', 'Folder', readonly=True)
    mail_ids = fields.One2many(
        'fetchmail.attach.mail.manually.mail', 'wizard_id', 'Emails')

    @api.model
    def default_get(self, fields_list):
        defaults = super(AttachMailManually, self).default_get(fields_list)
        defaults['mail_ids'] = []
        folder_model = self.env['fetchmail.server.folder']
        folder_id = self.env.context.get('folder_id')
        defaults['folder_id'] = folder_id
        folder = folder_model.browse([folder_id])
        connection = folder.server_id.connect()
        connection.select(folder.path)
        criteria = 'FLAGGED' if folder.flag_nonmatching else 'UNDELETED'
        msgids = folder.get_msgids(connection, criteria)
        for msgid in msgids[0].split():
            mail_message, message_org = folder.fetch_msg(connection, msgid)
            defaults['mail_ids'].append((0, 0, {
                'msgid': msgid,
                'subject': mail_message.get('subject', ''),
                'date': mail_message.get('date', ''),
                'object_id': '%s,-1' % folder.model_id.model}))
        connection.close()
        return defaults

    @api.multi
    def attach_mails(self):
        self.ensure_one()
        folder = self.folder_id
        server = folder.server_id
        connection = server.connect()
        connection.select(folder.path)
        for mail in self.mail_ids:
            if not mail.object_id:
                continue
            msgid = mail.msgid
            mail_message, message_org = folder.fetch_msg(connection, msgid)
            folder.attach_mail(mail.object_id, mail_message)
            folder.update_msg(
                connection, msgid, matched=True,
                flagged=folder.flag_nonmatching)
        connection.close()
        return {'type': 'ir.actions.act_window_close'}

    @api.model
    def fields_view_get(
            self, view_id=None, view_type='form',
            toolbar=False, submenu=False):
        result = super(AttachMailManually, self).fields_view_get(
            view_id=view_id, view_type=view_type, toolbar=toolbar,
            submenu=submenu)
        folder_model = self.env['fetchmail.server.folder']
        folder_id = self.env.context.get('folder_id')
        folder = folder_model.browse([folder_id])
        tree = result['fields']['mail_ids']['views']['tree']
        tree['fields']['object_id']['selection'] = [
            (folder.model_id.model, folder.model_id.name)]
        return result


class AttachMailManuallyMail(models.TransientModel):
    _name = 'fetchmail.attach.mail.manually.mail'

    wizard_id = fields.Many2one(
        'fetchmail.attach.mail.manually', readonly=True)
    msgid = fields.Char('Message id', readonly=True)
    subject = fields.Char('Subject', readonly=True)
    date = fields.Datetime('Date', readonly=True)
    object_id = fields.Reference(
        lambda self: [
            (m.model, m.name)
            for m in self.env['ir.model'].search([])],
        string='Object')