[IMP] survey_extra_fields: new params max_file_size and allowed_extensions for file question type

survey_extra_fields/models/survey_question.py

@@ -9,3 +9,12 @@ class SurveyQuestion(models.Model):
     question_type = fields.Selection(
         selection_add=[("file", "File")]
     )
+    max_file_size = fields.Integer(
+        string="Max File Size (MB)",
+        default=10,
+        help="Maximum file size in MB. Leave 0 for no limit.",
+    )
+    allowed_extensions = fields.Char(
+        string="Allowed Extensions",
+        help="Comma-separated list of allowed extensions (e.g. .pdf,.docx). Leave empty to allow all types.",
+    )

survey_extra_fields/models/survey_user_input.py

@@ -1,14 +1,23 @@
 # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
 
-import json
+from __future__ import annotations
 
-from odoo import models
+import base64
+import json
+import os
+from typing import TYPE_CHECKING
+
+from odoo import _, models
+from odoo.exceptions import ValidationError
+
+if TYPE_CHECKING:
+    from odoo.addons.survey_extra_fields.models.survey_question import SurveyQuestion
 
 
 class SurveyUserInput(models.Model):
     _inherit = "survey.user_input"
 
-    def save_lines(self, question, answer, comment=None):
+    def save_lines(self, question: SurveyQuestion, answer: str | None, comment: str | None = None) -> None:
         if question.question_type == "file":
             old_answers = self.env["survey.user_input.line"].search([
                 ("user_input_id", "=", self.id),
@@ -22,8 +31,11 @@ class SurveyUserInput(models.Model):
             }
             if answer:
                 file_data = json.loads(answer)
-                vals["value_file"] = file_data.get("data")
-                vals["value_file_fname"] = file_data.get("name")
+                file_b64 = file_data.get("data", "")
+                file_name = file_data.get("name", "")
+                self._check_file_constraints(question, file_b64, file_name)
+                vals["value_file"] = file_b64
+                vals["value_file_fname"] = file_name
             else:
                 vals.update(answer_type=None, skipped=True)
             if old_answers:
@@ -32,3 +44,30 @@ class SurveyUserInput(models.Model):
                 self.env["survey.user_input.line"].create(vals)
         else:
             return super().save_lines(question, answer, comment=comment)
+
+    def _check_file_constraints(
+            self,
+            question: SurveyQuestion,
+            file_b64: str,
+            file_name: str
+    ) -> None:
+        if question.max_file_size:
+            file_size = len(base64.b64decode(file_b64))
+            max_bytes = question.max_file_size * 1024 * 1024
+            if file_size > max_bytes:
+                raise ValidationError(
+                    _("The file '%(name)s' exceeds the maximum allowed size of %(size)s MB.",
+                      name=file_name, size=question.max_file_size)
+                )
+        if question.allowed_extensions:
+            allowed = [
+                allowed_extension.strip().lower()
+                for allowed_extension in question.allowed_extensions.split(",")
+                if allowed_extension.strip()
+            ]
+            file_extension = os.path.splitext(file_name)[1].lower()
+            if file_extension not in allowed:
+                raise ValidationError(
+                    _("The file '%(name)s' is not allowed. Accepted formats: %(exts)s.",
+                      name=file_name, exts=question.allowed_extensions)
+                )