Fix inconsistency in submission objects

CybercentreCanada · Jul 22, 2021 · edff383 · edff383
1 parent b7ecd43
commit edff383
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/assemblyline/odm/messages/submission.py b/assemblyline/odm/messages/submission.py
@@ -20,6 +20,7 @@ class Submission(odm.Model):
     metadata: Dict[str, str] = odm.FlattenedObject(default={})           # Metadata submitted with the file
     notification: Notification = odm.Compound(Notification, default={})  # Notification queue parameters
     params: SubmissionParams = odm.Compound(SubmissionParams)            # Parameters of the submission
+    scan_key: str = odm.Optional(odm.Keyword())
 
 
 def from_datastore_submission(submission: DatabaseSubmission):
@@ -31,7 +32,8 @@ def from_datastore_submission(submission: DatabaseSubmission):
         'sid': submission.sid,
         'files': submission.files,
         'metadata': submission.metadata,
-        'params': submission.params
+        'params': submission.params,
+        'scan_key': submission.scan_key
     })