server-tools/fetchmail_attach_from_folder/model/fetchmail_server_folder.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/>.
#
########################################################################

from openerp.osv import fields
from openerp.osv.orm import Model
from .. import match_algorithm


class fetchmail_server_folder(Model):
    _name = 'fetchmail.server.folder'
    _rec_name = 'path'

    def _get_match_algorithms(self):
        def get_all_subclasses(cls):
            return cls.__subclasses__() + [subsub
                                           for sub in cls.__subclasses__()
                                           for subsub in get_all_subclasses(sub)]
        return dict([(cls.__name__, cls) for cls in get_all_subclasses(
            match_algorithm.base.base)])

    def _get_match_algorithms_sel(self, cr, uid, context=None):
        algorithms = []
        for cls in self._get_match_algorithms().itervalues():
            algorithms.append((cls.__name__, cls.name))
        algorithms.sort()
        return algorithms

    _columns = {
        'sequence': fields.integer('Sequence'),
        'path': fields.char(
            'Path', size=256, help='The path to your mail '
            "folder. Typically would be something like 'INBOX.myfolder'",
            required=True
        ),
        'model_id': fields.many2one(
            'ir.model', 'Model', required=True,
            help='The model to attach emails to'
        ),
        'model_field': fields.char(
            'Field (model)', size=128,
            help='The field in your model that contains the field to match '
                 'against.\n'
                 'Examples:\n'
                 "'email' if your model is res.partner, or "
                 "'partner_id.email' if you're matching sale orders"
        ),
        'model_order': fields.char(
            'Order (model)', size=128,
            help='Fields to order by, this mostly useful in conjunction '
                 "with 'Use 1st match'"
        ),
        'match_algorithm': fields.selection(
            _get_match_algorithms_sel,
            'Match algorithm', required=True, translate=True,
            help='The algorithm used to determine which object an email '
                 'matches.'
        ),
        'mail_field': fields.char(
            'Field (email)', size=128,
            help='The field in the email used for matching. Typically '
                 "this is 'to' or 'from'"
        ),
        'server_id': fields.many2one('fetchmail.server', 'Server'),
        'delete_matching': fields.boolean(
            'Delete matches',
            help='Delete matched emails from server'
        ),
        'flag_nonmatching': fields.boolean(
            'Flag nonmatching',
            help="Flag emails in the server that don't match any object "
                 'in OpenERP'
        ),
        'match_first': fields.boolean(
            'Use 1st match',
            help='If there are multiple matches, use the first one. If '
                 'not checked, multiple matches count as no match at all'
        ),
        'domain': fields.char(
            'Domain', size=128, help='Fill in a search '
            'filter to narrow down objects to match'
        ),
        'msg_state': fields.selection(
            [
                ('sent', 'Sent'),
                ('received', 'Received'),
            ],
            'Message state',
            help='The state messages fetched from this folder should be '
                 'assigned in OpenERP'
        ),
    }

    _defaults = {
        'flag_nonmatching': True,
        'msg_state': 'received',
    }

    def get_algorithm(self, cr, uid, ids, context=None):
        for this in self.browse(cr, uid, ids, context):
            return self._get_match_algorithms()[this.match_algorithm]()

    def button_attach_mail_manually(self, cr, uid, ids, context=None):
        for this in self.browse(cr, uid, ids, context):
            context.update({'default_folder_id': this.id})
        return {
            'type': 'ir.actions.act_window',
            'res_model': 'fetchmail.attach.mail.manually',
            'target': 'new',
            'context': context,
            'view_type': 'form',
            'view_mode': 'form',
        }