Fixed ported urllib code from old cura

plugins/SliceInfoPlugin/SliceInfo.py

@@ -3,12 +3,104 @@
 
 from UM.Extension import Extension
 from UM.Application import Application
+from UM.Preferences import Preferences
+from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator
+from UM.Scene.SceneNode import SceneNode
 
+import collections
+import json
+import os.path
+import copy
+import platform
+import math
+import urllib.request
+import urllib.parse
+
+
+##      This Extension runs in the background and sends several bits of information to the Ultimaker servers.
+#       The data is only sent when the user in question gave permission to do so. All data is anonymous and
+#       no model files are being sent (Just a SHA256 hash of the model).
 class SliceInfo(Extension):
     def __init__(self):
         super().__init__()
         Application.getInstance().getOutputDeviceManager().writeStarted.connect(self._onWriteStarted)
+        Preferences.getInstance().addPreference("info/send_slice_info", True)
 
     def _onWriteStarted(self, output_device):
-        print("Write started")
-        pass
+        print("Starting sending")
+        if not Preferences.getInstance().getValue("info/send_slice_info"):
+            return # Do nothing, user does not want to send data
+
+        settings = Application.getInstance().getActiveMachine()
+        profile = settings.getActiveProfile()
+        profile_values = None
+
+        # Load all machine definitions and put them in machine_settings dict
+        setting_file_name = Application.getInstance().getActiveMachine()._json_file
+        machine_settings = {}
+        with open(setting_file_name, "rt", -1, "utf-8") as f:
+            data = json.load(f, object_pairs_hook = collections.OrderedDict)
+        machine_settings[os.path.basename(setting_file_name)] = copy.deepcopy(data)
+        # Loop through inherited json files
+        while True:
+            if "inherits" in data:
+                inherited_setting_file_name = os.path.dirname(setting_file_name) + "/" + data["inherits"]
+                with open(inherited_setting_file_name, "rt", -1, "utf-8") as f:
+                    data = json.load(f, object_pairs_hook = collections.OrderedDict)
+                machine_settings[os.path.basename(inherited_setting_file_name)] = copy.deepcopy(data)
+                print("Inherited:", os.path.basename(inherited_setting_file_name))
+            else:
+                break
+
+        if profile:
+            profile_values = profile.getChangedSettings()
+
+        # Get total material used (in mm^3)
+        print_information = Application.getInstance().getPrintInformation()
+        material_radius = 0.5 * settings.getSettingValueByKey("material_diameter")
+        material_used = math.pi * material_radius * material_radius * print_information.materialAmount #Volume of material used
+
+        # Get model information (bounding boxes, hashes and transformation matrix)
+        models_info = []
+        for node in DepthFirstIterator(Application.getInstance().getController().getScene().getRoot()):
+            if type(node) is SceneNode and node.getMeshData() and node.getMeshData().getVertices() is not None:
+                if not getattr(node, "_outside_buildarea", False):
+                    model_info = {}
+                    model_info["hash"] = node.getMeshData().getHash()
+                    model_info["bounding_box"] = {}
+                    model_info["bounding_box"]["minimum"] = {}
+                    model_info["bounding_box"]["minimum"]["x"] = node.getBoundingBox().minimum.x
+                    model_info["bounding_box"]["minimum"]["y"] = node.getBoundingBox().minimum.y
+                    model_info["bounding_box"]["minimum"]["z"] = node.getBoundingBox().minimum.z
+
+                    model_info["bounding_box"]["maximum"] = {}
+                    model_info["bounding_box"]["maximum"]["x"] = node.getBoundingBox().maximum.x
+                    model_info["bounding_box"]["maximum"]["y"] = node.getBoundingBox().maximum.y
+                    model_info["bounding_box"]["maximum"]["z"] = node.getBoundingBox().maximum.z
+                    model_info["transformation"] = str(node.getWorldTransformation().getData())
+
+                    models_info.append(model_info)
+
+        # Bundle the collected data
+        submitted_data = {
+            "processor": platform.processor(),
+            "machine": platform.machine(),
+            "platform": platform.platform(),
+            "machine_settings": json.dumps(machine_settings),
+            "version": "15.10",#Application.getInstance().getVersion(),
+            "printtime": print_information.currentPrintTime,
+            "filament": material_used
+        }
+
+        # Convert data to bytes
+        submitted_data = urllib.parse.urlencode(submitted_data)
+        binary_data = submitted_data.encode('utf-8')
+
+        # Submit data
+        try:
+            f = urllib.request.urlopen("https://stats.youmagine.com/curastats/slice", data = binary_data, timeout = 1)
+        except Exception as e:
+            print("Exception occured", e)
+
+        print("Result: ", f.read())
+        f.close()