mail-addons/mail_move_message/controllers/main.py

# -*- coding: utf-8 -*-
from odoo.addons.web.controllers.main import DataSet
from odoo.tools.translate import _
from odoo import http
from odoo.http import request
from odoo.addons.bus.controllers.main import BusController


class MailChatController(BusController):
    # -----------------------------
    # Extends BUS Controller Poll
    # -----------------------------

    def _poll(self, dbname, channels, last, options):
        if request.session.uid:
            channels.append((request.db, 'mail_move_message'))
            channels.append((request.db, 'mail_move_message.delete_message'))
        return super(MailChatController, self)._poll(dbname, channels, last, options)


class DataSetCustom(DataSet):

    def _extend_name(self, model, records):
        Model = request.env[model]
        fields = Model.fields_get()
        contact_field = False
        for n, f in fields.iteritems():
            if f['type'] == 'many2one' and f['relation'] == 'res.partner':
                contact_field = n
                break
        partner_info = {}
        if contact_field:
            partner_info = Model.browse([r[0] for r in records]).read([contact_field])
            partner_info = dict([(p['id'], p[contact_field]) for p in partner_info])
        res = []
        for r in records:
            if partner_info.get(r[0]):
                res.append((r[0], _('%s [%s] ID %s') % (r[1], partner_info.get(r[0])[1], r[0])))
            else:
                res.append((r[0], _('%s ID %s') % (r[1], r[0])))
        return res

    @http.route('/web/dataset/call_kw/<model>/name_search', type='json', auth="user")
    def name_search(self, model, method, args, kwargs):
        context = kwargs.get('context')
        if context and context.get('extended_name_with_contact'):
            # add order by ID desc
            Model = request.env[model]
            search_args = list(kwargs.get('args') or [])
            limit = int(kwargs.get('limit') or 100)
            operator = kwargs.get('operator')
            name = kwargs.get('name')
            if Model._rec_name and (not name == '' and operator == 'ilike'):
                search_args += [(Model._rec_name, operator, name)]
            records = Model.search(search_args, limit=limit, order='id desc')
            res = records.name_get()
            return self._extend_name(model, res)

        return self._call_kw(model, method, args, kwargs)

    @http.route('/web/dataset/call_kw/<model>/name_get', type='json', auth="user")
    def name_get(self, model, method, args, kwargs):
        res = self._call_kw(model, method, args, kwargs)
        context = kwargs.get('context')
        if context and context.get('extended_name_with_contact'):
            res = self._extend_name(model, res)
        return res