tryton/trytond/ir/ui/menu.py

#This file is part of Tryton.  The COPYRIGHT file at the top level of
#this repository contains the full copyright notices and license terms.
"UI menu"
from trytond.osv import fields, OSV

def one_in(i, j):
    """Check the presence of an element of setA in setB
    """
    for k in i:
        if k in j:
            return True
    return False

ICONS = [(x, x) for x in [
    'tryton-accessories',
    'tryton-attachment',
    'tryton-clear',
    'tryton-close',
    'tryton-calculator',
    'tryton-calendar',
    'tryton-clock',
    'tryton-connect',
    'tryton-copy',
    'tryton-currency',
    'tryton-delete',
    'tryton-development',
    'tryton-dialog-error',
    'tryton-dialog-information',
    'tryton-dialog-warning',
    'tryton-disconnect',
    'tryton-executable',
    'tryton-find',
    'tryton-find-replace',
    'tryton-folder-new',
    'tryton-folder-saved-search',
    'tryton-fullscreen',
    'tryton-graph',
    'tryton-go-home',
    'tryton-go-jump',
    'tryton-go-next',
    'tryton-go-previous',
    'tryton-help',
    'tryton-image-missing',
    'tryton-information',
    'tryton-lock',
    'tryton-list',
    'tryton-list-add',
    'tryton-list-remove',
    'tryton-locale',
    'tryton-log-out',
    'tryton-mail-message-new',
    'tryton-new',
    'tryton-noimage',
    'tryton-open',
    'tryton-package',
    'tryton-preferences',
    'tryton-preferences-system',
    'tryton-preferences-system-session',
    'tryton-presentation',
    'tryton-print',
    'tryton-readonly',
    'tryton-refresh',
    'tryton-save-as',
    'tryton-save',
    'tryton-spreadsheet',
    'tryton-start-here',
    'tryton-tree',
    'tryton-system',
    'tryton-system-file-manager',
    'tryton-users',
    'tryton-web-browser',
]]


class UIMenu(OSV):
    "UI menu"
    _name = 'ir.ui.menu'
    _description = __doc__
    name = fields.Char('Menu', required=True, translate=True)
    sequence = fields.Integer('Sequence')
    childs = fields.One2Many('ir.ui.menu', 'parent','Childs')
    parent = fields.Many2One('ir.ui.menu', 'Parent Menu', select=1)
    groups = fields.Many2Many('res.group', 'ir_ui_menu_group_rel',
       'menu_id', 'gid', 'Groups', ondelete_target='CASCADE')
    complete_name = fields.Function('get_full_name',
       string='Complete Name', type='char', order_field='name')
    icon = fields.Selection(ICONS, 'Icon', translate=False)
    action = fields.Function('get_action', fnct_inv='action_inv',
       type='reference', string='Action',
       selection=[
           ('ir.action.report', 'ir.action.report'),
           ('ir.action.act_window', 'ir.action.act_window'),
           ('ir.action.wizard', 'ir.action.wizard'),
           ('ir.action.url', 'ir.action.url'),
           ])
    active = fields.Boolean('Active')

    def __init__(self):
        super(UIMenu, self).__init__()
        self._order.insert(0, ('sequence', 'ASC'))

    def default_icon(self, cursor, user, context=None):
        return 'tryton-open'

    def default_sequence(self, cursor, user, context=None):
        return 10

    def default_active(self, cursor, user, context=None):
        return True

    def get_full_name(self, cursor, user, ids, name, args, context):
        res = {}
        for menu in self.browse(cursor, user, ids, context=context):
            res[menu.id] = self._get_one_full_name(menu)
        return res

    def _get_one_full_name(self, menu, level=6):
        if level <= 0:
            return '...'
        if menu.parent:
            parent_path = self._get_one_full_name(menu.parent, level-1) + "/"
        else:
            parent_path = ''
        return parent_path + menu.name

    def get_action(self, cursor, user, ids, name, arg, context=None):
        action_keyword_obj = self.pool.get('ir.action.keyword')

        if context is None:
            context = {}
        ctx = context.copy()
        ctx['active_test'] = False

        res = {}
        for menu_id in ids:
            res[menu_id] = False
        action_keyword_ids = action_keyword_obj.search(cursor, user, [
            ('keyword', '=', 'tree_open'),
            ('model', 'in', [self._name + ',' + str(x) for x in ids]),
            ], context=ctx)
        for action_keyword in action_keyword_obj.browse(cursor, user,
                action_keyword_ids, context=context):
            model_id = int(
                    action_keyword.model.split(',')[1].split(',')[0].strip('('))
            action_obj = self.pool.get(action_keyword.action.type)
            action_id = action_obj.search(cursor, user, [
                ('action', '=', action_keyword.action.id),
                ], context=ctx)
            if action_id:
                action_id = action_id[0]
                action = action_obj.browse(cursor, user, action_id,
                        context=context)
                action_name = action.rec_name
            else:
                action_id = 0
                action_name = ''
            res[model_id] = action_keyword.action.type + \
                    ',(' + str(action_id) + ',"' + action_name + '")'
        return res

    def action_inv(self, cursor, user, menu_id, name, value, arg,
            context=None):
        if context is None:
            context = {}
        if not value:
            return
        ctx = context.copy()
        if '_timestamp' in ctx:
            del ctx['_timestamp']
        action_keyword_obj = self.pool.get('ir.action.keyword')
        action_keyword_ids = action_keyword_obj.search(cursor, user, [
            ('keyword', '=', 'tree_open'),
            ('model', '=', self._name + ',' + str(menu_id)),
            ], context=context)
        if action_keyword_ids:
            action_keyword_obj.delete(cursor, user, action_keyword_ids,
                    context=ctx)
        action_type, action_id = value.split(',')
        if not int(action_id):
            return
        action_obj = self.pool.get(action_type)
        action = action_obj.browse(cursor, user, int(action_id),
                context=context)
        action_keyword_obj.create(cursor, user, {
            'keyword': 'tree_open',
            'model': self._name + ',' + str(menu_id),
            'action': action.action.id,
            }, context=ctx)

UIMenu()