From 6ecfbcb2226e730c4188a9e93586d591e5686aeb Mon Sep 17 00:00:00 2001
From: fieldOfView <aldo@fieldofview.com>
Date: Sat, 17 Feb 2018 10:29:42 +0100
Subject: [PATCH] Fix postprocessing

---
 plugins/BlackBeltPlugin/BlackBeltPlugin.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/plugins/BlackBeltPlugin/BlackBeltPlugin.py b/plugins/BlackBeltPlugin/BlackBeltPlugin.py
index b3e5773133..5ea144d1fa 100644
--- a/plugins/BlackBeltPlugin/BlackBeltPlugin.py
+++ b/plugins/BlackBeltPlugin/BlackBeltPlugin.py
@@ -132,8 +132,15 @@ class BlackBeltPlugin(Extension):
             return
 
         scene = Application.getInstance().getController().getScene()
-        if hasattr(scene, "gcode_list"):
-            gcode_list = getattr(scene, "gcode_list")
+        gcode_dict = getattr(scene, "gcode_dict", {})
+        if not gcode_dict: # this also checks for an empty dict
+            Logger.log("w", "Scene has no gcode to process")
+            return
+
+        dict_changed = False
+
+        for plate_id in gcode_dict:
+            gcode_list = gcode_dict[plate_id]
             if gcode_list:
                 if ";BLACKBELTPROCESSED" not in gcode_list[0]:
                     search_regex = re.compile(r"M106 S(\d*\.?\d*?)")
@@ -143,10 +150,14 @@ class BlackBeltPlugin(Extension):
                         gcode_list[layer_number] = re.sub(search_regex, replace_pattern, layer) #Replace all.
 
                     gcode_list[0] += ";BLACKBELTPROCESSED\n"
-                    setattr(scene, "gcode_list", gcode_list)
+                    gcode_dict[plate_id] = gcode_list
+                    dict_changed = True
                 else:
                     Logger.log("e", "Already post processed")
 
+        if dict_changed:
+            setattr(scene, "gcode_dict", gcode_dict)
+
 ## QML-accessible singleton for access to extended data on definition and variants
 class BlackBeltSingleton(QObject):
     def __init__(self):