initial commit 2

1 files changed, 22 insertions, 0 deletions

diff --git a/addons/website_slides_survey/controllers/survey.py b/addons/website_slides_survey/controllers/survey.py
new file mode 100644
index 00000000..85a87001
--- /dev/null
+++ b/addons/website_slides_survey/controllers/survey.py
@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo.addons.survey.controllers.main import Survey
+
+
+class Survey(Survey):
+    def _prepare_survey_finished_values(self, survey, answer, token=False):
+        result = super(Survey, self)._prepare_survey_finished_values(survey, answer, token)
+        if answer.slide_id:
+            result['channel_url'] = answer.slide_id.channel_id.website_url
+
+        return result
+
+    def _prepare_retry_additional_values(self, answer):
+        result = super(Survey, self)._prepare_retry_additional_values(answer)
+        if answer.slide_id:
+            result['slide_id'] = answer.slide_id.id
+        if answer.slide_partner_id:
+            result['slide_partner_id'] = answer.slide_partner_id.id
+
+        return result