Fix: Fixed key issue when splitting results

PUNCH-Cyber · May 25, 2016 · 35b84d9 · 35b84d9
1 parent 5116ee3
commit 35b84d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stoq-framework/plugins.py b/stoq-framework/plugins.py
@@ -753,7 +753,7 @@ def start(self, payload=None, **kwargs):
                 worker_result['source_meta'].pop(k, None)
 
         worker_result['payload_id'] = 0
-        results['plugins'].update({0: self.name})
+        results['plugins'].update({"0": self.name})
 
         # Keep track of our total count of payloads, in case yara dispatch
         # finds something
@@ -810,7 +810,7 @@ def start(self, payload=None, **kwargs):
                             self.save_payload(yara_result[1], self.archive_connector)
 
                         results['results'].append(dispatch_result)
-                        results['plugins'].update({payload_id: dispatch_result['plugin']})
+                        results['plugins'].update({str(payload_id): dispatch_result['plugin']})
 
                         payload_id += 1