Merge pull request #340 from CybercentreCanada/hotfix/default_service…

…_params Hotfix/default service params
CybercentreCanada · Aug 27, 2021 · 5592736 · 5592736
2 parents 5ab4886 + b9c2503
commit 5592736
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/assemblyline/odm/models/service.py b/assemblyline/odm/models/service.py
@@ -74,6 +74,7 @@ class SubmissionParams(odm.Model):
     type = odm.Enum(values=['str', 'int', 'list', 'bool'])
     value = odm.Any()
     list = odm.Optional(odm.Any())
+    hide = odm.Boolean(default=False)
 
 
 @odm.model(index=True, store=False)

diff --git a/assemblyline/odm/models/service_delta.py b/assemblyline/odm/models/service_delta.py
@@ -68,6 +68,7 @@ class SubmissionParamsDelta(odm.Model):
     type = odm.Optional(odm.Enum(values=['str', 'int', 'list', 'bool']))
     value = odm.Optional(odm.Any())
     list = odm.Optional(odm.Any())
+    hide = odm.Optional(odm.Any())
 
 
 @odm.model(index=True, store=False)

diff --git a/assemblyline/odm/models/user_settings.py b/assemblyline/odm/models/user_settings.py
@@ -22,7 +22,7 @@ class UserSettings(odm.Model):                                      # User's def
     malicious = odm.Boolean(default=False)                            # Is the file submitted known to be malicious
     priority = odm.Integer(default=1000)                              # Default priority for the submissions
     profile = odm.Boolean(default=False)                              # Should the submission do extra profiling
-    service_spec = odm.Mapping(odm.Any(), default={})                 # Default service specific settings
+    service_spec = odm.Mapping(odm.Mapping(odm.Any()), default={})    # Default service specific settings
     services = odm.Compound(ServiceSelection, default={})             # Default service selection
     submission_view = odm.Enum(values=VIEWS, default="report")        # Default view for completed submissions
     ttl = odm.Integer(default=30)                                     # Default submission Time to Live (days)