# -*- coding: utf-8 -*- ############################################################################## # # OpenERP, Open Source Management Solution # Copyright (C) 2004-2010 Tiny SPRL (). # # 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 . # ############################################################################## from osv import fields,osv import tools class hr_department(osv.osv): def name_get(self, cr, uid, ids, context=None): if not len(ids): return [] reads = self.read(cr, uid, ids, ['name','parent_id'], context) res = [] for record in reads: name = record['name'] if record['parent_id']: name = record['parent_id'][1]+' / '+name res.append((record['id'], name)) return res def _dept_name_get_fnc(self, cr, uid, ids, prop, unknow_none, context): res = self.name_get(cr, uid, ids, context) return dict(res) _name = "hr.department" _columns = { 'name': fields.char('Department Name', size=64, required=True), 'complete_name': fields.function(_dept_name_get_fnc, method=True, type="char", string='Name'), 'company_id': fields.many2one('res.company', 'Company', select=True, required=False), 'parent_id': fields.many2one('hr.department', 'Parent Department', select=True), 'child_ids': fields.one2many('hr.department', 'parent_id', 'Child Departments'), 'note': fields.text('Note'), 'manager_id': fields.many2one('res.users', 'Manager', required=True), 'member_ids': fields.many2many('res.users', 'hr_department_user_rel', 'department_id', 'user_id', 'Members'), } def _get_members(self,cr, uid, context={}): mids = self.search(cr, uid, [('manager_id','=',uid)]) result = {uid:1} for m in self.browse(cr, uid, mids, context): for user in m.member_ids: result[user.id] = 1 return result.keys() def _check_recursion(self, cr, uid, ids): level = 100 while len(ids): cr.execute('select distinct parent_id from hr_department where id =ANY(%s)',(ids,)) ids = filter(None, map(lambda x:x[0], cr.fetchall())) if not level: return False level -= 1 return True _constraints = [ (_check_recursion, 'Error! You can not create recursive departments.', ['parent_id']) ] hr_department() class ir_action_window(osv.osv): _inherit = 'ir.actions.act_window' def read(self, cr, uid, ids, fields=None, context=None, load='_classic_read'): select = ids if isinstance(ids, (int, long)): select = [ids] res = super(ir_action_window, self).read(cr, uid, select, fields=fields, context=context, load=load) for r in res: mystring = 'department_users_get()' if mystring in (r.get('domain', '[]') or ''): r['domain'] = r['domain'].replace(mystring, str( self.pool.get('hr.department')._get_members(cr, uid))) if isinstance(ids, (int, long)): if res: return res[0] else: return False return res ir_action_window() class res_users(osv.osv): _inherit = 'res.users' _description = 'User' def _parent_compute(self, cr, uid, ids, name, args, context={}): result = {} obj_dept = self.pool.get('hr.department') for user_id in ids: ids_dept = obj_dept.search(cr, uid, [('member_ids', 'in', [user_id])]) parent_ids = [] if ids_dept: data_dept = obj_dept.read(cr, uid, ids_dept, ['manager_id']) parent_ids = map(lambda x: x['manager_id'][0], data_dept) result[user_id] = parent_ids return result def _parent_search(self, cr, uid, obj, name, args, context): parent = [] for arg in args: if arg[0] == 'parent_id': parent = arg[2] child_ids = self._child_compute(cr, uid, parent,name, args, {}) if not child_ids: return [('id', 'in', [0])] return [('id', 'in', child_ids.get(uid,[]))] def _child_compute(self, cr, uid, ids, name, args, context={}): obj_dept = self.pool.get('hr.department') obj_user = self.pool.get('res.users') result = {} for manager_id in ids: child_ids = [] mgnt_dept_ids = obj_dept.search(cr, uid, [('manager_id', '=', manager_id)]) ids_dept = obj_dept.search(cr, uid, [('id', 'child_of', mgnt_dept_ids)]) if ids_dept: data_dept = obj_dept.read(cr, uid, ids_dept, ['member_ids']) childs = map(lambda x: x['member_ids'], data_dept) childs = tools.flatten(childs) childs = obj_user.search(cr, uid, [('id','in',childs),('active','=',True)]) if manager_id in childs: childs.remove(manager_id) child_ids.extend(tools.flatten(childs)) set = {} map(set.__setitem__, child_ids, []) child_ids = set.keys() else: child_ids = [] result[manager_id] = child_ids return result def _child_search(self, cr, uid, obj, name, args, context): parent = [] for arg in args: if arg[0] == 'child_ids': parent = arg[2] child_ids = self._child_compute(cr, uid, parent,name, args, {}) if not child_ids: return [('id', 'in', [0])] return [('id', 'in', child_ids.get(uid,[]))] _columns = { 'parent_id': fields.function(_parent_compute, relation='res.users',fnct_search=_parent_search, method=True, string="Managers", type='many2many'), 'child_ids': fields.function(_child_compute, relation='res.users', fnct_search=_child_search,method=True, string="Subordinates", type='many2many'), 'context_department_id': fields.many2one('hr.department', 'Departments'), } res_users() # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: