import logging import textwrap import uuid from dateutil.relativedelta import relativedelta from odoo import api, fields, models, _ from odoo.exceptions import ValidationError from odoo.tools import float_is_zero _logger = logging.getLogger(__name__) class SurveyUserInput(models.Model): _inherit = 'survey.user_input' registration_id = fields.Many2one('event.registration', 'Event registration') #registration created automaticaly on survey post def _prepare_registration(self): """Extract registration values from the answers""" elegible_inputs = self.user_input_line_ids.filtered( lambda x: x.question_id.event_registration_field and not x.skipped ) vals = {} for line in elegible_inputs: if line.question_id.event_registration_field.ttype == 'many2one': vals[line.question_id.event_registration_field.name] = line.suggested_answer_id.record_reference else: vals[line.question_id.event_registration_field.name] = line[f"value_{line.answer_type}"] return vals def _create_registration_post_process(self, registration): """After creating the event registration send an internal message with the input link""" registration.message_post_with_view( "mail.message_origin_link", values={"self": registration, "origin": self}, subtype_id=self.env.ref("mail.mt_note").id, ) def _mark_done(self): """Generate registration when the survey is submitted""" for user_input in self.filtered( lambda r: r.survey_id.generate_registration and not self.registration_id ): vals = user_input._prepare_registration() # check doublon : if old draft registration already exists : update it email = vals.get('email') event_id = vals.get('event_id') old_registration = False if email and event_id: old_registration = self.env["event.registration"].search([('email','=',email),('event_id','=',event_id)]) if old_registration: old_registration = old_registration[0] if old_registration.state == 'draft': registration = old_registration registration.write(vals) registration.message_post_with_view( "mail.message_origin_link", values={"edit":True, "self": registration, "origin": user_input}, subtype_id=self.env.ref("mail.mt_note").id, ) if not old_registration: registration = self.env["event.registration"].create(vals) self._create_registration_post_process(registration) self.update({"registration_id": registration.id}) res = super()._mark_done() # after all, set partner id of registration as the partner of user input for user_input in self: if user_input.registration_id and user_input.partner_id: user_input.registration_id.partner_id = user_input.partner_id return res