assemblyline-core 4.5.1.dev359__tar.gz → 4.5.1.dev361__tar.gz

{assemblyline-core-4.5.1.dev359 → assemblyline-core-4.5.1.dev361}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: assemblyline-core
-Version: 4.5.1.dev359
+Version: 4.5.1.dev361
 Summary: Assemblyline 4 - Core components
 Home-page: https://github.com/CybercentreCanada/assemblyline-core/
 Author: CCCS Assemblyline development team

assemblyline-core-4.5.1.dev361/assemblyline_core/VERSION

@@ -0,0 +1 @@
+4.5.1.dev361

{assemblyline-core-4.5.1.dev359 → assemblyline-core-4.5.1.dev361}/assemblyline_core/dispatching/dispatcher.py

@@ -11,6 +11,7 @@ import json
 import enum
 from queue import PriorityQueue, Empty, Queue
 import dataclasses
+from copy import deepcopy
 
 import elasticapm
 
@@ -102,6 +103,7 @@ class MonitorTask:
     # Should aservice be dispatched again when possible
     dispatch_needed: bool = dataclasses.field(default=False)
 
+
 @contextmanager
 def apm_span(client, span_name: str):
     try:
@@ -126,15 +128,20 @@ class ResultSummary:
 
 
 class TemporaryFileData:
-    def __init__(self, sha256: str, config: dict[str, str], shared: Optional[dict[str, Any]] = None) -> None:
+    def __init__(self,
+                 sha256: str,
+                 config: dict[str, str],
+                 shared: Optional[dict[str, Any]] = None,
+                 local: Optional[dict[str, Any]] = None
+                 ) -> None:
         self.sha256 = sha256
         self.config = config
-        self.shared_values: dict[str, Any] = dict() if shared is None else shared            
-        self.local_values: dict[str, Any] = {}
+        self.shared_values: dict[str, Any] = {} if shared is None else shared
+        self.local_values: dict[str, Any] = {} if local is None else local
 
     def new_file(self, sha256: str) -> TemporaryFileData:
         """Create an entry for another file with reference to the shared values."""
-        return TemporaryFileData(sha256, self.config, self.shared_values)
+        return TemporaryFileData(sha256, self.config, self.shared_values, deepcopy(self.local_values))
 
     def read(self) -> dict[str, Any]:
         """Get a copy of the current data"""
@@ -304,7 +311,8 @@ class SubmissionTask:
         """
         parent_temp = self.temporary_data[parent]
         for child in children:
-            self.temporary_data.setdefault(child, parent_temp.new_file(child))
+            if child not in self.temporary_data:
+                self.temporary_data[child] = parent_temp.new_file(child)
             try:
                 self._parent_map[child].add(parent)
             except KeyError:

{assemblyline-core-4.5.1.dev359 → assemblyline-core-4.5.1.dev361}/assemblyline_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: assemblyline-core
-Version: 4.5.1.dev359
+Version: 4.5.1.dev361
 Summary: Assemblyline 4 - Core components
 Home-page: https://github.com/CybercentreCanada/assemblyline-core/
 Author: CCCS Assemblyline development team

assemblyline-core-4.5.1.dev359/assemblyline_core/VERSION

@@ -1 +0,0 @@
-4.5.1.dev359