server-tools/fetchmail_attach_from_folder/wizard/attach_mail_manually.py

# -*- encoding: utf-8 -*-
##############################################################################
#
#    OpenERP, Open Source Management Solution
#    This module copyright (C) 2013 Therp BV (<http://therp.nl>)
#    All Rights Reserved
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

import logging
from openerp.osv import fields
from openerp.osv.orm import TransientModel


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

    _columns = {
        'folder_id': fields.many2one(
            'fetchmail.server.folder', 'Folder', readonly=True),
        'mail_ids': fields.one2many(
            'fetchmail.attach.mail.manually.mail', 'wizard_id', 'Emails'),
    }

    def default_get(self, cr, uid, fields_list, context=None):
        if context is None:
            context = {}

        defaults = super(attach_mail_manually, self).default_get(
            cr, uid, fields_list, context)

        for folder in self.pool.get('fetchmail.server.folder').browse(
                cr, uid, [context.get('default_folder_id')], context):
            defaults['mail_ids'] = []
            connection = folder.server_id.connect()
            connection.select(folder.path)
            result, msgids = connection.search(
                None,
                'FLAGGED' if folder.flag_nonmatching else 'UNDELETED')
            if result != 'OK':
                logging.error(
                    'Could not search mailbox %s on %s',
                    folder.path,
                    folder.server_id.server)
                continue
            attach_mail_manually_mail._columns['object_id'].selection = [
                (folder.model_id.model, folder.model_id.name)
            ]
            for msgid in msgids[0].split():
                result, msgdata = connection.fetch(msgid, '(RFC822)')
                if result != 'OK':
                    logging.error(
                        'Could not fetch %s in %s on %s',
                        msgid, folder.path, folder.server_id.server)
                    continue
                mail_message = self.pool.get('mail.message').parse_message(
                    msgdata[0][1])
                defaults['mail_ids'].append(
                    (0, 0,
                     {
                         'msgid': msgid,
                         'subject': mail_message.get('subject', ''),
                         'date': mail_message.get('date', ''),
                         'object_id': folder.model_id.model + ',False'
                     }))
            connection.close()

        return defaults

    def attach_mails(self, cr, uid, ids, context=None):
        for this in self.browse(cr, uid, ids, context):
            for mail in this.mail_ids:
                connection = this.folder_id.server_id.connect()
                connection.select(this.folder_id.path)
                result, msgdata = connection.fetch(mail.msgid, '(RFC822)')
                if result != 'OK':
                    logging.error(
                        'Could not fetch %s in %s on %s',
                        mail.msgid, this.folder_id.path,
                        this.folder_id.server_id.server)
                    continue

                mail_message = self.pool.get('mail.message').parse_message(
                    msgdata[0][1], this.folder_id.server_id.original)

                this.folder_id.server_id.attach_mail(
                    connection,
                    mail.object_id.id, this.folder_id, mail_message,
                    mail.msgid)
                connection.close()
        return {'type': 'ir.actions.act_window_close'}


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

    _columns = {
        'wizard_id': fields.many2one(
            'fetchmail.attach.mail.manually', readonly=True),
        'msgid': fields.char('Message id', size=16, readonly=True),
        'subject': fields.char('Subject', size=128, readonly=True),
        'date': fields.datetime('Date', readonly=True),
        'object_id': fields.reference(
            'Object',
            selection=lambda self, cr, uid, context:
            [(m.model, m.name) for m in
             self.pool.get('ir.model').browse(
                 cr, uid, self.pool.get('ir.model').search(cr, uid, []),
                 context)],
            size=128),
    }