NetworkClusterPrinterOutputDevice.py 30 KB

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