initial commit 2

1 files changed, 16 insertions, 0 deletions

diff --git a/addons/survey/models/badge.py b/addons/survey/models/badge.py
new file mode 100644
index 00000000..7ed12731
--- /dev/null
+++ b/addons/survey/models/badge.py
@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+
+class GamificationBadge(models.Model):
+    _inherit = 'gamification.badge'
+
+    survey_ids = fields.One2many('survey.survey', 'certification_badge_id', 'Survey Ids')
+    survey_id = fields.Many2one('survey.survey', 'Survey', compute='_compute_survey_id', store=True)
+
+    @api.depends('survey_ids.certification_badge_id')
+    def _compute_survey_id(self):
+        for badge in self:
+            badge.survey_id = badge.survey_ids[0] if badge.survey_ids else None