odoo-survey-addons/survey_attachment/models/survey_user_input_line.py

# -*- coding: utf-8 -*-
import base64
from odoo import models, fields, api, _
from .survey_question import TYPES

FILE_TYPES = [
    ('image', _('Image')),
    ('pdf', _('PDF file')),
]


class SurveyUserInputLine(models.Model):
    _inherit = 'survey.user_input_line'

    answer_type = fields.Selection(selection=TYPES)
    file = fields.Binary(string='Uploaded file')
    filename = fields.Char(string='Uploaded file name')
    file_type = fields.Selection(selection=FILE_TYPES, string="File type")

    @api.model
    def save_line_upload_file(self, user_input_id, question, post, answer_tag):
        vals = {
            'user_input_id': user_input_id,
            'question_id': question.id,
            'survey_id': question.survey_id.id,
            'skipped': False
        }
        if question.constr_mandatory:
            file = base64.encodebytes(post[answer_tag].read())
        else:
            file = base64.encodebytes(post[answer_tag].read()) if post[answer_tag] else None
        if answer_tag in post:
            vals.update({
                'answer_type': 'upload_file',
                'file': file,
                'filename': post[answer_tag].filename,
            })
            if post[answer_tag].content_type == 'application/pdf':
                vals.update({'file_type': 'pdf'})
            if 'image/' in post[answer_tag].content_type:
                vals.update({'file_type': 'image'})
        else:
            vals.update({
                'answer_type': None,
                'file': False,
                'filename': False,
                'skipped': True,
            })
        old_uil = self.search([
            ('user_input_id', '=', user_input_id),
            ('survey_id', '=', question.survey_id.id),
            ('question_id', '=', question.id)
        ])
        if old_uil:
            old_uil.write(vals)
        else:
            old_uil.create(vals)
        return True