def test_getAndUpdate(data): model = PrinterOutputModel(MagicMock()) # Convert the first letter into a capital attribute = list(data["attribute"]) attribute[0] = attribute[0].capitalize() attribute = "".join(attribute) # mock the correct emit setattr(model, data["attribute"] + "Changed", MagicMock()) # Attempt to set the value getattr(model, "update" + attribute)(data["value"]) # Check if signal fired. signal = getattr(model, data["attribute"] + "Changed") assert signal.emit.call_count == 1 # Ensure that the value got set assert getattr(model, data["attribute"]) == data["value"] # Attempt to set the value again getattr(model, "update" + attribute)(data["value"]) # The signal should not fire again assert signal.emit.call_count == 1
def _createPrinterList(self) -> None: printer = PrinterOutputModel( output_controller=self._output_controller, number_of_extruders=self._number_of_extruders) printer.updateName(self.name) self._printers = [printer] self.printersChanged.emit()
def test_availableConfigurations_addConfiguration(): model = PrinterOutputModel(MagicMock()) configuration = MagicMock(spec=PrinterConfigurationModel) model.addAvailableConfiguration(configuration) assert model.availableConfigurations == [configuration]
def test_uniqueConfigurations(printer_output_device): printer = PrinterOutputModel(MagicMock()) # Add a printer and fire the signal that ensures they get hooked up correctly. printer_output_device._printers = [printer] printer_output_device._onPrintersChanged() assert printer_output_device.uniqueConfigurations == [] configuration = PrinterConfigurationModel() printer.addAvailableConfiguration(configuration) assert printer_output_device.uniqueConfigurations == [configuration] # Once the type of printer is set, it's active configuration counts as being set. # In that case, that should also be added to the list of available configurations printer.updateType("blarg!") loaded_material = MaterialOutputModel(guid="", type="PLA", color="Blue", brand="Generic", name="Blue PLA") loaded_left_extruder = ExtruderConfigurationModel(0) loaded_left_extruder.setMaterial(loaded_material) loaded_right_extruder = ExtruderConfigurationModel(1) loaded_right_extruder.setMaterial(loaded_material) printer.printerConfiguration.setExtruderConfigurations( [loaded_left_extruder, loaded_right_extruder]) assert printer_output_device.uniqueConfigurations == [ configuration, printer.printerConfiguration ]
def createOutputModel( self, controller: PrinterOutputController) -> PrinterOutputModel: model = PrinterOutputModel(controller, len(self.configuration), firmware_version=self.firmware_version) self.updateOutputModel(model) return model
def _onGlobalContainerStackChanged(self): container_stack = CuraApplication.getInstance().getGlobalContainerStack() num_extruders = container_stack.getProperty("machine_extruder_count", "value") # Ensure that a printer is created. controller = GenericOutputController(self) controller.setCanUpdateFirmware(True) self._printers = [PrinterOutputModel(output_controller = controller, number_of_extruders = num_extruders)] self._printers[0].updateName(container_stack.getName())
def _createPrinterModel(self, data: Dict[str, Any]) -> PrinterOutputModel: printer = PrinterOutputModel( output_controller=ClusterUM3PrinterOutputController(self), number_of_extruders=self._number_of_extruders) printer.setCameraUrl( QUrl("http://" + data["ip_address"] + ":8080/?action=stream")) self._printers.append(printer) return printer
def test_removeAlreadyRemovedConfiguration(): model = PrinterOutputModel(MagicMock()) configuration = MagicMock(spec=PrinterConfigurationModel) model.availableConfigurationsChanged = MagicMock() model.removeAvailableConfiguration(configuration) assert model.availableConfigurationsChanged.emit.call_count == 0 assert model.availableConfigurations == []
def createOutputModel( self, controller: PrinterOutputController) -> PrinterOutputModel: # FIXME # Note that we're using '2' here as extruder count. We have hardcoded this for now to prevent issues where the # amount of extruders coming back from the API is actually lower (which it can be if a printer was just added # to a cluster). This should be fixed in the future, probably also on the cluster API side. model = PrinterOutputModel(controller, 2, firmware_version=self.firmware_version) self.updateOutputModel(model) return model
def createOutputModel( self, controller: PrinterOutputController) -> PrinterOutputModel: """Creates a new output model. :param controller: - The controller of the model. """ model = PrinterOutputModel(controller, len(self.configuration), firmware_version=self.firmware_version) self.updateOutputModel(model) return model
def test_availableConfigurations_addConfigTwice(): model = PrinterOutputModel(MagicMock()) configuration = MagicMock(spec=PrinterConfigurationModel) model.setAvailableConfigurations([configuration]) assert model.availableConfigurations == [configuration] # Adding it again should not have any effect model.addAvailableConfiguration(configuration) assert model.availableConfigurations == [configuration]
def _build_printer_output_model(self) -> PrinterOutputModel: """Returns printer Output Model for this device.""" printer_output_model = PrinterOutputModel( output_controller=self._printer_output_controller, number_of_extruders=_NUM_EXTRUDERS) printer_output_model.updateKey(self._address) printer_output_model.updateName(self.address) printer_output_model.updateState('idle') printer_output_model.setAvailableConfigurations( [_build_printer_conf_model()]) printer_output_model.updateType('Monoprice Select Mini') printer_output_model.updateActivePrintJob(self._print_job_model) return printer_output_model
def test_uniqueConfigurations_empty_is_filtered_out(printer_output_device): printer = PrinterOutputModel(MagicMock()) # Add a printer and fire the signal that ensures they get hooked up correctly. printer_output_device._printers = [printer] printer_output_device._onPrintersChanged() printer.updateType("blarg!") empty_material = MaterialOutputModel(guid = "", type = "empty", color = "empty", brand = "Generic", name = "Empty") empty_left_extruder = ExtruderConfigurationModel(0) empty_left_extruder.setMaterial(empty_material) empty_right_extruder = ExtruderConfigurationModel(1) empty_right_extruder.setMaterial(empty_material) printer.printerConfiguration.setExtruderConfigurations([empty_left_extruder, empty_right_extruder]) assert printer_output_device.uniqueConfigurations == []
def test_peripherals(): model = PrinterOutputModel(MagicMock()) model.peripheralsChanged = MagicMock() peripheral = MagicMock(spec=Peripheral) peripheral.name = "test" peripheral2 = MagicMock(spec=Peripheral) peripheral2.name = "test2" model.addPeripheral(peripheral) assert model.peripheralsChanged.emit.call_count == 1 model.addPeripheral(peripheral2) assert model.peripheralsChanged.emit.call_count == 2 assert model.peripherals == "test, test2" model.removePeripheral(peripheral) assert model.peripheralsChanged.emit.call_count == 3 assert model.peripherals == "test2"
def _conectionStateChanged(self, new_state): if new_state == True: container_stack = CuraApplication.getInstance( ).getGlobalContainerStack() num_extruders = container_stack.getProperty( "machine_extruder_count", "value") # Ensure that a printer is created. printer = PrinterOutputModel( output_controller=self._output_controller, number_of_extruders=num_extruders, firmware_version=self.firmwareVersion) printer.updateName(container_stack.getName()) self._printers = [printer] self.setConnectionState(ConnectionState.Connected) self.printersChanged.emit() else: #self._printers = None self.setConnectionState(ConnectionState.Connecting) if self.printers[0]: self.printers[0].updateState("offline")
"value": ["yay"] }, ] test_validate_data_get_update = [ { "attribute": "configuration", "value": PrinterConfigurationModel() }, { "attribute": "owner", "value": "WHOO" }, { "attribute": "assignedPrinter", "value": PrinterOutputModel(MagicMock()) }, { "attribute": "key", "value": "YAY" }, { "attribute": "name", "value": "Turtles" }, { "attribute": "timeTotal", "value": 10 }, { "attribute": "timeElapsed",
def _onGetPrinterDataFinished(self, reply): status_code = reply.attribute(QNetworkRequest.HttpStatusCodeAttribute) if status_code == 200: try: result = json.loads(bytes(reply.readAll()).decode("utf-8")) except json.decoder.JSONDecodeError: Logger.log("w", "Received an invalid printer state message: Not valid JSON.") return if not self._printers: # Quickest way to get the firmware version is to grab it from the zeroconf. firmware_version = self._properties.get(b"firmware_version", b"").decode("utf-8") self._printers = [PrinterOutputModel(output_controller=self._output_controller, number_of_extruders=self._number_of_extruders, firmware_version=firmware_version)] self._printers[0].setCameraUrl(QUrl("http://" + self._address + ":8080/?action=stream")) for extruder in self._printers[0].extruders: extruder.activeMaterialChanged.connect(self.materialIdChanged) extruder.hotendIDChanged.connect(self.hotendIdChanged) self.printersChanged.emit() # LegacyUM3 always has a single printer. printer = self._printers[0] printer.updateBedTemperature(result["bed"]["temperature"]["current"]) printer.updateTargetBedTemperature(result["bed"]["temperature"]["target"]) printer.updateState(result["status"]) try: # If we're still handling the request, we should ignore remote for a bit. if not printer.getController().isPreheatRequestInProgress(): printer.updateIsPreheating(result["bed"]["pre_heat"]["active"]) except KeyError: # Older firmwares don't support preheating, so we need to fake it. pass head_position = result["heads"][0]["position"] printer.updateHeadPosition(head_position["x"], head_position["y"], head_position["z"]) for index in range(0, self._number_of_extruders): temperatures = result["heads"][0]["extruders"][index]["hotend"]["temperature"] extruder = printer.extruders[index] extruder.updateTargetHotendTemperature(temperatures["target"]) extruder.updateHotendTemperature(temperatures["current"]) material_guid = result["heads"][0]["extruders"][index]["active_material"]["guid"] if extruder.activeMaterial is None or extruder.activeMaterial.guid != material_guid: # Find matching material (as we need to set brand, type & color) containers = ContainerRegistry.getInstance().findInstanceContainers(type="material", GUID=material_guid) if containers: color = containers[0].getMetaDataEntry("color_code") brand = containers[0].getMetaDataEntry("brand") material_type = containers[0].getMetaDataEntry("material") name = containers[0].getName() else: # Unknown material. color = "#00000000" brand = "Unknown" material_type = "Unknown" name = "Unknown" material = MaterialOutputModel(guid=material_guid, type=material_type, brand=brand, color=color, name = name) extruder.updateActiveMaterial(material) try: hotend_id = result["heads"][0]["extruders"][index]["hotend"]["id"] except KeyError: hotend_id = "" printer.extruders[index].updateHotendID(hotend_id) else: Logger.log("w", "Got status code {status_code} while trying to get printer data".format(status_code = status_code))