NetworkClusterPrinterOutputDevice.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738
  1. import datetime
  2. import getpass
  3. import gzip
  4. import json
  5. import os
  6. import os.path
  7. import time
  8. from enum import Enum
  9. from PyQt5.QtNetwork import QNetworkRequest, QHttpPart, QHttpMultiPart
  10. from PyQt5.QtCore import QUrl, pyqtSlot, pyqtProperty, QCoreApplication, QTimer, pyqtSignal, QObject
  11. from PyQt5.QtGui import QDesktopServices
  12. from PyQt5.QtNetwork import QNetworkAccessManager, QNetworkReply
  13. from UM.Application import Application
  14. from UM.Logger import Logger
  15. from UM.Message import Message
  16. from UM.OutputDevice import OutputDeviceError
  17. from UM.i18n import i18nCatalog
  18. from UM.Qt.Duration import Duration, DurationFormat
  19. from UM.PluginRegistry import PluginRegistry
  20. from . import NetworkPrinterOutputDevice
  21. i18n_catalog = i18nCatalog("cura")
  22. class OutputStage(Enum):
  23. ready = 0
  24. uploading = 2
  25. class NetworkClusterPrinterOutputDevice(NetworkPrinterOutputDevice.NetworkPrinterOutputDevice):
  26. printJobsChanged = pyqtSignal()
  27. printersChanged = pyqtSignal()
  28. selectedPrinterChanged = pyqtSignal()
  29. def __init__(self, key, address, properties, api_prefix):
  30. super().__init__(key, address, properties, api_prefix)
  31. # Store the address of the master.
  32. self._master_address = address
  33. name_property = properties.get(b"name", b"")
  34. if name_property:
  35. name = name_property.decode("utf-8")
  36. else:
  37. name = key
  38. self._authentication_state = NetworkPrinterOutputDevice.AuthState.Authenticated # The printer is always authenticated
  39. self.setName(name)
  40. description = i18n_catalog.i18nc("@action:button Preceded by 'Ready to'.", "Print over network")
  41. self.setShortDescription(description)
  42. self.setDescription(description)
  43. self._stage = OutputStage.ready
  44. host_override = os.environ.get("CLUSTER_OVERRIDE_HOST", "")
  45. if host_override:
  46. Logger.log(
  47. "w",
  48. "Environment variable CLUSTER_OVERRIDE_HOST is set to [%s], cluster hosts are now set to this host",
  49. host_override)
  50. self._host = "http://" + host_override
  51. else:
  52. self._host = "http://" + address
  53. # is the same as in NetworkPrinterOutputDevicePlugin
  54. self._cluster_api_version = "1"
  55. self._cluster_api_prefix = "/cluster-api/v" + self._cluster_api_version + "/"
  56. self._api_base_uri = self._host + self._cluster_api_prefix
  57. self._file_name = None
  58. self._progress_message = None
  59. self._request = None
  60. self._reply = None
  61. # The main reason to keep the 'multipart' form data on the object
  62. # is to prevent the Python GC from claiming it too early.
  63. self._multipart = None
  64. self._print_view = None
  65. self._request_job = []
  66. self._job_list = []
  67. self._monitor_view_qml_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "ClusterMonitorItem.qml")
  68. self._control_view_qml_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "ClusterControlItem.qml")
  69. self._print_jobs = []
  70. self._print_job_by_printer_uuid = {}
  71. self._print_job_by_uuid = {} # Print jobs by their own uuid
  72. self._printers = []
  73. self._printers_dict = {} # by unique_name
  74. self._connected_printers_type_count = []
  75. self._automatic_printer = {"unique_name": "", "friendly_name": "Automatic"} # empty unique_name IS automatic selection
  76. self._selected_printer = self._automatic_printer
  77. self._cluster_status_update_timer = QTimer()
  78. self._cluster_status_update_timer.setInterval(5000)
  79. self._cluster_status_update_timer.setSingleShot(False)
  80. self._cluster_status_update_timer.timeout.connect(self._requestClusterStatus)
  81. self._can_pause = True
  82. self._can_abort = True
  83. self._can_pre_heat_bed = False
  84. self._can_control_manually = False
  85. self._cluster_size = int(properties.get(b"cluster_size", 0))
  86. self._cleanupRequest()
  87. #These are texts that are to be translated for future features.
  88. temporary_translation = i18n_catalog.i18n("This printer is not set up to host a group of connected Ultimaker 3 printers.")
  89. temporary_translation2 = i18n_catalog.i18nc("Count is number of printers.", "This printer is the host for a group of {count} connected Ultimaker 3 printers.").format(count = 3)
  90. temporary_translation3 = i18n_catalog.i18n("{printer_name} has finished printing '{job_name}'. Please collect the print and confirm clearing the build plate.") #When finished.
  91. temporary_translation4 = i18n_catalog.i18n("{printer_name} is reserved to print '{job_name}'. Please change the printer's configuration to match the job, for it to start printing.") #When configuration changed.
  92. ## No authentication, so requestAuthentication should do exactly nothing
  93. @pyqtSlot()
  94. def requestAuthentication(self, message_id = None, action_id = "Retry"):
  95. pass # Cura Connect doesn't do any authorization
  96. def setAuthenticationState(self, auth_state):
  97. self._authentication_state = NetworkPrinterOutputDevice.AuthState.Authenticated # The printer is always authenticated
  98. def _verifyAuthentication(self):
  99. pass
  100. def _checkAuthentication(self):
  101. Logger.log("d", "_checkAuthentication Cura Connect - nothing to be done")
  102. @pyqtProperty(QObject, notify=selectedPrinterChanged)
  103. def controlItem(self):
  104. # TODO: Probably not the nicest way to do this. This needs to be done better at some point in time.
  105. if not self._control_item:
  106. self._createControlViewFromQML()
  107. name = self._selected_printer.get("friendly_name")
  108. if name == self._automatic_printer.get("friendly_name") or name == "":
  109. return self._control_item
  110. # Let cura use the default.
  111. return None
  112. @pyqtSlot(int, result = str)
  113. def getTimeCompleted(self, time_remaining):
  114. current_time = time.time()
  115. datetime_completed = datetime.datetime.fromtimestamp(current_time + time_remaining)
  116. return "{hour:02d}:{minute:02d}".format(hour = datetime_completed.hour, minute = datetime_completed.minute)
  117. @pyqtSlot(int, result = str)
  118. def getDateCompleted(self, time_remaining):
  119. current_time = time.time()
  120. datetime_completed = datetime.datetime.fromtimestamp(current_time + time_remaining)
  121. return (datetime_completed.strftime("%a %b ") + "{day}".format(day=datetime_completed.day)).upper()
  122. @pyqtProperty(int, constant = True)
  123. def clusterSize(self):
  124. return self._cluster_size
  125. @pyqtProperty(str, notify=selectedPrinterChanged)
  126. def name(self):
  127. # Show the name of the selected printer.
  128. # This is not the nicest way to do this, but changes to the Cura UI are required otherwise.
  129. name = self._selected_printer.get("friendly_name")
  130. if name != self._automatic_printer.get("friendly_name"):
  131. return name
  132. # Return name of cluster master.
  133. return self._properties.get(b"name", b"").decode("utf-8")
  134. def connect(self):
  135. super().connect()
  136. self._cluster_status_update_timer.start()
  137. def close(self):
  138. super().close()
  139. self._cluster_status_update_timer.stop()
  140. def _setJobState(self, job_state):
  141. if not self._selected_printer:
  142. return
  143. selected_printer_uuid = self._printers_dict[self._selected_printer["unique_name"]]["uuid"]
  144. if selected_printer_uuid not in self._print_job_by_printer_uuid:
  145. return
  146. print_job_uuid = self._print_job_by_printer_uuid[selected_printer_uuid]["uuid"]
  147. url = QUrl(self._api_base_uri + "print_jobs/" + print_job_uuid + "/action")
  148. put_request = QNetworkRequest(url)
  149. put_request.setHeader(QNetworkRequest.ContentTypeHeader, "application/json")
  150. data = '{"action": "' + job_state + '"}'
  151. self._manager.put(put_request, data.encode())
  152. def _requestClusterStatus(self):
  153. # TODO: Handle timeout. We probably want to know if the cluster is still reachable or not.
  154. url = QUrl(self._api_base_uri + "printers/")
  155. printers_request = QNetworkRequest(url)
  156. self._addUserAgentHeader(printers_request)
  157. self._manager.get(printers_request)
  158. # See _finishedPrintersRequest()
  159. if self._printers: # if printers is not empty
  160. url = QUrl(self._api_base_uri + "print_jobs/")
  161. print_jobs_request = QNetworkRequest(url)
  162. self._addUserAgentHeader(print_jobs_request)
  163. self._manager.get(print_jobs_request)
  164. # See _finishedPrintJobsRequest()
  165. def _finishedPrintJobsRequest(self, reply):
  166. try:
  167. json_data = json.loads(bytes(reply.readAll()).decode("utf-8"))
  168. except json.decoder.JSONDecodeError:
  169. Logger.log("w", "Received an invalid print job state message: Not valid JSON.")
  170. return
  171. self.setPrintJobs(json_data)
  172. def _finishedPrintersRequest(self, reply):
  173. try:
  174. json_data = json.loads(bytes(reply.readAll()).decode("utf-8"))
  175. except json.decoder.JSONDecodeError:
  176. Logger.log("w", "Received an invalid print job state message: Not valid JSON.")
  177. return
  178. self.setPrinters(json_data)
  179. def materialHotendChangedMessage(self, callback):
  180. # When there is just one printer, the activate configuration option is enabled
  181. if (self._cluster_size == 1):
  182. super().materialHotendChangedMessage(callback = callback)
  183. def _startCameraStream(self):
  184. ## Request new image
  185. url = QUrl("http://" + self._printers_dict[self._selected_printer["unique_name"]]["ip_address"] + ":8080/?action=stream")
  186. self._image_request = QNetworkRequest(url)
  187. self._addUserAgentHeader(self._image_request)
  188. self._image_reply = self._manager.get(self._image_request)
  189. self._image_reply.downloadProgress.connect(self._onStreamDownloadProgress)
  190. def spawnPrintView(self):
  191. if self._print_view is None:
  192. path = os.path.join(self._plugin_path, "PrintWindow.qml")
  193. self._print_view = Application.getInstance().createQmlComponent(path, {"OutputDevice": self})
  194. if self._print_view is not None:
  195. self._print_view.show()
  196. ## Store job info, show Print view for settings
  197. def requestWrite(self, nodes, file_name=None, filter_by_machine=False, file_handler=None, **kwargs):
  198. self._selected_printer = self._automatic_printer # reset to default option
  199. self._request_job = [nodes, file_name, filter_by_machine, file_handler, kwargs]
  200. # the build plates to be sent
  201. gcode_dict = getattr(Application.getInstance().getController().getScene(), "gcode_dict")
  202. self._job_list = list(gcode_dict.keys())
  203. Logger.log("d", "build plates to be sent to printer: %s", (self._job_list))
  204. if self._stage != OutputStage.ready:
  205. if self._error_message:
  206. self._error_message.hide()
  207. self._error_message = Message(
  208. i18n_catalog.i18nc("@info:status",
  209. "Sending new jobs (temporarily) blocked, still sending the previous print job."))
  210. self._error_message.show()
  211. return
  212. self._add_build_plate_number = len(self._job_list) > 1
  213. self.writeStarted.emit(self) # Allow postprocessing before sending data to the printer
  214. if len(self._printers) > 1:
  215. self.spawnPrintView() # Ask user how to print it.
  216. elif len(self._printers) == 1:
  217. # If there is only one printer, don't bother asking.
  218. self.selectAutomaticPrinter()
  219. self.sendPrintJob()
  220. else:
  221. # Cluster has no printers, warn the user of this.
  222. if self._error_message:
  223. self._error_message.hide()
  224. self._error_message = Message(
  225. i18n_catalog.i18nc("@info:status",
  226. "Unable to send new print job: this 3D printer is not (yet) set up to host a group of connected Ultimaker 3 printers."))
  227. self._error_message.show()
  228. ## Actually send the print job, called from the dialog
  229. # :param: require_printer_name: name of printer, or ""
  230. @pyqtSlot()
  231. def sendPrintJob(self):
  232. nodes, file_name, filter_by_machine, file_handler, kwargs = self._request_job
  233. output_build_plate_number = self._job_list.pop(0)
  234. gcode_list = getattr(Application.getInstance().getController().getScene(), "gcode_dict")[output_build_plate_number]
  235. if not gcode_list: # Empty build plate
  236. Logger.log("d", "Skipping empty job (build plate number %d).", output_build_plate_number)
  237. if self._job_list:
  238. return self.sendPrintJob()
  239. else:
  240. return
  241. self._send_gcode_start = time.time()
  242. Logger.log("d", "Sending print job [%s] to host, build plate [%s]..." % (file_name, output_build_plate_number))
  243. if self._stage != OutputStage.ready:
  244. Logger.log("d", "Unable to send print job as the state is %s", self._stage)
  245. raise OutputDeviceError.DeviceBusyError()
  246. self._stage = OutputStage.uploading
  247. if self._add_build_plate_number:
  248. self._file_name = "%s_%d.gcode.gz" % (file_name, output_build_plate_number)
  249. else:
  250. self._file_name = "%s.gcode.gz" % (file_name)
  251. self._showProgressMessage()
  252. require_printer_name = self._selected_printer["unique_name"]
  253. new_request = self._buildSendPrintJobHttpRequest(require_printer_name, gcode_list)
  254. if new_request is None or self._stage != OutputStage.uploading:
  255. return
  256. self._request = new_request
  257. self._reply = self._manager.post(self._request, self._multipart)
  258. self._reply.uploadProgress.connect(self._onUploadProgress)
  259. # See _finishedPrintJobPostRequest()
  260. def _buildSendPrintJobHttpRequest(self, require_printer_name, gcode_list):
  261. api_url = QUrl(self._api_base_uri + "print_jobs/")
  262. request = QNetworkRequest(api_url)
  263. # Create multipart request and add the g-code.
  264. self._multipart = QHttpMultiPart(QHttpMultiPart.FormDataType)
  265. # Add gcode
  266. part = QHttpPart()
  267. part.setHeader(QNetworkRequest.ContentDispositionHeader,
  268. 'form-data; name="file"; filename="%s"' % (self._file_name))
  269. compressed_gcode = self._compressGcode(gcode_list)
  270. if compressed_gcode is None:
  271. return None # User aborted print, so stop trying.
  272. part.setBody(compressed_gcode)
  273. self._multipart.append(part)
  274. # require_printer_name "" means automatic
  275. if require_printer_name:
  276. self._multipart.append(self.__createKeyValueHttpPart("require_printer_name", require_printer_name))
  277. user_name = self.__get_username()
  278. if user_name is None:
  279. user_name = "unknown"
  280. self._multipart.append(self.__createKeyValueHttpPart("owner", user_name))
  281. self._addUserAgentHeader(request)
  282. return request
  283. def _compressGcode(self, gcode_list):
  284. self._compressing_print = True
  285. batched_line = ""
  286. max_chars_per_line = int(1024 * 1024 / 4) # 1 / 4 MB
  287. byte_array_file_data = b""
  288. def _compressDataAndNotifyQt(data_to_append):
  289. compressed_data = gzip.compress(data_to_append.encode("utf-8"))
  290. self._progress_message.setProgress(-1) # Tickle the message so that it's clear that it's still being used.
  291. QCoreApplication.processEvents() # Ensure that the GUI does not freeze.
  292. # Pretend that this is a response, as zipping might take a bit of time.
  293. self._last_response_time = time.time()
  294. return compressed_data
  295. if gcode_list is None:
  296. Logger.log("e", "Unable to find sliced gcode, returning empty.")
  297. return byte_array_file_data
  298. for line in gcode_list:
  299. if not self._compressing_print:
  300. self._progress_message.hide()
  301. return None # Stop trying to zip, abort was called.
  302. batched_line += line
  303. # if the gcode was read from a gcode file, self._gcode will be a list of all lines in that file.
  304. # Compressing line by line in this case is extremely slow, so we need to batch them.
  305. if len(batched_line) < max_chars_per_line:
  306. continue
  307. byte_array_file_data += _compressDataAndNotifyQt(batched_line)
  308. batched_line = ""
  309. # Also compress the leftovers.
  310. if batched_line:
  311. byte_array_file_data += _compressDataAndNotifyQt(batched_line)
  312. return byte_array_file_data
  313. def __createKeyValueHttpPart(self, key, value):
  314. metadata_part = QHttpPart()
  315. metadata_part.setHeader(QNetworkRequest.ContentTypeHeader, 'text/plain')
  316. metadata_part.setHeader(QNetworkRequest.ContentDispositionHeader, 'form-data; name="%s"' % (key))
  317. metadata_part.setBody(bytearray(value, "utf8"))
  318. return metadata_part
  319. def __get_username(self):
  320. try:
  321. return getpass.getuser()
  322. except:
  323. Logger.log("d", "Could not get the system user name, returning 'unknown' instead.")
  324. return None
  325. def _finishedPrintJobPostRequest(self, reply):
  326. self._stage = OutputStage.ready
  327. if self._progress_message:
  328. self._progress_message.hide()
  329. self._progress_message = None
  330. self.writeFinished.emit(self)
  331. if reply.error():
  332. self._showRequestFailedMessage(reply)
  333. self.writeError.emit(self)
  334. else:
  335. self._showRequestSucceededMessage()
  336. self.writeSuccess.emit(self)
  337. self._cleanupRequest()
  338. if self._job_list: # start sending next job
  339. self.sendPrintJob()
  340. def _showRequestFailedMessage(self, reply):
  341. if reply is not None:
  342. Logger.log("w", "Unable to send print job to group {cluster_name}: {error_string} ({error})".format(
  343. cluster_name = self.getName(),
  344. error_string = str(reply.errorString()),
  345. error = str(reply.error())))
  346. error_message_template = i18n_catalog.i18nc("@info:status", "Unable to send print job to group {cluster_name}.")
  347. message = Message(text=error_message_template.format(
  348. cluster_name = self.getName()))
  349. message.show()
  350. def _showRequestSucceededMessage(self):
  351. confirmation_message_template = i18n_catalog.i18nc(
  352. "@info:status",
  353. "Sent {file_name} to group {cluster_name}."
  354. )
  355. file_name = os.path.basename(self._file_name).split(".")[0]
  356. message_text = confirmation_message_template.format(cluster_name = self.getName(), file_name = file_name)
  357. message = Message(text=message_text)
  358. button_text = i18n_catalog.i18nc("@action:button", "Show print jobs")
  359. button_tooltip = i18n_catalog.i18nc("@info:tooltip", "Opens the print jobs interface in your browser.")
  360. message.addAction("open_browser", button_text, "globe", button_tooltip)
  361. message.actionTriggered.connect(self._onMessageActionTriggered)
  362. message.show()
  363. def setPrintJobs(self, print_jobs):
  364. #TODO: hack, last seen messes up the check, so drop it.
  365. for job in print_jobs:
  366. del job["last_seen"]
  367. # Strip any extensions
  368. job["name"] = self._removeGcodeExtension(job["name"])
  369. if self._print_jobs != print_jobs:
  370. old_print_jobs = self._print_jobs
  371. self._print_jobs = print_jobs
  372. self._notifyFinishedPrintJobs(old_print_jobs, print_jobs)
  373. self._notifyConfigurationChangeRequired(old_print_jobs, print_jobs)
  374. # Yes, this is a hacky way of doing it, but it's quick and the API doesn't give the print job per printer
  375. # for some reason. ugh.
  376. self._print_job_by_printer_uuid = {}
  377. self._print_job_by_uuid = {}
  378. for print_job in print_jobs:
  379. if "printer_uuid" in print_job and print_job["printer_uuid"] is not None:
  380. self._print_job_by_printer_uuid[print_job["printer_uuid"]] = print_job
  381. self._print_job_by_uuid[print_job["uuid"]] = print_job
  382. self.printJobsChanged.emit()
  383. def _removeGcodeExtension(self, name):
  384. parts = name.split(".")
  385. if parts[-1].upper() == "GZ":
  386. parts = parts[:-1]
  387. if parts[-1].upper() == "GCODE":
  388. parts = parts[:-1]
  389. return ".".join(parts)
  390. def _notifyFinishedPrintJobs(self, old_print_jobs, new_print_jobs):
  391. """Notify the user when any of their print jobs have just completed.
  392. Arguments:
  393. old_print_jobs -- the previous list of print job status information as returned by the cluster REST API.
  394. new_print_jobs -- the current list of print job status information as returned by the cluster REST API.
  395. """
  396. if old_print_jobs is None:
  397. return
  398. username = self.__get_username()
  399. if username is None:
  400. return
  401. our_old_print_jobs = self.__filterOurPrintJobs(old_print_jobs)
  402. our_old_not_finished_print_jobs = [pj for pj in our_old_print_jobs if pj["status"] != "wait_cleanup"]
  403. our_new_print_jobs = self.__filterOurPrintJobs(new_print_jobs)
  404. our_new_finished_print_jobs = [pj for pj in our_new_print_jobs if pj["status"] == "wait_cleanup"]
  405. old_not_finished_print_job_uuids = set([pj["uuid"] for pj in our_old_not_finished_print_jobs])
  406. for print_job in our_new_finished_print_jobs:
  407. if print_job["uuid"] in old_not_finished_print_job_uuids:
  408. printer_name = self.__getPrinterNameFromUuid(print_job["printer_uuid"])
  409. if printer_name is None:
  410. printer_name = i18n_catalog.i18nc("@label Printer name", "Unknown")
  411. message_text = (i18n_catalog.i18nc("@info:status",
  412. "Printer '{printer_name}' has finished printing '{job_name}'.")
  413. .format(printer_name=printer_name, job_name=print_job["name"]))
  414. message = Message(text=message_text, title=i18n_catalog.i18nc("@info:status", "Print finished"))
  415. Application.getInstance().showMessage(message)
  416. Application.getInstance().showToastMessage(
  417. i18n_catalog.i18nc("@info:status", "Print finished"),
  418. message_text)
  419. def __filterOurPrintJobs(self, print_jobs):
  420. username = self.__get_username()
  421. return [print_job for print_job in print_jobs if print_job["owner"] == username]
  422. def _notifyConfigurationChangeRequired(self, old_print_jobs, new_print_jobs):
  423. if old_print_jobs is None:
  424. return
  425. old_change_required_print_jobs = self.__filterConfigChangePrintJobs(self.__filterOurPrintJobs(old_print_jobs))
  426. new_change_required_print_jobs = self.__filterConfigChangePrintJobs(self.__filterOurPrintJobs(new_print_jobs))
  427. old_change_required_print_job_uuids = set([pj["uuid"] for pj in old_change_required_print_jobs])
  428. for print_job in new_change_required_print_jobs:
  429. if print_job["uuid"] not in old_change_required_print_job_uuids:
  430. printer_name = self.__getPrinterNameFromUuid(print_job["assigned_to"])
  431. if printer_name is None:
  432. # don't report on yet unknown printers
  433. continue
  434. message_text = (i18n_catalog.i18n("{printer_name} is reserved to print '{job_name}'. Please change the printer's configuration to match the job, for it to start printing.")
  435. .format(printer_name=printer_name, job_name=print_job["name"]))
  436. message = Message(text=message_text, title=i18n_catalog.i18nc("@label:status", "Action required"))
  437. Application.getInstance().showMessage(message)
  438. Application.getInstance().showToastMessage(
  439. i18n_catalog.i18nc("@label:status", "Action required"),
  440. message_text)
  441. def __filterConfigChangePrintJobs(self, print_jobs):
  442. return filter(self.__isConfigurationChangeRequiredPrintJob, print_jobs)
  443. def __isConfigurationChangeRequiredPrintJob(self, print_job):
  444. if print_job["status"] == "queued":
  445. changes_required = print_job.get("configuration_changes_required", [])
  446. return len(changes_required) != 0
  447. return False
  448. def __getPrinterNameFromUuid(self, printer_uuid):
  449. for printer in self._printers:
  450. if printer["uuid"] == printer_uuid:
  451. return printer["friendly_name"]
  452. return None
  453. def setPrinters(self, printers):
  454. if self._printers != printers:
  455. self._connected_printers_type_count = []
  456. printers_count = {}
  457. self._printers = printers
  458. self._printers_dict = dict((p["unique_name"], p) for p in printers) # for easy lookup by unique_name
  459. for printer in printers:
  460. variant = printer["machine_variant"]
  461. if variant in printers_count:
  462. printers_count[variant] += 1
  463. else:
  464. printers_count[variant] = 1
  465. for type in printers_count:
  466. self._connected_printers_type_count.append({"machine_type": type, "count": printers_count[type]})
  467. self.printersChanged.emit()
  468. @pyqtProperty("QVariantList", notify=printersChanged)
  469. def connectedPrintersTypeCount(self):
  470. return self._connected_printers_type_count
  471. @pyqtProperty("QVariantList", notify=printersChanged)
  472. def connectedPrinters(self):
  473. return self._printers
  474. @pyqtProperty(int, notify=printJobsChanged)
  475. def numJobsPrinting(self):
  476. num_jobs_printing = 0
  477. for job in self._print_jobs:
  478. if job["status"] in ["printing", "wait_cleanup", "sent_to_printer", "pre_print", "post_print"]:
  479. num_jobs_printing += 1
  480. return num_jobs_printing
  481. @pyqtProperty(int, notify=printJobsChanged)
  482. def numJobsQueued(self):
  483. num_jobs_queued = 0
  484. for job in self._print_jobs:
  485. if job["status"] == "queued":
  486. num_jobs_queued += 1
  487. return num_jobs_queued
  488. @pyqtProperty("QVariantMap", notify=printJobsChanged)
  489. def printJobsByUUID(self):
  490. return self._print_job_by_uuid
  491. @pyqtProperty("QVariantMap", notify=printJobsChanged)
  492. def printJobsByPrinterUUID(self):
  493. return self._print_job_by_printer_uuid
  494. @pyqtProperty("QVariantList", notify=printJobsChanged)
  495. def printJobs(self):
  496. return self._print_jobs
  497. @pyqtProperty("QVariantList", notify=printersChanged)
  498. def printers(self):
  499. return [self._automatic_printer, ] + self._printers
  500. @pyqtSlot(str, str)
  501. def selectPrinter(self, unique_name, friendly_name):
  502. self.stopCamera()
  503. self._selected_printer = {"unique_name": unique_name, "friendly_name": friendly_name}
  504. Logger.log("d", "Selected printer: %s %s", friendly_name, unique_name)
  505. # TODO: Probably not the nicest way to do this. This needs to be done better at some point in time.
  506. if unique_name == "":
  507. self._address = self._master_address
  508. else:
  509. self._address = self._printers_dict[self._selected_printer["unique_name"]]["ip_address"]
  510. self.selectedPrinterChanged.emit()
  511. def _updateJobState(self, job_state):
  512. name = self._selected_printer.get("friendly_name")
  513. if name == "" or name == "Automatic":
  514. # TODO: This is now a bit hacked; If no printer is selected, don't show job state.
  515. if self._job_state != "":
  516. self._job_state = ""
  517. self.jobStateChanged.emit()
  518. else:
  519. if self._job_state != job_state:
  520. self._job_state = job_state
  521. self.jobStateChanged.emit()
  522. @pyqtSlot()
  523. def selectAutomaticPrinter(self):
  524. self.stopCamera()
  525. self._selected_printer = self._automatic_printer
  526. self.selectedPrinterChanged.emit()
  527. @pyqtProperty("QVariant", notify=selectedPrinterChanged)
  528. def selectedPrinterName(self):
  529. return self._selected_printer.get("unique_name", "")
  530. def getPrintJobsUrl(self):
  531. return self._host + "/print_jobs"
  532. def getPrintersUrl(self):
  533. return self._host + "/printers"
  534. def _showProgressMessage(self):
  535. progress_message_template = i18n_catalog.i18nc("@info:progress",
  536. "Sending <filename>{file_name}</filename> to group {cluster_name}")
  537. file_name = os.path.basename(self._file_name).split(".")[0]
  538. self._progress_message = Message(progress_message_template.format(file_name = file_name, cluster_name = self.getName()), 0, False, -1)
  539. self._progress_message.addAction("Abort", i18n_catalog.i18nc("@action:button", "Cancel"), None, "")
  540. self._progress_message.actionTriggered.connect(self._onMessageActionTriggered)
  541. self._progress_message.show()
  542. def _addUserAgentHeader(self, request):
  543. request.setRawHeader(b"User-agent", b"CuraPrintClusterOutputDevice Plugin")
  544. def _cleanupRequest(self):
  545. self._request = None
  546. self._stage = OutputStage.ready
  547. self._file_name = None
  548. def _onFinished(self, reply):
  549. super()._onFinished(reply)
  550. reply_url = reply.url().toString()
  551. status_code = reply.attribute(QNetworkRequest.HttpStatusCodeAttribute)
  552. if status_code == 500:
  553. Logger.log("w", "Request to {url} returned a 500.".format(url = reply_url))
  554. return
  555. if reply.error() == QNetworkReply.ContentOperationNotPermittedError:
  556. # It was probably "/api/v1/materials" for legacy UM3
  557. return
  558. if reply.error() == QNetworkReply.ContentNotFoundError:
  559. # It was probably "/api/v1/print_job" for legacy UM3
  560. return
  561. if reply.operation() == QNetworkAccessManager.PostOperation:
  562. if self._cluster_api_prefix + "print_jobs" in reply_url:
  563. self._finishedPrintJobPostRequest(reply)
  564. return
  565. # We need to do this check *after* we process the post operation!
  566. # If the sending of g-code is cancelled by the user it will result in an error, but we do need to handle this.
  567. if reply.error() != QNetworkReply.NoError:
  568. Logger.log("e", "After requesting [%s] we got a network error [%s]. Not processing anything...", reply_url, reply.error())
  569. return
  570. elif reply.operation() == QNetworkAccessManager.GetOperation:
  571. if self._cluster_api_prefix + "print_jobs" in reply_url:
  572. self._finishedPrintJobsRequest(reply)
  573. elif self._cluster_api_prefix + "printers" in reply_url:
  574. self._finishedPrintersRequest(reply)
  575. @pyqtSlot()
  576. def openPrintJobControlPanel(self):
  577. Logger.log("d", "Opening print job control panel...")
  578. QDesktopServices.openUrl(QUrl(self.getPrintJobsUrl()))
  579. @pyqtSlot()
  580. def openPrinterControlPanel(self):
  581. Logger.log("d", "Opening printer control panel...")
  582. QDesktopServices.openUrl(QUrl(self.getPrintersUrl()))
  583. def _onMessageActionTriggered(self, message, action):
  584. if action == "open_browser":
  585. QDesktopServices.openUrl(QUrl(self.getPrintJobsUrl()))
  586. if action == "Abort":
  587. Logger.log("d", "User aborted sending print to remote.")
  588. self._progress_message.hide()
  589. self._compressing_print = False
  590. if self._reply:
  591. self._reply.abort()
  592. self._stage = OutputStage.ready
  593. Application.getInstance().getController().setActiveStage("PrepareStage")
  594. @pyqtSlot(int, result=str)
  595. def formatDuration(self, seconds):
  596. return Duration(seconds).getDisplayString(DurationFormat.Format.Short)
  597. ## For cluster below
  598. def _get_plugin_directory_name(self):
  599. current_file_absolute_path = os.path.realpath(__file__)
  600. directory_path = os.path.dirname(current_file_absolute_path)
  601. _, directory_name = os.path.split(directory_path)
  602. return directory_name
  603. @property
  604. def _plugin_path(self):
  605. return PluginRegistry.getInstance().getPluginPath(self._get_plugin_directory_name())