NetworkClusterPrinterOutputDevice.py 28 KB

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