initial commit 2

1 files changed, 31 insertions, 0 deletions

diff --git a/addons/hr_gamification/wizard/gamification_badge_user_wizard.py b/addons/hr_gamification/wizard/gamification_badge_user_wizard.py
new file mode 100644
index 00000000..6bb1370b
--- /dev/null
+++ b/addons/hr_gamification/wizard/gamification_badge_user_wizard.py
@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models, _
+from odoo.exceptions import UserError, AccessError
+
+
+class GamificationBadgeUserWizard(models.TransientModel):
+    _inherit = 'gamification.badge.user.wizard'
+
+    employee_id = fields.Many2one('hr.employee', string='Employee', required=True)
+    user_id = fields.Many2one('res.users', string='User', related='employee_id.user_id',
+        store=False, readonly=True, compute_sudo=True)
+
+    def action_grant_badge(self):
+        """Wizard action for sending a badge to a chosen employee"""
+        if not self.user_id:
+            raise UserError(_('You can send badges only to employees linked to a user.'))
+
+        if self.env.uid == self.user_id.id:
+            raise UserError(_('You can not send a badge to yourself.'))
+
+        values = {
+            'user_id': self.user_id.id,
+            'sender_id': self.env.uid,
+            'badge_id': self.badge_id.id,
+            'employee_id': self.employee_id.id,
+            'comment': self.comment,
+        }
+
+        return self.env['gamification.badge.user'].create(values)._send_badge()