# -*- coding: utf-8 -*- ############################################################################## # # OpenERP, Open Source Management Solution # Copyright (C) 2013 OpenERP SA () # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU 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 General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see # ############################################################################## from openerp.osv import osv class res_users_gamification_group(osv.Model): """ Update of res.users class - if adding groups to an user, check gamification.challenge linked to this group, and the user. This is done by overriding the write method. """ _name = 'res.users' _inherit = ['res.users'] def write(self, cr, uid, ids, vals, context=None): write_res = super(res_users_gamification_group, self).write(cr, uid, ids, vals, context=context) if vals.get('groups_id'): # form: {'group_ids': [(3, 10), (3, 3), (4, 10), (4, 3)]} or {'group_ids': [(6, 0, [ids]} user_group_ids = [command[1] for command in vals['groups_id'] if command[0] == 4] user_group_ids += [id for command in vals['groups_id'] if command[0] == 6 for id in command[2]] challenge_obj = self.pool.get('gamification.challenge') challenge_ids = challenge_obj.search(cr, uid, [('autojoin_group_id', 'in', user_group_ids)], context=context) if challenge_ids: challenge_obj.write(cr, uid, challenge_ids, {'user_ids': [(4, user_id) for user_id in ids]}, context=context) if vals.get('image'): goal_definition_id = self.pool.get('ir.model.data').get_object(cr, uid, 'gamification', 'definition_base_avatar', context) goal_ids = self.pool.get('gamification.goal').search(cr, uid, [('definition_id', '=', goal_definition_id.id), ('user_id', 'in', ids)], context=context) values = {'state': 'reached', 'current': 1} self.pool.get('gamification.goal').write(cr, uid, goal_ids, values, context=context) return write_res def get_goals_todo_info(self, cr, uid, context=None): """Return the list of goals assigned to the user, grouped by challenge This method intends to return processable data in javascript in the goal_list_to_do template. The output format is not constant as the required information is different between individual and board goal definitions :return: list of dictionnaries for each goal to display """ all_goals_info = [] challenge_obj = self.pool.get('gamification.challenge') challenge_ids = challenge_obj.search(cr, uid, [('user_ids', 'in', uid), ('state', '=', 'inprogress')], context=context) for challenge in challenge_obj.browse(cr, uid, challenge_ids, context=context): # serialize goals info to be able to use it in javascript serialized_goals_info = { 'id': challenge.id, 'name': challenge.name, 'visibility_mode': challenge.visibility_mode, } user = self.browse(cr, uid, uid, context=context) serialized_goals_info['currency'] = user.company_id.currency_id.id if challenge.visibility_mode == 'board': # board report should be grouped by line for all users goals_info = challenge_obj.get_board_goal_info(cr, uid, challenge, subset_goal_ids=False, context=context) if len(goals_info) == 0: # challenge with no valid lines continue serialized_goals_info['lines'] = [] for line_board in goals_info: vals = {'definition_name': line_board['goal_definition'].name, 'definition_description': line_board['goal_definition'].description, 'definition_condition': line_board['goal_definition'].condition, 'computation_mode': line_board['goal_definition'].computation_mode, 'definition_monetary': line_board['goal_definition'].monetary, 'definition_suffix': line_board['goal_definition'].suffix, 'definition_action': True if line_board['goal_definition'].action_id else False, 'definition_display': line_board['goal_definition'].display_mode, 'target_goal': line_board['target_goal'], 'goals': []} for goal in line_board['board_goals']: # Keep only the Top 3 and the current user if goal[0] > 2 and goal[1].user_id.id != uid: continue vals['goals'].append({ 'rank': goal[0] + 1, 'id': goal[1].id, 'user_id': goal[1].user_id.id, 'user_name': goal[1].user_id.name, 'state': goal[1].state, 'completeness': goal[1].completeness, 'current': goal[1].current, 'target_goal': goal[1].target_goal, }) if uid == goal[1].user_id.id: vals['own_goal_id'] = goal[1].id serialized_goals_info['lines'].append(vals) else: # individual report are simply a list of goal goals_info = challenge_obj.get_indivual_goal_info(cr, uid, uid, challenge, subset_goal_ids=False, context=context) if not goals_info: continue serialized_goals_info['goals'] = [] for goal in goals_info: serialized_goals_info['goals'].append({ 'id': goal.id, 'definition_name': goal.definition_id.name, 'definition_description': goal.definition_description, 'definition_condition': goal.definition_id.condition, 'definition_monetary': goal.definition_id.monetary, 'definition_suffix': goal.definition_id.suffix, 'definition_action': True if goal.definition_id.action_id else False, 'definition_display': goal.definition_id.display_mode, 'state': goal.state, 'completeness': goal.completeness, 'computation_mode': goal.computation_mode, 'current': goal.current, 'target_goal': goal.target_goal, }) all_goals_info.append(serialized_goals_info) return all_goals_info def get_challenge_suggestions(self, cr, uid, context=None): """Return the list of challenges suggested to the user""" challenge_info = [] challenge_obj = self.pool.get('gamification.challenge') challenge_ids = challenge_obj.search(cr, uid, [('invited_user_ids', 'in', uid), ('state', '=', 'inprogress')], context=context) for challenge in challenge_obj.browse(cr, uid, challenge_ids, context=context): values = { 'id': challenge.id, 'name': challenge.name, 'description': challenge.description, } challenge_info.append(values) return challenge_info class res_groups_gamification_group(osv.Model): """ Update of res.groups class - if adding users from a group, check gamification.challenge linked to this group, and the user. This is done by overriding the write method. """ _name = 'res.groups' _inherit = 'res.groups' def write(self, cr, uid, ids, vals, context=None): write_res = super(res_groups_gamification_group, self).write(cr, uid, ids, vals, context=context) if vals.get('users'): # form: {'group_ids': [(3, 10), (3, 3), (4, 10), (4, 3)]} or {'group_ids': [(6, 0, [ids]} user_ids = [command[1] for command in vals['users'] if command[0] == 4] user_ids += [id for command in vals['users'] if command[0] == 6 for id in command[2]] challenge_obj = self.pool.get('gamification.challenge') challenge_ids = challenge_obj.search(cr, uid, [('autojoin_group_id', 'in', ids)], context=context) if challenge_ids: challenge_obj.write(cr, uid, challenge_ids, {'user_ids': [(4, user_id) for user_id in user_ids]}, context=context) return write_res # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: