1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2011 Tiny SPRL (<http://tiny.be>).
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as
9 # published by the Free Software Foundation, either version 3 of the
10 # License, or (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU Affero General Public License for more details.
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 ##############################################################################
22 from osv import osv, fields
25 class portal(osv.osv):
27 _description = 'Portal'
29 'name': fields.char(string='Name', size=64, required=True),
30 'menu_id': fields.many2one('ir.actions.actions', required=True,
32 help="What replaces the standard menu for the portal's users"),
33 'user_ids': fields.one2many('res.users', 'portal_id',
35 help='The set of users associated to this portal'),
36 'group_ids': fields.many2many('res.groups', 'res_portals_groups_rel', 'pid', 'gid',
38 help='Users of this portal automatically belong to those groups'),
41 ('unique_name', 'UNIQUE(name)', 'Portals must have different names.')
44 def copy(self, cr, uid, id, defaults, context=None):
45 """ override copy() to not copy the portal users """
46 # find an unused name of the form "old_name [N]" for some random N
47 old_name = self.browse(cr, uid, id, context).name
48 new_name = copy_random(old_name)
49 while self.search(cr, uid, [('name', '=', new_name)], limit=1, context=context):
50 new_name = copy_random(old_name)
52 defaults['name'] = new_name
53 defaults['user_ids'] = []
54 return super(portal, self).copy(cr, uid, id, defaults, context)
56 def create(self, cr, uid, values, context=None):
57 """ extend create() to assign the portal menu and groups to users """
58 # as 'user_ids' is a many2one relation, values['user_ids'] must be a
59 # list of tuples of the form (0, 0, {values})
60 for op, _, user_values in values['user_ids']:
62 user_values['menu_id'] = values['menu_id']
63 user_values['groups_id'] = values['group_ids']
65 return super(portal, self).create(cr, uid, values, context)
67 def write(self, cr, uid, ids, values, context=None):
68 """ extend write() to reflect menu and groups changes on users """
70 # analyse groups changes, and determine how to change users
72 for change in values.get('group_ids', []):
73 if change[0] in [0, 5, 6]: # change creates or sets groups,
74 groups_diff = None # must compute per-portal diff
76 if change[0] in [3, 4]: # change add or remove group,
77 groups_diff.append(change) # add or remove group on users
79 if groups_diff is None:
80 return self._write_compute_diff(cr, uid, ids, values, context)
82 return self._write_diff(cr, uid, ids, values, groups_diff, context)
84 def _write_diff(self, cr, uid, ids, values, groups_diff, context=None):
85 """ perform write() and apply groups_diff on users """
86 # first apply portal changes
87 super(portal, self).write(cr, uid, ids, values, context)
89 # then apply menu and group changes on their users
91 if 'menu_id' in values:
92 user_values['menu_id'] = values['menu_id']
94 user_values['groups_id'] = groups_diff
98 for p in self.browse(cr, uid, ids, context):
99 user_ids += get_browse_ids(p.user_ids)
100 self.pool.get('res.users').write(cr, uid, user_ids, user_values, context)
104 def _write_compute_diff(self, cr, uid, ids, values, context=None):
105 """ perform write(), then compute and apply groups_diff on each portal """
106 # read group_ids before write() to compute groups_diff
108 for p in self.browse(cr, uid, ids, context):
109 old_group_ids[p.id] = get_browse_ids(p.group_ids)
111 # apply portal changes
112 super(portal, self).write(cr, uid, ids, values, context)
114 # the changes to apply on users
115 user_object = self.pool.get('res.users')
117 if 'menu_id' in values:
118 user_values['menu_id'] = values['menu_id']
120 # compute groups_diff on each portal, and apply them on users
121 for p in self.browse(cr, uid, ids, context):
122 old_groups = set(old_group_ids[p.id])
123 new_groups = set(get_browse_ids(p.group_ids))
124 # groups_diff: [(3, UNLINKED_ID), ..., (4, LINKED_ID), ...]
125 user_values['groups_id'] = \
126 [(3, g) for g in (old_groups - new_groups)] + \
127 [(4, g) for g in (new_groups - old_groups)]
128 user_ids = get_browse_ids(p.user_ids)
129 user_object.write(cr, uid, user_ids, user_values, context)
133 def action_create_menu(self, cr, uid, ids, context=None):
134 """ create a menu for this portal """
136 raise ValueError("portal.action_create_menu() applies to one portal only")
137 portal_name = self.browse(cr, uid, ids[0], context).name
139 # create a menuitem under 'portal.portal_menu_tree'
142 'parent_id': self._get_res_id(cr, uid, 'portal', 'portal_menu_tree'),
144 item_id = self.pool.get('ir.ui.menu').create(cr, uid, item_data, context)
146 # create an action to open the menuitems under item_id
149 'type': 'ir.actions.act_window',
151 'res_model': 'ir.ui.menu',
153 'view_id': self._get_res_id(cr, uid, 'base', 'view_menu'),
154 'domain': [('parent_id', '=', item_id)],
156 action_id = self.pool.get('ir.actions.act_window').create(cr, uid, action_data, context)
158 # set the portal menu_id to action_id
159 return self.write(cr, uid, ids, {'menu_id': action_id}, context)
161 def _get_res_id(self, cr, uid, module, xml_id):
162 """ return the resource id associated to the given xml_id """
163 ir_model_data_obj = self.pool.get('ir.model.data')
164 record_id = ir_model_data_obj._get_id(cr, uid, module, xml_id)
165 record_data = ir_model_data_obj.read(cr, uid, [record_id], ['res_id'])
166 assert (len(record_data) == 1) and ('res_id' in record_data[0])
167 return record_data[0]['res_id']
171 class users(osv.osv):
173 _inherit = 'res.users'
175 'portal_id': fields.many2one('res.portal', string='Portal',
176 help='If given, the portal defines customized menu and access rules'),
179 def default_get(self, cr, uid, fields, context=None):
180 """ override default values of menu_id and groups_id for portal users """
182 # How it works: the values of 'menu_id' and 'groups_id' are passed in
183 # context by the portal form view
184 if ('menu_id' in context) and ('menu_id' in fields):
185 fields.remove('menu_id')
186 others['menu_id'] = context['menu_id']
187 if ('groups_id' in context) and ('groups_id' in fields):
188 fields.remove('groups_id')
189 others['groups_id'] = get_many2many(context['groups_id'])
190 # the remaining fields use inherited defaults
191 defs = super(users, self).default_get(cr, uid, fields, context)
198 def get_browse_id(obj):
199 """ return the id of a browse() object, or None """
200 return (obj and obj.id or None)
202 def get_browse_ids(objs):
203 """ return the ids of a list of browse() objects """
204 return map(get_browse_id, objs)
206 def get_many2many(arg):
207 """ get the list of ids from a many2many 'values' field """
208 assert len(arg) == 1 and arg[0][0] == 6 # arg = [(6, _, IDs)]
211 def copy_random(name):
212 """ return "name [N]" for some random integer N """
213 return "%s [%s]" % (name, random.choice(xrange(1000000)))